Java MyBatis返回两个字段作为Map的key和value问题
作者：天航星
使用MyBatis查询两个字段并返回Map时,需要注意数据量和值的类型,直接返回Map会导致报错,使用@MapKey注解可以生成Map,但值是对象而不是直接值,为了解决这个问题,可以自定义一个Map结果处理器MapResultHandler
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 了。
总结
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。