MyBatis-plus 自动生成 条件构造器 QueryWrapper

背景

在使用Wrapper构建条件时，经常因为需要构建的条件过多需要写半个多小时，还容易粗心写错字段，所以就想搞个可以直接自动构建QueryWrapper的工具类。

思路

我们需要一个方法，能够根据条件参数对象的属性值来设置QueryWrapper的条件，同时需要返回一个我们所需要对应范型的QueryWrapper。

使用范型来确定返回的QueryWrapper的范型，使用反射来获取参数类中的属性值，同时需要区分字段的类型来确定选择该字段使用哪种查询方式。

实现

再获取字段名的时候有可能参数类的字段名不能直接根数据库表的字段名直接对应上，所以我直接复用了@TableField 注解，若是字段上有该注解就直接使用其对应的value属性值，若是没有该注解则就使用字段名，将字段名的驼峰命名法转换成下划线分割的格式。

    /**
     * 创建对应的wrapper
     * @param param 参数对象
     * @param <T> 范型
     * @return 创建好的wrapper
     */
    public static <T> QueryWrapper<T> createWrapper(Object param) {
        QueryWrapper<T> wrapper = new QueryWrapper<>();
        Class<?> dataClass = param.getClass();
        try {
            for (Field field : dataClass.getDeclaredFields()) {
                field.setAccessible(true);
                //获取字段类型和字段值
                Class<?> type = field.getType();
                Object value = field.get(param);
                if (value == null){
                    continue;
                }
                //将字段名转为数据库中的字段名
                String fieldName;
                TableField tableFieldAnnotation = field.getAnnotation(TableField.class);
                if (tableFieldAnnotation != null) {
                    fieldName = tableFieldAnnotation.value();
                }else{
                    fieldName = convertToSnakeCase(field.getName());
                }
                //根据字段的类型来选择不同的查询方式
                if (type == String.class) {
                    wrapper.like(fieldName, value);
                }else if (type == Long.class){
                    wrapper.eq(fieldName, value);
                }else if (type == List.class){
                    wrapper.in(fieldName, (List<?>)value);
                }else if (type == Between.class){
                    Between val= (Between)value;
                    wrapper.between(fieldName, val.getStart(), val.getEnd());
                }else{
                    wrapper.eq(fieldName, value);
                }
            }
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        }
        return wrapper;
    }


    /**
     * 将驼峰命名法的字段名转为下划线隔开的形式
     * @param input 驼峰命名法的字段名
     * @return 下划线隔开的数据库字段名
     */
    private static String convertToSnakeCase(String input) {
        StringBuilder output = new StringBuilder();
        for (int i = 0; i < input.length(); i++) {
            char ch = input.charAt(i);
            if (Character.isUpperCase(ch)) {
                if (i > 0) {
                    output.append(StrPool.C_UNDERLINE);
                }
                output.append(Character.toLowerCase(ch));
            } else {
                output.append(ch);
            }
        }
        return output.toString();
    }

测试

可以看到dto参数对象的属性值都已经填充到了QueryWrapper中