MySQL.MyBatis怎么将查询的两个字段作为Map的key和value

问题的由来

前端使用Echarts图标显示汇总数据。需要形式如下的数据：

{
  "C20": 42.01,
  "C30": 1693.10,
  "C35": 231.00
}

但是经过以下的SQL查询

SELECT
	gsom.intensity_level AS  强度等级,
	SUM(gsom.quantity) AS  生产方量
FROM
	gt_szls_out_master gsom
WHERE org_id = '8100000000002334' AND dev_sn IN('00010001')
GROUP BY
	gsom.intensity_level

数据库查询出来的数据结果如下：

解决方案

其实观察一下前端需要的代码形式，我们只需要把SQL查询的结果中强度等级作为map的key，生产方量作为map的value即可。

实现上面的方案有一下几种方法：

• 1、可以将查询的结果在java代码中进行遍历，组装成相应的结构返回给前端
• 2、使用注解+拦截器的方式直接返回符合条件的Map（后面会讲）
• 3、就是我们本次要讲的实现方式：通过改造SQL以及处理返回结果

1、MapResultHandler.java

import org.apache.ibatis.session.ResultContext;
import org.apache.ibatis.session.ResultHandler;

import java.util.HashMap;
import java.util.Map;

/**
 * @Desc
 * @Creator Hardy
 * @Create_Date 2022/3/4 9:40
 * @Updater
 * @Update_Date
 **/
public class MapResultHandler implements ResultHandler<Map> {
    private final Map<String, String> mappedResults = new HashMap();

    @Override
    public void handleResult(ResultContext resultContext) {
        Map<String, String> m = (Map) resultContext.getResultObject();
        mappedResults.put(m.get("key"), m.get("value"));
    }

    public Map getMappedResults() {
        return mappedResults;
    }
}

2、xxxDao.java

void getMixingPlantProductionSummary(@Param("orgId") Long orgId, @Param("devSNs") String devSNs, MapResultHandler mapResultHandler);

3、Mybatis.xml

<resultMap id="mapResult" type="java.util.HashMap">
    <result property="key" column="name"/>
    <result property="value" column="count"/>
</resultMap>

<select id="getMixingPlantProductionSummary" resultMap="mapResult">
    SELECT
    gsom.intensity_level AS name,
    SUM(gsom.quantity) AS count
    FROM
    gt_szls_out_master gsom
    <where>
        <if test="orgId != null">
            and gsom.org_id = #{orgId}
        </if>

        <if test="devSNs != null">
            and gsom.dev_sn in (#{devSNs})
        </if>
    </where>
    GROUP BY gsom.intensity_level
</select>

4、使用方式

MapResultHandler handler = new MapResultHandler();
xxxDao.getMixingPlantProductionSummary(orgId, devSNs, handler);
Map resultMap = handler.getMappedResults();

注意：

1、xxxDao.java中定义的接口返回值类型必须是void，如上面void getMixingPlantProductionSummary()

2、MyBatis.xml中定义的resultMap返回值类型是HashMap，其里面定义的两个result中的key要与MapResultHandler.java中代码mappedResults.put(m.get("key"), m.get("value"))保持一致；其value属性必须和Mybatis.xml中sql的返回值名称一致

3、Map resultMap = handler.getMappedResults();中的resultMap就是我们一开始讲到的前端需要的数据类型