MyBatis-Plus中实现自定义复杂排序逻辑

在使用MyBatis-Plus进行数据库操作时，有时我们需要根据复杂的条件进行数据排序，特别是当排序条件不仅仅基于单一字段，而是需要多个字段和特定的排序规则时。本文将通过一个实际的例子说明如何在MyBatis-Plus中实现这种复杂的排序逻辑。

场景描述

假设我们有一个车辆保险信息的查询功能，需要根据gateNo进行排序。gateNo字段的格式为X-YYY，其中X可以是数字或特殊字符串（如“天”），YYY是一个数字。我们的目标是按X的数值排序，将X为“天”的情况排在最后，同时YYY也需要按数值进行排序。

解决方案

在MyBatis-Plus中，QueryWrapper提供了丰富的方法来构建查询条件，但对于复杂的排序逻辑，我们需要利用原始SQL片段来实现。以下是具体的实现步骤和代码：

步骤1：定义QueryWrapper

首先，我们创建一个QueryWrapper实例。如果查询模型（model）为空，我们直接返回一个空的QueryWrapper实例。

QueryWrapper<VCzInsuranceEntity> queryWrapper = new QueryWrapper<>();
if (model == null) {
    return queryWrapper;
}

步骤2：添加自定义排序逻辑

为了根据gateNo进行排序，我们使用last方法来添加自定义的SQL排序片段。这里使用CASE语句来处理特殊值“天”，并使用CAST函数和SUBSTRING_INDEX来分离和转换字符串为数值。

String customOrderSql = "CASE WHEN gate_no LIKE '天-%' THEN 2 ELSE 1 END, " +
    "CAST(SUBSTRING_INDEX(gate_no, '-', 1) AS UNSIGNED), " +
    "CAST(SUBSTRING_INDEX(gate_no, '-', -1) AS UNSIGNED)";
queryWrapper.last("ORDER BY " + customOrderSql);

详细解释

1. 特殊值处理：

   • 使用 CASE 语句判断 gate_no 是否以“天-”开头。如果是，则赋值为 2，这样的记录会被排在查询结果的最后。

2. 按 X 排序：

   • 使用 SUBSTRING_INDEX 函数提取 gate_no 中 ‘-’ 前的部分（即 X），然后用 CAST 函数将其转换为无符号整数，确保按数值进行排序。

3. 按 YYY 排序：

   • 同样使用 SUBSTRING_INDEX 函数但这次提取 ‘-’ 后的部分（即 YYY），再通过 CAST 进行数值转换，保证在相同的 X 内部，根据 YYY 的值进行排序。

步骤3：添加其他查询条件

在实际的应用场景中，除了排序之外，我们通常还需要根据多个字段添加过滤条件。例如，我们可以根据车辆ID、所有者ID和其他相关字段来过滤数据。这里为了简化，我们只展示几个示例条件：

queryWrapper
    .eq(model.getCarId() != null && model.getCarId() != 0, "car_id", model.getCarId())
    .like(StringUtils.isNotBlank(model.getLicensePlate()), "license_plate", model.getLicensePlate());

完整方法

将以上步骤组合，我们得到了一个完整的方法，用于构建带有自定义排序的查询包装器：

@Override
public QueryWrapper<VCzInsuranceEntity> onSelectWhere(VCzInsuranceEntity model) {
    QueryWrapper<VCzInsuranceEntity> queryWrapper = new QueryWrapper<>();
    if (model == null) {
        return queryWrapper;
    }

    String customOrderSql = "CASE WHEN gate_no LIKE '天-%' THEN 2 ELSE 1 END, " +
        "CAST(SUBSTRING_INDEX(gate_no, '-', 1) AS UNSIGNED), " +
        "CAST(SUBSTRING_INDEX(gate_no, '-', -1) AS UNSIGNED)";
    queryWrapper.last("ORDER BY " + customOrderSql);

    queryWrapper
        .eq(model.getCarId() != null && model.getCarId() != 0, "car_id", model.getCarId())
        .like(StringUtils.isNotBlank(model.getLicensePlate()), "license_plate", model.getLicensePlate());

    return queryWrapper;
}

结论

通过使用MyBatis-Plus的QueryWrapper和SQL原始片段，我们可以灵活地实现复杂的数据排序逻辑。这种方法尤其适用于需要对数据进行特定规则排序的场景