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

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

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

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

場景描述

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

解決方案

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

步驟1：定義QueryWrapper

首先，我們創(chuàng)建一個QueryWrapper實例。如果查詢模型（model）為空，我們直接返回一個空的QueryWrapper實例。

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

步驟2：添加自定義排序邏輯

為了根據(jù)gateNo進行排序，我們使用last方法來添加自定義的SQL排序片段。這里使用CASE語句來處理特殊值“天”，并使用CAST函數(shù)和SUBSTRING_INDEX來分離和轉(zhuǎn)換字符串為數(shù)值。

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，這樣的記錄會被排在查詢結(jié)果的最后。

按 X 排序：

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

按 YYY 排序：

同樣使用 SUBSTRING_INDEX 函數(shù)但這次提取 ‘-’ 后的部分（即 YYY），再通過 CAST 進行數(shù)值轉(zhuǎn)換，保證在相同的 X 內(nèi)部，根據(jù) YYY 的值進行排序。

步驟3：添加其他查詢條件

在實際的應用場景中，除了排序之外，我們通常還需要根據(jù)多個字段添加過濾條件。例如，我們可以根據(jù)車輛ID、所有者ID和其他相關(guān)字段來過濾數(shù)據(jù)。這里為了簡化，我們只展示幾個示例條件：

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

完整方法

將以上步驟組合，我們得到了一個完整的方法，用于構(gòu)建帶有自定義排序的查詢包裝器：

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