mybatis查询多个结果返回map--@MapKey使用方法

柔情只为你懂 2024-03-31 13:23 66阅读 0赞

##  ##

**目录**

介绍

源码分析：

--------------------

## 介绍 ##

在使用mybatis的时候遇到一种情况，查询结果集返回多个，想用map接收，以前是用List<对象>接收，然后java代码中转换成map。其实mybatis是支持返回map的。下面就介绍使用方法。

测试代码：

@GetMapping(value = "dd")
        public String test4(String params){
    
            log.info("dsds" + params);
    
            Map<Long, BusinessPricePO> map = businessPriceMapper.getmaphh();
            log.info("dsva ",map);
            // 保存数据
            return "1122";
        }

下面@MapKey中的id是BusinessPricePO中的一个属性id，否则会报错：There is no getter for property named 'id' in 'class java.lang.Class'

public interface BusinessPriceMapper extends BaseMapper<BusinessPricePO> {
    
        @MapKey("id")
        Map<Long, BusinessPricePO> getmaphh();
    }

<select id="getmaphh" resultType="com.ccreate.cnpc.apply.entity.BusinessPricePO">
            SELECT id,apply_id FROM business_price LIMIT 10
        </select>

查询结果：

![e9850283084f45198db18e806307cad4.png][]

## 源码分析： ##

在spring-boot继承mybatis，执行的时候会进入com.baomidou.mybatisplus.core.override.MybatisMapperMethod\#execute

这个类，不同继承方式不同，这个自己打断点进入，查看

![bef4d58f7d0547eab87413fafb0a6a9a.png][]

执行进入：com.baomidou.mybatisplus.core.override.MybatisMapperMethod\#executeForMap

private <K, V> Map<K, V> executeForMap(SqlSession sqlSession, Object[] args) {
            Map<K, V> result;
            Object param = method.convertArgsToSqlCommandParam(args);
            if (method.hasRowBounds()) {
                RowBounds rowBounds = method.extractRowBounds(args);
                result = sqlSession.selectMap(command.getName(), param, method.getMapKey(), rowBounds);
            } else {
    // 执行这里的查询map方法
                result = sqlSession.selectMap(command.getName(), param, method.getMapKey());
            }
            return result;
        }

再执行：org.apache.ibatis.session.defaults.DefaultSqlSession\#selectMap(java.lang.String, java.lang.Object, java.lang.String, org.apache.ibatis.session.RowBounds)

![42c4172fe229459199bacf9b7fbe6f5b.png][]

public <K, V> Map<K, V> selectMap(String statement, Object parameter, String mapKey, RowBounds rowBounds) {
        final List<? extends V> list = selectList(statement, parameter, rowBounds);
        final DefaultMapResultHandler<K, V> mapResultHandler = new DefaultMapResultHandler<>(mapKey,
                configuration.getObjectFactory(), configuration.getObjectWrapperFactory(), configuration.getReflectorFactory());
        final DefaultResultContext<V> context = new DefaultResultContext<>();
        for (V o : list) {
    // 将返回结果，循环处理放入context中
          context.nextResultObject(o);
    // 这里再进行转换
          mapResultHandler.handleResult(context);
        }
    // 处理完，再拿出转换后的结果，返回
        return mapResultHandler.getMappedResults();
      }

处理返回Map的handler

org.apache.ibatis.executor.result.DefaultMapResultHandler\#handleResult

public void handleResult(ResultContext<? extends V> context) {
        final V value = context.getResultObject();
        final MetaObject mo = MetaObject.forObject(value, objectFactory, objectWrapperFactory, reflectorFactory);
        // TODO is that assignment always true?
    // 通过反射，找到key的get方法，再获取值
        final K key = (K) mo.getValue(mapKey);
    // 将对象转换为map
        mappedResults.put(key, value);
      }

最终返回。

[e9850283084f45198db18e806307cad4.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/31/f3b5fcfd27b8498dae6b1684a17596da.png
[bef4d58f7d0547eab87413fafb0a6a9a.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/31/879b62e8371e445cb6d37e76c12def15.png
[42c4172fe229459199bacf9b7fbe6f5b.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/31/038d6fe36a564bd8906f1a2e459bbc57.png