1. 程式人生 > >Mybatis 結果對映下劃線轉駝峰

Mybatis 結果對映下劃線轉駝峰

mybatis 結果對映下劃線轉駝峰

Spring Boot 配置:

#下劃線轉駝峰
mybatis.configuration.map-underscore-to-camel-case=true

mybatis XML 配置

<configuration>
    <!-- 全域性配置 -->
    <settings>
       <!-- 是否開啟自動駝峰命名規則(camel case)對映,即從資料庫列名 A_COLUMN 到屬性名 aColumn 的類似對映 -->
        <setting name="mapUnderscoreToCamelCase" value="true"/>
    </settings>
</configuration>

當resultTpye=map時,以上的配置無效


通過實現ObjectWrapperFactory介面,可以判斷當object 是 Map 型別時,返回 true,然後在下面的 getWrapperFor 中返回一個可以處理 key 為駝峰的 Wrapper 類即可。

實現ObjectWrapperFactory

Spring boot 配置

@Configuration
public class MybatisConfig {
    /**
     * mybatis resultType為map時下劃線鍵值轉小寫駝峰形式插
     */
    @Bean
    public ConfigurationCustomizer configurationCustomizer() {
        return configuration -> configuration.setObjectWrapperFactory(new MapWrapperFactory());
    }

    
    static 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);
        }
    }

    static class MyMapWrapper extends MapWrapper {
        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.contains("_"))) {
                return underlineToCamelhump(name);
            }
            return name;
        }

        /**
         * 將下劃線風格替換為駝峰風格
         *
         * @param inputString
         * @return
         */
        private 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();
        }
    }
}

還需要配置

#下劃線轉駝峰
mybatis.configuration.map-underscore-to-camel-case=true

mybatis XML配置

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

還需要

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