mybatis查询返回map的问题

2年前 (2022) 程序员胖胖胖虎阿
269 0 0

文章目录

  • 背景
  • 1、mybatis只返回单个map
  • 2、查询返回map的list
  • 3、利用mybatis的@MapKey注解返回map
  • 4、重写handler

背景

假设背景:
想获取某个省下各个市有多少问题,以
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.csdn.net/sou_liu/article/details/47755635

版权声明:程序员胖胖胖虎阿 发表于 2022年11月3日 下午6:24。
转载请注明:mybatis查询返回map的问题 | 胖虎的工具箱-编程导航

相关文章

暂无评论

暂无评论...