前言

虽然目前我们大多使用Mybatis、JPA进行数据操作，但是在特殊情况下，我们还是可能通过JDBCTemplate继续对数据库进行操作的。前天在项目开发中，使用JDBCTemplate进行开发时发现一个异常。

问题重现

就是如果使用JDBCTemplate.queryForObject进行查询数据，如果查询数据为空（null）时会抛出异常。

org.springframework.dao.IncorrectResultSizeDataAccessException: Incorrect result size: expected 1, actual 2

问题分析：
通过追查源码发现是因为，在源码中方法返回结果判断时如果为空就抛出throw new EmptyResultDataAccessException(1);异常；

如果返回值的数量大于1 的时候也会抛出：throw new IncorrectResultSizeDataAccessException(1, results.size());的异常。

但是我想要的功能就是如果数据库为空那就返回空，而不是抛出异常。

源码：

@Nullable
    public <T> T queryForObject(String sql, @Nullable Object[] args, RowMapper<T> rowMapper) throws DataAccessException {
        List<T> results = (List)this.query((String)sql, (Object[])args, (ResultSetExtractor)(new RowMapperResultSetExtractor(rowMapper, 1)));
        return DataAccessUtils.nullableSingleResult(results);
    }

    @Nullable
    public static <T> T nullableSingleResult(@Nullable Collection<T> results) throws IncorrectResultSizeDataAccessException {
        if (CollectionUtils.isEmpty(results)) {
            throw new EmptyResultDataAccessException(1);
        } else if (results.size() > 1) {
            throw new IncorrectResultSizeDataAccessException(1, results.size());
        } else {
            return results.iterator().next();
        }
    }

解决方案

查询结果为空抛出异常解决方法

如果是因为查询结果为空的情况报错：解决方案如下
其实解决方法有两种，分别为：

通过try catch的方式如果抛出这个异常就，设定返回值为空（不推荐，因为不经不准确，而且影响效率）
重写这个判断方法

首相我们写一个增强类继承 JDBCTemplate.
然后将对应的queryForObject方法进行重写。
修改其判断方法。然后在创建JDBCTemplate的时候，创建我们自己定义的这个就可以了。
注意：自己定义的类的构造器要调用父类的（JDBCTemplate）具体实现看下方代码。
在这里插入图片描述

package cn.hncht.plug.qywx.util;

import org.springframework.dao.DataAccessException;
import org.springframework.dao.IncorrectResultSizeDataAccessException;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.RowMapper;
import org.springframework.jdbc.core.RowMapperResultSetExtractor;
import org.springframework.lang.Nullable;
import org.springframework.util.CollectionUtils;

import javax.sql.DataSource;
import java.util.Collection;
import java.util.List;

public class JdbcTemplateEnhance extends JdbcTemplate {

    public JdbcTemplateEnhance () {
    }

    public JdbcTemplateEnhance (DataSource dataSource) {
        super(dataSource);
    }

    @Override
    public <T> T queryForObject(String sql, RowMapper<T> rowMapper, Object... args) throws DataAccessException {
        List<T> query = super.query(sql, args, new RowMapperResultSetExtractor<>(rowMapper, 1));
        return canReturnEmptyResultSet(query);
    }


    private static <T> T canReturnEmptyResultSet(@Nullable Collection<T> results) throws IncorrectResultSizeDataAccessException {
        if (CollectionUtils.isEmpty(results)) {
            return null;
        } else if (results.size() > 1) {
            throw new IncorrectResultSizeDataAccessException(1, results.size());
        } else {
            return results.iterator().next();
        }
    }

}