Java MyBatis返回两个字段作为Map的key和value

使用 MyBatis 时，可能会遇到这种情况：只查询两个字段，需要返回一个 Map，其中第一个字段作为 key，第二个字段作为 value。

这种查询在某些场景非常好用，比如查询字典，查询出的 key 和 value 就是字典的 value 和 label，利用 HashMap 的 get 方法时间复杂度为 O(1) 的特点，可以实现字典的快速映射。

此时的写法是有讲究的：

• 直接返回 Map：当数据量超过 1 时，会报错。因为此时将 Map 作为一个对象，一个对象不能存储多个内容。
• 加上 <font style="color:rgb(77, 77, 77);">@MapKey</font>注解：需要指定返回的哪个字段作为 key。虽然能生成 Map，但其中的 value 是一个对象，不是直接值，与预期不符。

要达到预期的返回值，需要使用 MyBatis 提供的 ResultHandler 结果处理器。

1. 自定义一个 Map 结果处理器 <font style="color:#080808;background-color:#ffffff;">MapResultHandler</font>

public class MapResultHandler<K,V> implements ResultHandler<Map<K,V>> {
    
    private final Map<K,V> mappedResults = new HashMap<>();

    @Override
    public void handleResult(ResultContext context) {
        Map map = (Map) context.getResultObject();
        mappedResults.put((K)map.get("key"), (V)map.get("value"));
    }

    public Map<K,V> getMappedResults() {
        return mappedResults;
    }
}

2. 编写 Mapper.java

void selectMap(MapResultHandler<String, String> mapResultHandler);

如果要传参，在前面指定即可，比如：

void selectMap(@Param("type") String type, MapResultHandler<String, String> mapResultHandler);

3. 编写 Mapper.xml

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

<select id="selectMap" resultMap="mapResult">
    SELECT
        t1.`value`,
        t1.`label`
    FROM
        sys_dict t1
    WHERE
        t1.`type` = #{type}
        AND t1.`del_flag` = '0'
</select>

4. 调用

MapResultHandler<String, String> resultHandler = new MapResultHandler<>();
baseMapper.selectMapByLanguage(Global.getLanguage(), resultHandler);
Map<String, String> map = resultHandler.getMappedResults();

这样得到的 Map 就是将两个字段作为 Map 的 key 和 value 了。