如何使用JdbcTemplate按ID从数据库查询单条记录并优化实现？

ResultSetExtractor 实现方案 #

你担心的问题确实存在：使用RowMapper作为参数调用jdbcTemplate.query()时，方法会遍历整个ResultSet，把所有行都转换为对应的实体对象存入List后才返回，哪怕你只需要第一条数据，结果集较大时会造成不必要的内存开销。使用ResultSetExtractor可以自主控制ResultSet的遍历逻辑，拿到需要的结果就终止遍历，不会创建多余的对象，实现代码如下：

public Optional<Person> getPerson(int id) {
    String sql = "SELECT id, name FROM people WHERE id = ?";
    // ResultSetExtractor是函数式接口，可直接用lambda实现
    return jdbcTemplate.query(sql, resultSet -> {
        // 只读取第一行数据，不会遍历后续结果
        if (resultSet.next()) {
            return Optional.of(new Person(
                resultSet.getInt("id"),
                resultSet.getString("name")
            ));
        }
        return Optional.empty();
    }, id);
}

注意：如果你的查询条件是主键id，本身就具备唯一性，不会出现返回10万条结果的情况，上述优化更适用于可能返回大量结果、但你只需要读取部分数据的查询场景

主键单条查询的精简方案 #

如果你的业务场景就是按唯一主键查询单条数据，还可以直接使用jdbcTemplate.queryForObject()方法，更贴合使用场景，仅需要处理空结果的特殊情况即可：

public Optional<Person> getPerson(int id) {
    String sql = "SELECT id, name FROM people WHERE id = ?";
    try {
        Person person = jdbcTemplate.queryForObject(sql, (result, index) -> new Person(
            result.getInt("id"),
            result.getString("name")
        ), id);
        return Optional.ofNullable(person);
    } catch (EmptyResultDataAccessException e) {
        // 无匹配记录时返回空Optional
        return Optional.empty();
    }
}

try catch 的使用说明 #

• 基础资源与异常封装：JdbcTemplate内部已经实现了ResultSet、Connection、Statement的自动关闭，同时把JDBC层的受检SQLException转换为Spring的非受检DataAccessException系列异常抛出，你不需要手动捕获底层SQL异常、也不需要手动处理资源释放。
• 业务级异常处理：是否需要加try catch完全取决于你的业务需求：
  • 如果需要对特定异常做业务语义转换，比如上面的单条查询空结果场景，捕获EmptyResultDataAccessException转为空Optional返回，符合业务预期
  • 系统级异常（如SQL语法错误、数据库连接失败、权限不足等）不需要捕获，直接抛给上层统一异常拦截器处理即可

内容的提问来源于stack exchange，提问作者Dave