mybatis查询返回map的问题

背景

假设背景：
想获取某个省下各个市有多少问题，以
Map<String, Integer> 的形式返回，key 代表某个市的问题个数
数据库表为：

prov_code varchar  代表省的编码
city_code varchar  代表市的编码
problem varchar    代表拥有的某个问题

所以有一条该市的数据就说明该市多了一个问题，假设500为江苏省的编码，510位南京市的编码，520为苏州市的编码，数据为

500  510  问题1
500  510  问题2
500  510  问题3
500  520  问题1
500  520  问题3

正确返回的某个省下各个市有多少问题的map结果应为

“510”：3，
“520”：2

看下以下几种写法

1、mybatis只返回单个map

mapper接口：

Map<String,Object> selectCount(String provCode);

mybatis的xml文件中：

<select id="selectCount" resultType="map">
    select city_code as "cityCode",
    count(*) as "count"
    from prov_code_count
    where prov_code = #{provCode}
    group by city_code
  </select>

此时会报错，期待的是一个map，但是得到的结果是两个map，为

”cityCode“：”南京","count":3
”cityCode“：”苏州","count":2

可见mybatis返回的map是只能返回一个的，遇到多的就会报错，会将select后面的字段的名称作为key，值作为value。可参考：https://www.codeleading.com/article/52893335081/

2、查询返回map的list

这是第一个正确写法。
想要获取想要的效果，mybatis返回的是一个list的map，每个map同上一个结果，但是是list形式的

[
{”cityCode“：”南京","count":3},
{”cityCode“：”苏州","count":2}
]

mapper接口：

List<Map<String,Object>> selectCount(String provCode);

xml文件：

<select id="selectCount" resultType="map">
    select city_code as "cityCode",
    count(*) as "count"
    from prov_code_count
    where prov_code = #{provCode}
    group by city_code
  </select>

service层：

Map<String, Integer> branch2Count = toBranch2Count(mapper.selectCount(provCode);

private Map<String, Integer> toBranch2Count(List<Map<String, Object>> resMapList) {
        if (CollectionUtils.isEmpty(resMapList)) {
            return Collections.emptyMap();
        }

        Map<String, Integer> branch2Count = new HashMap<>(resMapList.size());
        resMapList.forEach(
                item -> {
                    branch2Count.put(String.valueOf(item.get("cityCode")), Integer.parseInt(String.valueOf(item.get("count"))));
                }
        );

        return branch2Count;
    }

可以看出这种方式mapper是返回的list形式的map，然后再在service层转换成所需的map的形式。

3、利用mybatis的@MapKey注解返回map

要先定义一个do对象，利用mybatis的@MapKey注解
do对象，分别为城市的编码，和该城市的问题的数量

Branch2CountDO.java
private String branchCode;
private Integer count;

mapper接口：

@MapKey("branchCode")
Map<String,Branch2CountDO> selectCount(String provCode);

xml文件：
先定义resultmap，再写sql

<resultMap id="result1" type="xxx.Branch2CountDO">
    <result column="city_code" property="branchCode"/>
    <result column="count" property="count"/>
</resultMap>

<select id="selectCount" resultMap="result1">
    select city_code , count(*) as "count"
    from prov_code_count where prov_code = #{provCode}
    group by city_code
</select>

service层：

Map<String, Branch2CountDO> res = mapper.selectCount(provCode);
Map<String, Integer> branch2Count = res.keySet().stream().collect(Collectors.toMap(branchCode -> branchCode, branchCode -> res.get(branchCode).getCount())

4、重写handler

参考：https://blog.youkuaiyun.com/sou_liu/article/details/47755635