实体对象转QueryWrapper 便捷实现条件查询

mybatis plus中提供了QueryWrapper做条件查询,

方法定义是public Children eq(boolean condition,R column,java.lang.Object val)

参数:condition - 执行条件;column - 字段;val - 值

我们通常是这样写的,

 1    public IhTimeSchedule  getTimeSchedule(String serviceCode,String appointDate, String noonType,
 2                                            String appointTimeInterval, String deptCode, String doctorCode,
 3                                            String clinicTypeCode,String serviceType) {
 4         QueryWrapper<IhTimeSchedule> wrapper=new QueryWrapper<IhTimeSchedule>()
 5                 .eq("DEPT_CODE",deptCode)
 6                 .eq("SERVICE_CODE",serviceCode)
 7                 .eq(StringUtils.isNotBlank(doctorCode),"DOCTOR_CODE",doctorCode)
 8                 .eq("CLINIC_DATE",appointDate+" 00:00:00")
 9                 .eq(StringUtils.isNotBlank(noonType),"NOON_TYPE",noonType).
10                 eq(StringUtils.isNotBlank(appointTimeInterval),"TIME_DIVISION",appointTimeInterval).
11                 eq("CLINIC_TYPE_CODE",clinicTypeCode);
12         IhTimeSchedule schedule = scheduleMapper.selectOne(wrapper);
13         return  schedule;
14     }

也可以这样写(省去写列名的麻烦)

 1 public List<HpSystemConfig> queryConfig(HpSystemConfig config) {
 2         if(config==null){
 3             throw new RRException(ResultCode.PARAM_IS_BLANK);
 4         }
 5         LambdaQueryWrapper<HpSystemConfig> wrapper=new LambdaQueryWrapper<>();
 6         wrapper.eq(StringUtils.isNotBlank(config.getConfigName()),HpSystemConfig::getConfigName,config.getConfigName()).
 7                 eq(StringUtils.isNotBlank(config.getConfigValue()),HpSystemConfig::getConfigValue,config.getConfigValue())
 8                 .eq(StringUtils.isNotBlank(config.getDeptCode()),HpSystemConfig::getDeptCode,config.getDeptCode())
 9                 .eq(StringUtils.isNotBlank(config.getUnitsCode()),HpSystemConfig::getUnitsCode,config.getUnitsCode())
10                 .eq(config.getId()!=null,HpSystemConfig::getId,config.getId());
11         return this.baseMapper.selectList(wrapper);
12     }

 

不难看出,当实体类中很多属性都需要作为查询条件的时候,QueryWrapper就需要定义的很长,也比较麻烦。

所以我就想,不如利用反射,写一个工具类,根据实体类自动生成QueryWrapper。

思路:

1.传入实体类对象作为参数,获取这个对象的Class对象。

2.遍历实体类声明的每个属性

3.获取属性上的mybatis注解TableField或者TableId,注解的value值即是数据库表的列名

4.构造QueryWrapper的eq条件(get方法返回值类型不同,则执行条件的判断有不同)

具体实现如下

 1 public static QueryWrapper entity2Wrapper(Object obj) {
 2         Class<?> aClass = obj.getClass();
 3         Field[] fields = obj.getClass().getDeclaredFields();
 4         QueryWrapper wrapper = new QueryWrapper();
 5         //遍历属性
 6         for (Field field : fields) {
 7             Method method = null;
 8             try {
 9                 String fieldName = field.getName();
10                 //跳过serialVersionUID
11                 if (fieldName.equals("serialVersionUID")) {
12                     continue;
13                 }
14                 //获取属性上的注解
15                 TableField fieldAnnotation = field.getAnnotation(TableField.class);
16                 TableId idAnnotation = field.getAnnotation(TableId.class);
17                 //拿到列名
18                 String value = fieldAnnotation == null ? idAnnotation.value() : fieldAnnotation.value();
19                 //get方法
20                 method = aClass.getDeclaredMethod("get" + captureName(fieldName), null);
21                 Object returnValue = method.invoke(obj);
22                 if (returnValue instanceof String) {
23                     String str = (String) returnValue;
24                     wrapper.eq(StringUtils.isNotBlank(str), value, returnValue);
25                 } else {
26                     wrapper.eq(returnValue != null, value, returnValue);
27                 }
28 
29             } catch (Exception e) {
30                 e.printStackTrace();
31                 return null;
32             }
33         }
34         return wrapper;
35     }
 1   /**
 2      * 将字符串的首字母转大写
 3      *
 4      * @param str 需要转换的字符串
 5      * @return
 6      */
 7     private static String captureName(String str) {
 8         // 进行字母的ascii编码前移,效率要高于截取字符串进行转换的操作
 9         char[] cs = str.toCharArray();
10         cs[0] -= 32;
11         return String.valueOf(cs);
12     }

使用方式如下

public List<HpQueueCallRoom> queryRoom(HpQueueCallRoom room) {
   if(room==null){
throw new RRException(ResultCode.PARAM_IS_BLANK);
}
QueryWrapper queryWrapper = WrapperUtil.entity2Wrapper(room);
return this.baseMapper.selectList(queryWrapper);
}


是不是简单了很多,通过工具类得到了QueryWrapper对象后,还可以链式附加别的需要的条件,比如select,orderBy.

 

swagger2测试

如下图,符合预期。

实体对象转QueryWrapper 便捷实现条件查询

 

 

 实体对象转QueryWrapper 便捷实现条件查询

上一篇:Cron表达式以及定时任务配置


下一篇:mybatis-plus学习笔记