top.lingkang.mm.override.MagicReflector Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mybatis-magic Show documentation
mybatis能力扩展框架，兼顾mybatis的mapper.xml编写操作数据库。
The newest version!
package top.lingkang.mm.override;

import cn.hutool.core.util.StrUtil;
import org.apache.ibatis.reflection.Reflector;
import org.apache.ibatis.reflection.invoker.Invoker;
import top.lingkang.mm.error.MagicException;

import java.lang.reflect.Field;
import java.util.HashSet;
import java.util.Locale;
import java.util.Map;

/**
 * 增加驼峰转化 get 、set
 *
 * @author lingkang
 * Created by 2024/3/3
 */
public class MagicReflector extends Reflector {
    private static Field getMethodsField, getTypesField, setMethodsField, setTypesField, mapField;

    static {
        try {
            getMethodsField = Reflector.class.getDeclaredField("getMethods");
            getMethodsField.setAccessible(true);

            getTypesField = Reflector.class.getDeclaredField("getTypes");
            getTypesField.setAccessible(true);

            setMethodsField = Reflector.class.getDeclaredField("setMethods");
            setMethodsField.setAccessible(true);

            setTypesField = Reflector.class.getDeclaredField("setTypes");
            setTypesField.setAccessible(true);

            mapField = Reflector.class.getDeclaredField("caseInsensitivePropertyMap");
            mapField.setAccessible(true);
        } catch (Exception e) {
            throw new MagicException(e);
        }
    }

    public MagicReflector(Class clazz) {
        super(clazz);
        // 驼峰转化
        try {
            // get
            Map getMethods = (Map) getMethodsField.get(this);
            Map> getTypes = (Map>) getTypesField.get(this);

            Object[] array = getMethods.keySet().toArray();
            for (Object key : array) {
                String name = StrUtil.toUnderlineCase((String) key);
                if (!getMethods.containsKey(name)) {
                    getMethods.put(name, getMethods.get(key));
                    getTypes.put(name, getTypes.get(key));
                }
            }
        } catch (Exception e) {
            throw new MagicException(e);
        }

        try {
            // set
            Map setMethodsMap = (Map) setMethodsField.get(this);
            Map> setTypesMap = (Map>) setTypesField.get(this);

            Object[] array = setMethodsMap.keySet().toArray();
            for (Object key : array) {
                String name = StrUtil.toUnderlineCase((String) key);
                if (!setMethodsMap.containsKey(name)) {
                    setMethodsMap.put(name, setMethodsMap.get(key));
                    setTypesMap.put(name, setTypesMap.get(key));
                }
            }
        } catch (Exception e) {
            throw new MagicException(e);
        }

        try {
            // 下划线的大写转换
            Map map = (Map) mapField.get(this);
            HashSet set = new HashSet<>(map.keySet());

            for (Object key : set) {
                String value = map.get(key).toUpperCase(Locale.ENGLISH).replace("_", "");
                if (!map.containsKey(value)) {
                    map.put(value, map.get(key));
                }
            }
        } catch (Exception e) {
            throw new MagicException(e);
        }
    }
}