From e9bfa1396ff47d171b3052a606e0931e6f93cc9c Mon Sep 17 00:00:00 2001
From: houzhongjian <houzhongyi@126.com>
Date: 星期四, 10 四月 2025 12:05:23 +0800
Subject: [PATCH] 还原framework代码
---
iailab-framework/iailab-common-mybatis/src/main/java/com/iailab/framework/mybatis/core/query/QueryWrapperX.java | 166 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 166 insertions(+), 0 deletions(-)
diff --git a/iailab-framework/iailab-common-mybatis/src/main/java/com/iailab/framework/mybatis/core/query/QueryWrapperX.java b/iailab-framework/iailab-common-mybatis/src/main/java/com/iailab/framework/mybatis/core/query/QueryWrapperX.java
new file mode 100644
index 0000000..bf79bf5
--- /dev/null
+++ b/iailab-framework/iailab-common-mybatis/src/main/java/com/iailab/framework/mybatis/core/query/QueryWrapperX.java
@@ -0,0 +1,166 @@
+package com.iailab.framework.mybatis.core.query;
+
+import cn.hutool.core.lang.Assert;
+import com.iailab.framework.mybatis.core.enums.SqlConstants;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.core.toolkit.ArrayUtils;
+import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
+import org.springframework.util.StringUtils;
+
+import java.util.Collection;
+
+/**
+ * 拓展 MyBatis Plus QueryWrapper 类，主要增加如下功能：
+ *
+ * 1. 拼接条件的方法，增加 xxxIfPresent 方法，用于判断值不存在的时候，不要拼接到条件中。
+ *
+ * @param <T> 数据类型
+ */
+public class QueryWrapperX<T> extends QueryWrapper<T> {
+
+ public QueryWrapperX<T> likeIfPresent(String column, String val) {
+ if (StringUtils.hasText(val)) {
+ return (QueryWrapperX<T>) super.like(column, val);
+ }
+ return this;
+ }
+
+ public QueryWrapperX<T> inIfPresent(String column, Collection<?> values) {
+ if (!CollectionUtils.isEmpty(values)) {
+ return (QueryWrapperX<T>) super.in(column, values);
+ }
+ return this;
+ }
+
+ public QueryWrapperX<T> inIfPresent(String column, Object... values) {
+ if (!ArrayUtils.isEmpty(values)) {
+ return (QueryWrapperX<T>) super.in(column, values);
+ }
+ return this;
+ }
+
+ public QueryWrapperX<T> eqIfPresent(String column, Object val) {
+ if (val != null) {
+ return (QueryWrapperX<T>) super.eq(column, val);
+ }
+ return this;
+ }
+
+ public QueryWrapperX<T> neIfPresent(String column, Object val) {
+ if (val != null) {
+ return (QueryWrapperX<T>) super.ne(column, val);
+ }
+ return this;
+ }
+
+ public QueryWrapperX<T> gtIfPresent(String column, Object val) {
+ if (val != null) {
+ return (QueryWrapperX<T>) super.gt(column, val);
+ }
+ return this;
+ }
+
+ public QueryWrapperX<T> geIfPresent(String column, Object val) {
+ if (val != null) {
+ return (QueryWrapperX<T>) super.ge(column, val);
+ }
+ return this;
+ }
+
+ public QueryWrapperX<T> ltIfPresent(String column, Object val) {
+ if (val != null) {
+ return (QueryWrapperX<T>) super.lt(column, val);
+ }
+ return this;
+ }
+
+ public QueryWrapperX<T> leIfPresent(String column, Object val) {
+ if (val != null) {
+ return (QueryWrapperX<T>) super.le(column, val);
+ }
+ return this;
+ }
+
+ public QueryWrapperX<T> betweenIfPresent(String column, Object val1, Object val2) {
+ if (val1 != null && val2 != null) {
+ return (QueryWrapperX<T>) super.between(column, val1, val2);
+ }
+ if (val1 != null) {
+ return (QueryWrapperX<T>) ge(column, val1);
+ }
+ if (val2 != null) {
+ return (QueryWrapperX<T>) le(column, val2);
+ }
+ return this;
+ }
+
+ public QueryWrapperX<T> betweenIfPresent(String column, Object[] values) {
+ if (values!= null && values.length != 0 && values[0] != null && values[1] != null) {
+ return (QueryWrapperX<T>) super.between(column, values[0], values[1]);
+ }
+ if (values!= null && values.length != 0 && values[0] != null) {
+ return (QueryWrapperX<T>) ge(column, values[0]);
+ }
+ if (values!= null && values.length != 0 && values[1] != null) {
+ return (QueryWrapperX<T>) le(column, values[1]);
+ }
+ return this;
+ }
+
+ // ========== 重写父类方法，方便链式调用 ==========
+
+ @Override
+ public QueryWrapperX<T> eq(boolean condition, String column, Object val) {
+ super.eq(condition, column, val);
+ return this;
+ }
+
+ @Override
+ public QueryWrapperX<T> eq(String column, Object val) {
+ super.eq(column, val);
+ return this;
+ }
+
+ @Override
+ public QueryWrapperX<T> orderByDesc(String column) {
+ super.orderByDesc(true, column);
+ return this;
+ }
+
+ @Override
+ public QueryWrapperX<T> last(String lastSql) {
+ super.last(lastSql);
+ return this;
+ }
+
+ @Override
+ public QueryWrapperX<T> in(String column, Collection<?> coll) {
+ super.in(column, coll);
+ return this;
+ }
+
+ /**
+ * 设置只返回最后一条
+ *
+ * TODO iailab：不是完美解，需要在思考下。如果使用多数据源，并且数据源是多种类型时，可能会存在问题：实现之返回一条的语法不同
+ *
+ * @return this
+ */
+ public QueryWrapperX<T> limitN(int n) {
+ Assert.notNull(SqlConstants.DB_TYPE, "获取不到数据库的类型");
+ switch (SqlConstants.DB_TYPE) {
+ case ORACLE:
+ case ORACLE_12C:
+ super.le("ROWNUM", n);
+ break;
+ case SQL_SERVER:
+ case SQL_SERVER2005:
+ super.select("TOP " + n + " *"); // 由于 SQL Server 是通过 SELECT TOP 1 实现限制一条，所以只好使用 * 查询剩余字段
+ break;
+ default:
+ super.last("LIMIT " + n);
+ }
+ return this;
+ }
+
+}
--
Gitblit v1.9.3