MyBatis Map结果的Key转为驼峰式

发布时间：2020-09-27 11:00:56 作者：isea533
来源：脚本之家阅读：217

MyBatis 配置文件中，支持下面这几种配置：

properties, settings, typeAliases, typeHandlers,
objectFactory, objectWrapperFactory, reflectorFactory,
plugins, environments, databaseIdProvider, mappers

我们使用objectWrapperFactory来解决这个问题。

配置这个属性时，必须遵守上面属性配置的顺序进行配置（在objectFactory后面，在reflectorFactory前面），否则就会出错。

objectWrapperFactory 接口如下：

public interface ObjectWrapperFactory {
 boolean hasWrapperFor(Object object);
 ObjectWrapper getWrapperFor(MetaObject metaObject, Object object);
}

通过实现这个接口，可以判断当object是Map类型时，返回true，然后在下面的getWrapperFor中返回一个可以处理 key 为驼峰的Wrapper 类即可。

首先，参考已有的MapWrapper实现自己的MyWrapper：

package tk.mybatis;
import org.apache.ibatis.reflection.MetaObject;
import org.apache.ibatis.reflection.wrapper.MapWrapper;
import java.util.Map;
/**
 * 驼峰处理
 */
public class MyMapWrapper extends MapWrapper {
  public MyMapWrapper(MetaObject metaObject, Map<String, Object> map) {
    super(metaObject, map);
  }
  @Override
  public String findProperty(String name, boolean useCamelCaseMapping) {
    if (useCamelCaseMapping
        && ((name.charAt(0) >= 'A' && name.charAt(0) <= 'Z')
           || name.indexOf("_") >= 0)) {
      return underlineToCamelhump(name);
    }
    return name;
  }
  /**
   * 将下划线风格替换为驼峰风格
   * @param inputString
   * @return
   */
  public String underlineToCamelhump(String inputString) {
    StringBuilder sb = new StringBuilder();
    boolean nextUpperCase = false;
    for (int i = 0; i < inputString.length(); i++) {
      char c = inputString.charAt(i);
      if (c == '_') {
        if (sb.length() > 0) {
          nextUpperCase = true;
        }
      } else {
        if (nextUpperCase) {
          sb.append(Character.toUpperCase(c));
          nextUpperCase = false;
        } else {
          sb.append(Character.toLowerCase(c));
        }
      }
    }
    return sb.toString();
  }
}

这种方式实现起来很容易。然后提供ObjectWrapperFactory实现类：

package tk.mybatis;
import org.apache.ibatis.reflection.MetaObject;
import org.apache.ibatis.reflection.wrapper.ObjectWrapper;
import org.apache.ibatis.reflection.wrapper.ObjectWrapperFactory;
import java.util.Map;
/**
 * Map 类型结果转驼峰
 */
public class MapWrapperFactory implements ObjectWrapperFactory {
  @Override
  public boolean hasWrapperFor(Object object) {
    return object != null && object instanceof Map;
  }
  @Override
  public ObjectWrapper getWrapperFor(MetaObject metaObject, Object object) {
    return new MyMapWrapper(metaObject, (Map) object);
  }
}

仍然很简单，在 MyBatis 配置文件中配置上objectWrapperFactory：

<objectWrapperFactory type="tk.mybatis.MapWrapperFactory"/>

因为代码中判断了useCamelCaseMapping，所以想要真正能够使用这个功能，还需要在settings中增加如下配置：

<settings>
  <setting name="mapUnderscoreToCamelCase" value="true"/>
  <!-- 其他配置 -->
</settings>

这种方式不需要使用拦截器，而且直接从源头对Map 的key 进行处理，不会产生任何额外的消耗，如果你经常设置返回值resultType为map，可以尝试该方法。

总结

以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，谢谢大家对亿速云的支持。如果你想了解更多相关内容请查看下面相关链接

MyBatis Map结果的Key转为驼峰式

相关阅读