脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 软件编程 → java → Java MyBatis返回两个字段作为Map的key和value

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

 更新时间：2024年12月14日 10:08:32   作者：天航星  
使用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 了。

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

最新评论