在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>Integer是id和String是the的地方name。如果表中有200个类别，则您需要200个条目，而不是200个列表。

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的默认哈希图，并创建单个主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，然后像这样提取您的主地图：

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

这两个之一应该为您工作。