實體對象轉QueryWrapper 便捷實現條件查詢

mybatis plus中提供了QueryWrapper做條件查詢，

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

參數:condition - 執行條件；column - 字段；val - 值

我們通常是這樣寫的，

   public IhTimeSchedule  getTimeSchedule(String serviceCode,String appointDate, String noonType,
                                           String appointTimeInterval, String deptCode, String doctorCode,
                                           String clinicTypeCode,String serviceType) {
        QueryWrapper<IhTimeSchedule> wrapper=new QueryWrapper<IhTimeSchedule>()
                .eq("DEPT_CODE",deptCode)
                .eq("SERVICE_CODE",serviceCode)
                .eq(StringUtils.isNotBlank(doctorCode),"DOCTOR_CODE",doctorCode)
                .eq("CLINIC_DATE",appointDate+" 00:00:00")
                .eq(StringUtils.isNotBlank(noonType),"NOON_TYPE",noonType).
                eq(StringUtils.isNotBlank(appointTimeInterval),"TIME_DIVISION",appointTimeInterval).
                eq("CLINIC_TYPE_CODE",clinicTypeCode);
        IhTimeSchedule schedule = scheduleMapper.selectOne(wrapper);
        return  schedule;
    }

也可以這樣寫（省去寫列名的麻煩）

public List<HpSystemConfig> queryConfig(HpSystemConfig config) {
        if(config==null){
            throw new RRException(ResultCode.PARAM_IS_BLANK);
        }
        LambdaQueryWrapper<HpSystemConfig> wrapper=new LambdaQueryWrapper<>();
        wrapper.eq(StringUtils.isNotBlank(config.getConfigName()),HpSystemConfig::getConfigName,config.getConfigName()).
                eq(StringUtils.isNotBlank(config.getConfigValue()),HpSystemConfig::getConfigValue,config.getConfigValue())
                .eq(StringUtils.isNotBlank(config.getDeptCode()),HpSystemConfig::getDeptCode,config.getDeptCode())
                .eq(StringUtils.isNotBlank(config.getUnitsCode()),HpSystemConfig::getUnitsCode,config.getUnitsCode())
                .eq(config.getId()!=null,HpSystemConfig::getId,config.getId());
        return this.baseMapper.selectList(wrapper);
    }

 

@Data
@TableName("HP_SCHEDULING_RECORD")
@ApiModel
public class HpSchedulingRecord implements Serializable {

    private static final long serialVersionUID = 1L;

    @TableId(value = "ID", type = IdType.ID_WORKER)
    private Long id;

    @ApiModelProperty(value = "醫院編號")
    @TableField("HSR_HOSPOTALCODE")
    private String hsrHospotalcode;

    @ApiModelProperty(value = "醫生編號")
    @TableField("HSR_DOCTOR")
    private Integer hsrDoctor;

    @ApiModelProperty(value = "房間編號")
    @TableField("HSR_ROOM")
    private Integer hsrRoom;

    @ApiModelProperty(value = "科室編號")
    @TableField("HSR_DEPT")
    private Integer hsrDept;

    @ApiModelProperty(value = "排班日期 形如yyyy-MM-dd")
    @TableField("HS_SHIFT_DATE")
    private String hsShiftDate;


}

 

不難看出，當實體類中很多屬性都需要作為查詢條件的時候，QueryWrapper就需要定義的很長，也比較麻煩。

可以看到上面我們的實體類，每個屬性都用mybatis注解標明了數據庫表的列名，所以我就想，不如利用反射，寫一個工具類，根據實體類自動生成QueryWrapper。

思路：

1.傳入實體類對象作為參數，獲取這個對象的Class對象。

2.遍歷實體類聲明的每個屬性

3.獲取屬性上的mybatis注解TableField或者TableId,注解的value值即是數據庫表的列名

4.構造QueryWrapper的eq條件（get方法返回值類型不同，則執行條件的判斷有不同）

具體實現如下

public static QueryWrapper entity2Wrapper(Object obj) {
        Class<?> aClass = obj.getClass();
        Field[] fields = obj.getClass().getDeclaredFields();
        QueryWrapper wrapper = new QueryWrapper();
        //遍歷屬性
        for (Field field : fields) {
            Method method = null;
            try {
                String fieldName = field.getName();
                //跳過serialVersionUID
                if (fieldName.equals("serialVersionUID")) {
                    continue;
                }
                //獲取屬性上的注解
                TableField fieldAnnotation = field.getAnnotation(TableField.class);
                TableId idAnnotation = field.getAnnotation(TableId.class);
                //拿到列名
                String value = fieldAnnotation == null ? idAnnotation.value() : fieldAnnotation.value();
                //get方法
                method = aClass.getDeclaredMethod("get" + captureName(fieldName), null);
                Object returnValue = method.invoke(obj);
                if (returnValue instanceof String) {
                    String str = (String) returnValue;
                    wrapper.eq(StringUtils.isNotBlank(str), value, returnValue);
                } else {
                    wrapper.eq(returnValue != null, value, returnValue);
                }

            } catch (Exception e) {
                e.printStackTrace();
                return null;
            }
        }
        return wrapper;
    }

  /**
     * 將字符串的首字母轉大寫
     *
     * @param str 需要轉換的字符串
     * @return
     */
    private static String captureName(String str) {
        // 進行字母的ascii編碼前移，效率要高於截取字符串進行轉換的操作
        char[] cs = str.toCharArray();
        cs[0] -= 32;
        return String.valueOf(cs);
    }

使用方式如下

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測試

如下圖，符合預期。