From 152781b05131e48bf6e94d71cc72dd54af52a3fb Mon Sep 17 00:00:00 2001
From: houzhongjian <houzhongyi@126.com>
Date: 星期四, 10 四月 2025 14:13:29 +0800
Subject: [PATCH] 恢复iailab-framework
---
iailab-framework/iailab-common-mybatis/src/main/java/com/iailab/framework/mybatis/core/util/MyBatisUtils.java | 106 +++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 106 insertions(+), 0 deletions(-)
diff --git a/iailab-framework/iailab-common-mybatis/src/main/java/com/iailab/framework/mybatis/core/util/MyBatisUtils.java b/iailab-framework/iailab-common-mybatis/src/main/java/com/iailab/framework/mybatis/core/util/MyBatisUtils.java
new file mode 100644
index 0000000..a8d8411
--- /dev/null
+++ b/iailab-framework/iailab-common-mybatis/src/main/java/com/iailab/framework/mybatis/core/util/MyBatisUtils.java
@@ -0,0 +1,106 @@
+package com.iailab.framework.mybatis.core.util;
+
+import cn.hutool.core.collection.CollectionUtil;
+import cn.hutool.core.util.StrUtil;
+import com.baomidou.mybatisplus.annotation.DbType;
+import com.baomidou.mybatisplus.core.metadata.OrderItem;
+import com.baomidou.mybatisplus.core.toolkit.StringPool;
+import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
+import com.baomidou.mybatisplus.extension.plugins.inner.InnerInterceptor;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.iailab.framework.common.pojo.PageParam;
+import com.iailab.framework.common.pojo.SortingField;
+import com.iailab.framework.mybatis.core.enums.DbTypeEnum;
+import net.sf.jsqlparser.expression.Alias;
+import net.sf.jsqlparser.schema.Column;
+import net.sf.jsqlparser.schema.Table;
+
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.List;
+import java.util.stream.Collectors;
+
+/**
+ * MyBatis 工具类
+ */
+public class MyBatisUtils {
+
+ private static final String MYSQL_ESCAPE_CHARACTER = "`";
+
+ public static <T> Page<T> buildPage(PageParam pageParam) {
+ return buildPage(pageParam, null);
+ }
+
+ public static <T> Page<T> buildPage(PageParam pageParam, Collection<SortingField> sortingFields) {
+ // 页码 + 数量
+ Page<T> page = new Page<>(pageParam.getPageNo(), pageParam.getPageSize());
+ // 排序字段
+ if (!CollectionUtil.isEmpty(sortingFields)) {
+ page.addOrder(sortingFields.stream().map(sortingField -> SortingField.ORDER_ASC.equals(sortingField.getOrder()) ?
+ OrderItem.asc(sortingField.getField()) : OrderItem.desc(sortingField.getField()))
+ .collect(Collectors.toList()));
+ }
+ return page;
+ }
+
+ /**
+ * 将拦截器添加到链中
+ * 由于 MybatisPlusInterceptor 不支持添加拦截器，所以只能全量设置
+ *
+ * @param interceptor 链
+ * @param inner 拦截器
+ * @param index 位置
+ */
+ public static void addInterceptor(MybatisPlusInterceptor interceptor, InnerInterceptor inner, int index) {
+ List<InnerInterceptor> inners = new ArrayList<>(interceptor.getInterceptors());
+ inners.add(index, inner);
+ interceptor.setInterceptors(inners);
+ }
+
+ /**
+ * 获得 Table 对应的表名
+ * <p>
+ * 兼容 MySQL 转义表名 `t_xxx`
+ *
+ * @param table 表
+ * @return 去除转移字符后的表名
+ */
+ public static String getTableName(Table table) {
+ String tableName = table.getName();
+ if (tableName.startsWith(MYSQL_ESCAPE_CHARACTER) && tableName.endsWith(MYSQL_ESCAPE_CHARACTER)) {
+ tableName = tableName.substring(1, tableName.length() - 1);
+ }
+ return tableName;
+ }
+
+ /**
+ * 构建 Column 对象
+ *
+ * @param tableName 表名
+ * @param tableAlias 别名
+ * @param column 字段名
+ * @return Column 对象
+ */
+ public static Column buildColumn(String tableName, Alias tableAlias, String column) {
+ if (tableAlias != null) {
+ tableName = tableAlias.getName();
+ }
+ return new Column(tableName + StringPool.DOT + column);
+ }
+
+ /**
+ * 跨数据库的 find_in_set 实现
+ *
+ * @param column 字段名称
+ * @param value 查询值(不带单引号)
+ * @return sql
+ */
+ public static String findInSet(String column, Object value) {
+ // 这里不用SqlConstants.DB_TYPE，因为它是使用 primary 数据源的 url 推断出来的类型
+ DbType dbType = JdbcUtils.getDbType();
+ return DbTypeEnum.getFindInSetTemplate(dbType)
+ .replace("#{column}", column)
+ .replace("#{value}", StrUtil.toString(value));
+ }
+
+}
--
Gitblit v1.9.3