MyBatis-Plus中實現自定義復雜排序邏輯的詳細步驟

更新時間：2024年07月23日 15:32:38 作者：一勺菠蘿丶

這篇文章主要介紹了MyBatis-Plus中實現自定義復雜排序邏輯,通過使用MyBatis-Plus的QueryWrapper和SQL原始片段,我們可以靈活地實現復雜的數據排序邏輯,這種方法尤其適用于需要對數據進行特定規(guī)則排序的場景,需要的朋友可以參考下

在使用MyBatis-Plus進行數據庫操作時，有時我們需要根據復雜的條件進行數據排序，特別是當排序條件不僅僅基于單一字段，而是需要多個字段和特定的排序規(guī)則時。本文將通過一個實際的例子說明如何在MyBatis-Plus中實現這種復雜的排序邏輯。

場景描述

假設我們有一個車輛保險信息的查詢功能，需要根據gateNo進行排序。gateNo字段的格式為X-YYY，其中X可以是數字或特殊字符串（如“天”），YYY是一個數字。我們的目標是按X的數值排序，將X為“天”的情況排在最后，同時YYY也需要按數值進行排序。

解決方案

在MyBatis-Plus中，QueryWrapper提供了豐富的方法來構建查詢條件，但對于復雜的排序邏輯，我們需要利用原始SQL片段來實現。以下是具體的實現步驟和代碼：

步驟1：定義QueryWrapper

首先，我們創(chuàng)建一個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);

詳細解釋

特殊值處理：

使用 CASE 語句判斷 gate_no 是否以“天-”開頭。如果是，則賦值為 2，這樣的記錄會被排在查詢結果的最后。

按 X 排序：

使用 SUBSTRING_INDEX 函數提取 gate_no 中 ‘-’ 前的部分（即 X），然后用 CAST 函數將其轉換為無符號整數，確保按數值進行排序。

按 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;
}