【动态SQL在业务中的应用——根据传参选择第二优先级排序】

左手的ㄟ右手 2024-03-16 20:54 15阅读 0赞

### 业务需求： ###

某某查询列表中，有三种排序规则，默认排序规则：

第一优先级：按部门/公司的顺序排序

第二优先级：按照机房等级，A＞B＞C，第二优先级可以改为机柜数量，点击可切换升序、降序

列表展示字段如图：

![f1313ec1e037456aba0cb5c765c90717.png][]

### 业务分析： ###

这是一个条件查询接口，我根据传参使用MybatisPlus里面的

LambdaQueryWrapper 构建查询条件，使用方法不在这里讲解了，优点是可以简化开发，给你封装好了sql语句，不用手写。以下是传参对象：

@Data
    @EqualsAndHashCode(callSuper = true)
    @SuperBuilder
    @NoArgsConstructor
    @AllArgsConstructor
    @ApiModel(value = "数字中心机房查询条件")
    public class MachineRoomQueryDTO extends PageDTO {
    
     //省略其他入参...
    
    
     @ApiModelProperty(value = "是否按机柜数量排序",dataType = "boolean", example = "false")
     @NotNull(message = "是否按机柜数量排序必填")
     private Boolean orderByMachineCount;
    
     @ApiModelProperty(value = "是否按机柜数量倒序排序",dataType = "boolean", example = "false")
     @NotNull(message = "是否按机柜数量倒序排序必填")
     private Boolean orderByDesc;
    }

组装好了查询条件，我先查出了数据表中所有符合条件的记录，然后根据入参选择不同的第二优先级排序规则。

//业务代码省略...
    
     LambdaQueryWrapper<DigitalMachineRoom> queryWrapper = new LambdaQueryWrapper<>();
    
    
    //省略组装查询条件...
    
    
    //先查出全部再排序
     List<DigitalMachineRoom> digitalMachineRooms = digitalMachineRoomDAO.selectList(queryWrapper);
    
     if (queryDTO.getOrderByMachineCount()) {
           if (queryDTO.getOrderByDesc()) {
               //第二优先级为机柜数量 降序
                 digitalMachineRooms = digitalMachineRoomDAO.selectListBySortRule(digitalMachineRooms, true);
             } else {
              //第二优先级为机柜数量 升序
                 digitalMachineRooms = digitalMachineRoomDAO.selectListBySortRule(digitalMachineRooms, false);
                }
            } else {
             //第二优先级为机房等级 升序
                digitalMachineRooms = digitalMachineRoomDAO.selectListBySortRule(digitalMachineRooms, null);
            }
    
     return digitalMachineRooms;

以下是Dao层抽取的方法：

@Repository
    public interface DigitalMachineRoomDAO extends MyBatisBaseDao<DigitalMachineRoom, Integer> {
        /**
         * 根据入参选择第二优先级排序规则查询机房列表
         * 第二优先级排序规则:
         * 1.orderByCountDesc=true——第二优先级按照机柜数量倒序排序
         * 2.orderByCountDesc=false——第二优先级按照机柜数量正序排序
         * 3.orderByCountDesc=null——第二优先级按照机房等级正序排序
         */
        List<DigitalMachineRoom> selectListBySortRule(@Param("list") List<DigitalMachineRoom> list, @Param("orderByCountDesc")Boolean orderByCountDesc);
    }

以下是上面方法对应的xml语句：

<select id="selectListBySortRule" resultType="com.cncico.progress.domain.DigitalMachineRoom">
        select dmr.*
        from digital_machine_room dmr
        left join org_config oc on oc.org_id = dmr.org_id
        <where>
          <if test="list != null and list.size() > 0">
            and dmr.id in (<foreach collection="list" item="item" index="index" separator=",">#{item.id}</foreach>)
          </if>
        </where>
        <choose>
          <when test="orderByCountDesc==true">
            order by oc.sort DESC, dmr.machine_count DESC
          </when>
          <when test="orderByCountDesc==false">
            order by oc.sort DESC, dmr.machine_count ASC
          </when>
          <otherwise>
            order by oc.sort DESC, dmr.`level` ASC
          </otherwise>
        </choose>
      </select>

注意点：

最开始用的queryWrapper，发现用了in条件查询后，会使后面的queryWrapper条件失效，解决方案是把eq或ne的条件写在范围查询in的前面，可避免条件失效，一定注意奥！

小节：

开始为了三种第二优先级排序规则写了三份sql方法，发现代码相似部分太多，被小组长点名改进，后面我想根据传参选择不同字段拼接sql的方案，但是要用到$符（有SQL注入风险），最终在小组长建议下改成了现在的样子，以上就是动态SQL的分享。

[f1313ec1e037456aba0cb5c765c90717.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/15/f8ede519653a495bbb6a861906e12113.png