Question

我在Java Web应用程序中使用Spring Boot和Mybatis（接口和mapper.xml）。

我的目的是将包含两列“名称”和“计数”的结果列表转换为地图。在该映射中，第一列应用作键，第二列应作为值。

我知道我应该重写ResultHandler，而如何通过使用接口才能生效？

Answer 1

已更新。我通过添加一个拦截器来处理这种情况并在mybatis-config.xml中添加插件以启用它来实现此方法。

首先，创建一个注释，然后可以通过添加注释，覆盖注释并将第一列设置为键，第二列设置为值来处理这种情况。

@Documented
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.METHOD, ElementType.TYPE})
@Inherited
public @interface ResultMap {

    String value();
}

第二，您应该实现一个拦截器来处理具有@ResultMap批注的mybatis resultSet

public class ResultMapInterceptor implements Interceptor {

    private static final Logger LOGGER = LoggerFactory.getLogger(ResultMapInterceptor.class);

    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        Object target = invocation.getTarget();
        DefaultResultSetHandler defaultResultSetHandler = (DefaultResultSetHandler) target;
        Field field = ReflectionUtils.findField(DefaultResultSetHandler.class, "mappedStatement");
        ReflectionUtils.makeAccessible(field);
        MappedStatement mappedStatement = (MappedStatement) field.get(defaultResultSetHandler);
        String className = StringUtils.substringBeforeLast(mappedStatement.getId(), ".");
        String methodName = StringUtils.substringAfterLast(mappedStatement.getId(), ".");

        Method[] methods = Class.forName(className).getDeclaredMethods();
        if (methods == null) {
            return invocation.proceed();
        }

        // get method "mappedStatement"
        for (Method method : methods) {
            if (methodName.equalsIgnoreCase(method.getName())) {
                // get annotation ResultMap, if null, then proceed
                ResultMap resultMap = method.getAnnotation(ResultMap.class);
                if (resultMap == null) {
                    return invocation.proceed();
                }
                // convert result map to specify result
                Statement statement = (Statement) invocation.getArgs()[0];
                return convert2Map(statement);
            }
        }
        return invocation.proceed();
    }

    @Override
    public Object plugin(Object target) {
        if (target instanceof ResultSetHandler) {
            return Plugin.wrap(target, this);
        } else {
            return target;
        }
    }

    @Override
    public void setProperties(Properties properties) {

    }

    private Object convert2Map(Statement statement) throws Throwable{
        ResultSet resultSet = statement.getResultSet();
        if (resultSet == null) {
            return null;
        }
        List<Object> resultList = new ArrayList(4);
        Map<Object, Object> map = new HashMap(8);
        while (resultSet.next()) {
            if (resultSet.getObject(1) != null) {
                map.put(resultSet.getObject(1), resultSet.getObject(2));
            }
        }
        resultList.add(map);
        LOGGER.debug("map result from DB, {}", resultList);
        return resultList;
    }

然后通过在mybatis-config.xml中添加插件来启用此拦截器

</configuration>
   <plugins>
        <plugin interceptor="com.cisco.migration.handler.ResultMapInterceptor"/>
    </plugins>
</configuration>

最后，在mybatis dao界面中使用此注释。

@ResultMap("")
Map getCountByAccType(SearchFilter searchFilter);

mapper.xml

<select id="getCountByAccType" resultMap="resultMap">
        select
            id，
            count(ACCOUNTTYPE) count
        from table t
</select>

Answer 2

最简单的方法是使用默认接口方法进行转换：

interfact MyMapper {

  @Select("SELECT name, count FROM whatever")
  List<Map<String, Object>> findCountsList();

  default Map<String, Integer> findCounts() {
      Map<String, Integer> result = new HashMap<>;
      for(Map<String, Object> entry:findCountsList()) {
          result.put(entry.get("name"), ((Number)entry.get("count")).intValue());
      }
      return result;
  }

}

如何使用mybatis接口而不是sqlSession处理地图结果（两列结果）

2 个答案: