java - 在mybatis中返回HashMap，在spring MVC中作为ModelAttribute使用

Question

我想使用 spring mvc @modelAttribute 在我的 Jsp 页面中显示类别列表。

在我的 mapper.xml 文件中是

<select id="selectAllCategories" resultMap="BaseResultMap">
  select id, name from categories  
</select>

在我的 Mapper.java 类中我有方法

List<Map<String, String>> selectAllCategories();

我想要一个这样的方法:

Map<Integer, String>`selectAllCategories();

而不是 List<Map<>> ，这可能吗？

Answer 1

您想获得 Map<Integer,String>其中整数是id字符串是 name .如果您的表中有 200 个类别，您会希望 map 中有 200 个条目，而不是 200 个 map 的列表。

MyBatis 不能完全做到开箱即用，但您可以使用它的工具来做到这一点。我看到两个选项。

选项 1:

第一个不是您所要求的，但值得展示。它给你一个Map<Integer,Category>其中类别是类别表的域对象，它具有 id、名称(可能还有类别表中的其他字段)。创建 Category 域对象后，在 MyBatis 中使用 @MapKey 很容易做到这一点。注释:

@Select("SELECT id, name FROM categories")
@MapKey("id")
Map<Integer,Category> getAllCategories();

在你的代码中你会这样做:

MyMapper mapper = session.getMapper(MyMapper.class);
Map<Integer,Category> m = mapper.getAllCategories();

这可能适用于您的用例，也可能不适用于您的用例，具体取决于您是否可以将名称提取为 Category 对象的属性。

选项 2:

获取 Map<Integer,String>你问的，我知道的最简单的方法是创建一个实现 MyBatis ResultHandler 的类界面。

您的 ResultHandler 将使用 MyBatis 创建的 column-name => column-value 的默认 hashmap 并创建单个主 Map。代码如下:

public class CategoryResultHandler implements ResultHandler {

  Map<Integer,String> inMap = new HashMap<Integer,String>(); 

  public Map<Integer, String> getIdNameMap() {
    return inMap;
  }

  @Override
  public void handleResult(ResultContext rc) {
    @SuppressWarnings("unchecked")
    Map<String,Object> m = (Map<String,Object>)rc.getResultObject();
    inMap.put((Integer)getFromMap(m, "id"), 
              (String)getFromMap(m, "name"));
  }

  // see note at bottom of answer as to why I include this method
  private Object getFromMap(Map<String, Object> map, String key) {
    if (map.containsKey(key.toLowerCase())) {
      return map.get(key.toLowerCase());
    } else {
      return map.get(key.toUpperCase());
    }
  }
}

handleResult 方法在类别表中每行调用一次。你告诉 MyBatis 使用 ResultHandler，然后像这样提取你的主 map :

CategoryResultHandler rh = new CategoryResultHandler();
session.select("getAllCategories", rh);
Map<Integer,String> m = rh.getIdNameMap();

这两个中的一个应该适合你。

最后几点说明:

1. 为什么我要包含 getFromMap()辅助方法？因为你不能总是控制 MyBatis 返回的 hashmap 中列名的大小写。更多详情:mybatis- 3.1.1. how to override the resultmap returned from mybatis

2. 我在 mybatis-koans 的 Koan26 中有这些解决方案的工作示例(我根据您的问题添加):https://github.com/midpeter444/mybatis-koans