Spring Boot整合Mybatis查询结果封装为Map类型

2018-06-28  本文已影响0人  李北北

使用Mybatis查询时候,将返回类型resultType设置为map,但是当查询结果中为null的字段不会被封装到map对象中,在网上看到的答案要么对我的项目无效,要么就需要配置mybatis-config.xml文件,但是我的项目没有这个文件。于是就自己追踪Mybatis源码执行过程,最后终于找到了解决办法。

解决办法就是在@Configuration配置类中配置SqlSessionFactory的@Bean方法内加入如下代码

//当查询结果封装为map对象时候, 让值为null的字段也放到map中
org.apache.ibatis.session.Configuration configuration = new org.apache.ibatis.session.Configuration();
configuration.setCallSettersOnNulls(true);
sessionFactoryBean.setConfiguration(configuration);

下面大致记录一下追踪源码过程
1、首先是在mybatis入口方法处打断点,然后请求执行方法。

public RespData selectList(DataMap param) {
        RespData data = new RespData();
        Page<XXXExample> page = new Page<>();
        page.setMap(param);
        page.setOrderBy("sequenceNo");
        List<DataMap> list = this.XXXMapperExt.selectList(page);//在该行代码打断点,然后按F5进入方法,经过一系列的反射之后会进入到Mybatis的核心类核心方法中
        data.setData(list);
        data.setTotal(list.size());
        return data;
}

2、追踪代码的目标是为了找到将数据封装到Map中的源码,具体过程不细述,下面是最终找到的源码

下面的方法所在类:org.apache.ibatis.executor.resultset.DefaultResultSetHandler

private boolean applyAutomaticMappings(ResultSetWrapper rsw, ResultMap resultMap, MetaObject metaObject, String columnPrefix) throws SQLException {
    List<UnMappedColumnAutoMapping> autoMapping = createAutomaticMappings(rsw, resultMap, metaObject, columnPrefix);
    boolean foundValues = false;
    if (autoMapping.size() > 0) {
      for (UnMappedColumnAutoMapping mapping : autoMapping) {
        final Object value = mapping.typeHandler.getResult(rsw.getResultSet(), mapping.column);
        if (value != null) {
          foundValues = true;
        }

        //此处是最关键的代码,当字段value不为null,或者isCallSettersOnNulls为true且value对象不是原始数据类型时候,才会以字段为key,value为值放入map中
        if (value != null || (configuration.isCallSettersOnNulls() && !mapping.primitive)) {
          // gcode issue #377, call setter on nulls (value is not 'found')
          metaObject.setValue(mapping.property, value);
        }
      }
    }
    return foundValues;
  }

从上面代码中可以看到,我们如果能够让configuration.isCallSettersOnNulls()返回的结果是true,那么值为null的字段也会被放入map对象中了,下面就该找到初始化configuration对象的地方。

3、接下来从上,上面类org.apache.ibatis.executor.resultset.DefaultResultSetHandler中的成员变量configuration入手,通过追踪其所有可能的被赋值的方法的上级调用关系,最终找到configuration对象的最初来源,就在下面的类中

所在类:org.apache.ibatis.session.defaults.DefaultSqlSessionFactory

private final Configuration configuration;
public DefaultSqlSessionFactory(Configuration configuration) {
    this.configuration = configuration;
}
private SqlSession openSessionFromDataSource(ExecutorType execType, TransactionIsolationLevel level, boolean autoCommit) {
    Transaction tx = null;
    try {
      final Environment environment = configuration.getEnvironment();
      final TransactionFactory transactionFactory = getTransactionFactoryFromEnvironment(environment);
      tx = transactionFactory.newTransaction(environment.getDataSource(), level, autoCommit);
      final Executor executor = configuration.newExecutor(tx, execType);
      return new DefaultSqlSession(configuration, executor, autoCommit);
    } catch (Exception e) {
      closeTransaction(tx); // may have fetched a connection so lets call close()
      throw ExceptionFactory.wrapException("Error opening session.  Cause: " + e, e);
    } finally {
      ErrorContext.instance().reset();
    }
  }

4、找到了configuration对象的最初来源,接下来就是找到程序启动时候初始化这个对象的地方,通过上面3的代码可以看到DefaultSqlSessionFactory类中的有一个有参构造方法,通过追踪这个方法被调用的上级代码,最终找到SqlSessionFactoryBean这个类,其部分代码如下:

类:org.mybatis.spring.SqlSessionFactoryBean

private Configuration configuration;
protected SqlSessionFactory buildSqlSessionFactory() throws IOException {

    Configuration configuration;
    。。。
    XMLConfigBuilder xmlConfigBuilder = null;
    if (this.configuration != null) {
      configuration = this.configuration;
      。。。。
    } 
    。。。。

    //这个方法下面的调用中有初始化第3部中的configuration的代码
    return this.sqlSessionFactoryBuilder.build(configuration);
  }

5、由此可见上面就是最初初始化configuration的代码所在,而我们的配置类中定义@Bean SqlSessionFactory的方法中就创建SqlSessionFactoryBean对象,SqlSessionFactoryBean类中有设置configuration的方法,下面是最终解决办法,其中前文提到解决办法所需要加入的代码就在下面中文注释的下方

    @Bean
    SqlSessionFactory sqlSessionFactory(@Qualifier("masterDataSource") DataSource masterDataSource,
                                        @Qualifier("slaverDataSource") DataSource slaverDataSource) {
        SqlSessionFactory factory = null;
        try {
            SqlSessionFactoryBean sessionFactoryBean = new SqlSessionFactoryBean();
            sessionFactoryBean.setDataSource(dynamicDataSource(masterDataSource, slaverDataSource));

            PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();

            org.springframework.core.io.Resource[] resources = resolver.getResources("classpath*:/*/mybatis/**/*.xml");
            List<org.springframework.core.io.Resource> filterResourceList = new ArrayList<org.springframework.core.io.Resource>();
            List<String> fileNameList = new ArrayList<String>();
            for (int i = 0; i < resources.length; i++) {
                org.springframework.core.io.Resource resource = resources[i];
                if (!fileNameList.contains(resource.getFilename())) {
                    filterResourceList.add(resource);
                    fileNameList.add(resource.getFilename());
                }
            }

            org.springframework.core.io.Resource[] result = new org.springframework.core.io.Resource[filterResourceList.size()];
            sessionFactoryBean.setMapperLocations(filterResourceList.toArray(result));
            
            //当查询结果封装为map对象时候, 让值为null的字段也放到map中
            org.apache.ibatis.session.Configuration configuration = new org.apache.ibatis.session.Configuration();
            configuration.setCallSettersOnNulls(true);
            sessionFactoryBean.setConfiguration(configuration);
            
            factory = (SqlSessionFactory) sessionFactoryBean.getObject();
        } catch (Exception e) {
            e.printStackTrace();
        }
        return factory;
    }

6、最附上两张调试过程打断点的截图


image.png image.png
上一篇下一篇

猜你喜欢

热点阅读