snowy/snowy-common/src/main/java/vip/xiaonuo/common/util/QueryWrapperHelp.java

package vip.xiaonuo.common.util;

import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import lombok.extern.slf4j.Slf4j;
import org.springframework.util.ObjectUtils;
import vip.xiaonuo.common.annotation.ListQueryMethod;
import vip.xiaonuo.common.enums.ListQueryType;

import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;

@Slf4j
public class QueryWrapperHelp {

    public static void createQueryCondition(QueryWrapper queryWrapper, Object obj) {
        try {
            Method[] methods = obj.getClass().getMethods();
            for (Method method : methods) {
                String name = method.getName();
                if (!name.startsWith("get")) {
                    continue;
                }
                Object result = method.invoke(obj, null);
                if (result == null) {
                    continue;
                }
                name = name.replace("get", "");
                String fildName = name.substring(0, 1).toLowerCase() + name.substring(1, name.length());

                Field field = QueryWrapperHelp.getField(obj.getClass(), fildName);
                if (field == null) {
                    continue;
                }
                ListQueryMethod listQueryMethod = field.getAnnotation(ListQueryMethod.class);
                if (listQueryMethod == null) {
                    continue;
                }
                String colName = StrUtil.toUnderlineCase(name);
                QueryWrapperHelp.setQueryWrapper(queryWrapper, listQueryMethod, colName, result);
            }
        } catch (IllegalAccessException | InvocationTargetException | NoSuchMethodException exception) {
            log.error(exception.getMessage(), exception);
        }
    }

    /**
     * 获得字段
     *
     * @param cls
     * @param fildName
     * @return
     */
    private static Field getField(Class cls, String fildName) {
        try {
            return cls.getDeclaredField(fildName);
        } catch (NoSuchFieldException e) {
        }
        if(cls.getSuperclass()==null){
            return null;
        }
        return QueryWrapperHelp.getField(cls.getSuperclass(),fildName);
    }

    private static void setQueryWrapper(QueryWrapper queryWrapper, ListQueryMethod listQueryMethod, String colName, Object value) throws NoSuchMethodException, InvocationTargetException, IllegalAccessException {
        Method method = queryWrapper.getClass().getMethod(listQueryMethod.value().getCode(), Object.class, Object.class);
        method.invoke(queryWrapper, colName, value);
    }
}