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

背景

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

我们需要一个方法，可以大概根据条件参数对象的属性值来设置QueryWrapper的条件，同时需要返回一个我们所需要对应范型的QueryWrapper。
利用范型来确定返回的QueryWrapper的范型，利用反射来获取参数类中的属性值，同时需要区分字段的类型来确定选择该字段利用哪种查询方式。
实现

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

1.     /**
2.      * 创建对应的wrapper
3.      * @param param 参数对象
4.      * @param <T> 范型
5.      * @return 创建好的wrapper
6.      */
7.     public static <T> QueryWrapper<T> createWrapper(Object param) {
8.         QueryWrapper<T> wrapper = new QueryWrapper<>();
9.         Class<?> dataClass = param.getClass();
10.         try {
11.             for (Field field : dataClass.getDeclaredFields()) {
12.                 field.setAccessible(true);
13.                 //获取字段类型和字段值
14.                 Class<?> type = field.getType();
15.                 Object value = field.get(param);
16.                 if (value == null){
17.                     continue;
18.                 }
19.                 //将字段名转为数据库中的字段名
20.                 String fieldName;
21.                 TableField tableFieldAnnotation = field.getAnnotation(TableField.class);
22.                 if (tableFieldAnnotation != null) {
23.                     fieldName = tableFieldAnnotation.value();
24.                 }else{
25.                     fieldName = convertToSnakeCase(field.getName());
26.                 }
27.                 //根据字段的类型来选择不同的查询方式
28.                 if (type == String.class) {
29.                     wrapper.like(fieldName, value);
30.                 }else if (type == Long.class){
31.                     wrapper.eq(fieldName, value);
32.                 }else if (type == List.class){
33.                     wrapper.in(fieldName, (List<?>)value);
34.                 }else if (type == Between.class){
35.                     Between val= (Between)value;
36.                     wrapper.between(fieldName, val.getStart(), val.getEnd());
37.                 }else{
38.                     wrapper.eq(fieldName, value);
39.                 }
40.             }
41.         } catch (IllegalAccessException e) {
42.             e.printStackTrace();
43.         }
44.         return wrapper;
45.     }
48.     /**
49.      * 将驼峰命名法的字段名转为下划线隔开的形式
50.      * @param input 驼峰命名法的字段名
51.      * @return 下划线隔开的数据库字段名
52.      */
53.     private static String convertToSnakeCase(String input) {
54.         StringBuilder output = new StringBuilder();
55.         for (int i = 0; i < input.length(); i++) {
56.             char ch = input.charAt(i);
57.             if (Character.isUpperCase(ch)) {
58.                 if (i > 0) {
59.                     output.append(StrPool.C_UNDERLINE);
60.                 }
61.                 output.append(Character.toLowerCase(ch));
62.             } else {
63.                 output.append(ch);
64.             }
65.         }
66.         return output.toString();
67.     }

复制代码

测试

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

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。