一尘不染

在mybatis中返回HashMap并将其用作Spring MVC中的ModelAttribute

spring-mvc

我想使用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<>>,这可能吗?


阅读 533

收藏
2020-06-01

共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();

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

2020-06-01