|
|
@@ -15,9 +15,39 @@
|
|
|
*/
|
|
|
package com.baomidou.mybatisplus.core.conditions;
|
|
|
|
|
|
+import static com.baomidou.mybatisplus.core.enums.SqlKeyword.AND;
|
|
|
+import static com.baomidou.mybatisplus.core.enums.SqlKeyword.BETWEEN;
|
|
|
+import static com.baomidou.mybatisplus.core.enums.SqlKeyword.EQ;
|
|
|
+import static com.baomidou.mybatisplus.core.enums.SqlKeyword.EXISTS;
|
|
|
+import static com.baomidou.mybatisplus.core.enums.SqlKeyword.GE;
|
|
|
+import static com.baomidou.mybatisplus.core.enums.SqlKeyword.GROUP_BY;
|
|
|
+import static com.baomidou.mybatisplus.core.enums.SqlKeyword.GT;
|
|
|
+import static com.baomidou.mybatisplus.core.enums.SqlKeyword.HAVING;
|
|
|
+import static com.baomidou.mybatisplus.core.enums.SqlKeyword.IN;
|
|
|
+import static com.baomidou.mybatisplus.core.enums.SqlKeyword.IS_NOT_NULL;
|
|
|
+import static com.baomidou.mybatisplus.core.enums.SqlKeyword.IS_NULL;
|
|
|
+import static com.baomidou.mybatisplus.core.enums.SqlKeyword.LE;
|
|
|
+import static com.baomidou.mybatisplus.core.enums.SqlKeyword.LIKE;
|
|
|
+import static com.baomidou.mybatisplus.core.enums.SqlKeyword.LT;
|
|
|
+import static com.baomidou.mybatisplus.core.enums.SqlKeyword.NE;
|
|
|
+import static com.baomidou.mybatisplus.core.enums.SqlKeyword.NOT;
|
|
|
+import static com.baomidou.mybatisplus.core.enums.SqlKeyword.OR;
|
|
|
+import static com.baomidou.mybatisplus.core.enums.SqlKeyword.ORDER_BY;
|
|
|
+
|
|
|
+import java.util.ArrayList;
|
|
|
+import java.util.Arrays;
|
|
|
+import java.util.HashMap;
|
|
|
+import java.util.List;
|
|
|
+import java.util.Map;
|
|
|
+import java.util.concurrent.atomic.AtomicInteger;
|
|
|
import java.util.function.Function;
|
|
|
+import java.util.stream.Collectors;
|
|
|
|
|
|
import com.baomidou.mybatisplus.core.enums.SqlKeyword;
|
|
|
+import com.baomidou.mybatisplus.core.toolkit.ArrayUtils;
|
|
|
+import com.baomidou.mybatisplus.core.toolkit.StringUtils;
|
|
|
+import com.baomidou.mybatisplus.core.toolkit.TableInfoHelper;
|
|
|
+import com.baomidou.mybatisplus.core.toolkit.support.Property;
|
|
|
|
|
|
|
|
|
/**
|
|
|
@@ -28,19 +58,368 @@ import com.baomidou.mybatisplus.core.enums.SqlKeyword;
|
|
|
* @author hubin
|
|
|
* @since 2017-05-26
|
|
|
*/
|
|
|
-public class QueryWrapper<T> extends Wrapper<T, QueryWrapper<T>> {
|
|
|
+public class QueryWrapper<T, Q extends QueryWrapper<T, Q>> extends Wrapper<T> {
|
|
|
+
|
|
|
+ private List<ISqlSegment> expression = new ArrayList<>();
|
|
|
+ private static final String MP_GENERAL_PARAMNAME = "MPGENVAL";
|
|
|
+ private final AtomicInteger paramNameSeq = new AtomicInteger(0);
|
|
|
+ private final Map<String, Object> paramNameValuePairs = new HashMap<>();
|
|
|
+ protected String paramAlias = null;
|
|
|
+ private static final String DEFAULT_PARAM_ALIAS = "ew";
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 占位符
|
|
|
+ */
|
|
|
+ private static final String PLACE_HOLDER = "{%s}";
|
|
|
+
|
|
|
+ private static final String MYBATIS_PLUS_TOKEN = "#{%s.paramNameValuePairs.%s}";
|
|
|
+
|
|
|
+ public Q apply(String condition) {
|
|
|
+ expression.add(() -> condition);
|
|
|
+ return typedThis();
|
|
|
+ }
|
|
|
+
|
|
|
+ public Q notIn(String condition) {
|
|
|
+ return not().in(condition);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * LIKE '%值%'
|
|
|
+ */
|
|
|
+ public Q like(String column, Object val) {
|
|
|
+ return like(true, column, val);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * LIKE '%值%'
|
|
|
+ */
|
|
|
+ public Q like(boolean condition, String column, Object val) {
|
|
|
+ return doIt(condition, () -> column, LIKE, () -> "'%", () -> formatSql("{0}", val), () -> "%'");
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * LIKE '%值'
|
|
|
+ */
|
|
|
+ public Q likeLeft(String column, Object val) {
|
|
|
+ return likeLeft(true, column, val);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * LIKE '%值'
|
|
|
+ */
|
|
|
+ public Q likeLeft(boolean condition, String column, Object val) {
|
|
|
+ return doIt(condition, () -> column, LIKE, () -> "'%", () -> formatSql("{0}", val), () -> "'");
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * LIKE '值%'
|
|
|
+ */
|
|
|
+ public Q likeRight(String column, Object val) {
|
|
|
+ return likeRight(true, column, val);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * LIKE '值%'
|
|
|
+ */
|
|
|
+ public Q likeRight(boolean condition, String column, Object val) {
|
|
|
+ return doIt(condition, () -> column, LIKE, () -> "'", () -> formatSql("{0}", val), () -> "%'");
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 等于 =
|
|
|
+ */
|
|
|
+ public Q eq(Property<T, ?> property, Object val) {
|
|
|
+ return eq(true, property, val);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 等于 =
|
|
|
+ */
|
|
|
+ public Q eq(boolean condition, Property<T, ?> property, Object val) {
|
|
|
+ return addCondition(condition, property, EQ, val);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 不等于 <>
|
|
|
+ */
|
|
|
+ public Q ne(Property<T, ?> property, Object val) {
|
|
|
+ return ne(true, property, val);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 不等于 <>
|
|
|
+ */
|
|
|
+ public Q ne(boolean condition, Property<T, ?> property, Object val) {
|
|
|
+ return addCondition(condition, property, NE, val);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 大于 >
|
|
|
+ */
|
|
|
+ public Q gt(Property<T, ?> property, Object val) {
|
|
|
+ return gt(true, property, val);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 大于 >
|
|
|
+ */
|
|
|
+ public Q gt(boolean condition, Property<T, ?> property, Object val) {
|
|
|
+ return addCondition(condition, property, GT, val);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 大于等于 >=
|
|
|
+ */
|
|
|
+ public Q ge(Property<T, ?> property, Object val) {
|
|
|
+ return ge(true, property, val);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 大于等于 >=
|
|
|
+ */
|
|
|
+ public Q ge(boolean condition, Property<T, ?> property, Object val) {
|
|
|
+ return addCondition(condition, property, GE, val);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 小于 <
|
|
|
+ */
|
|
|
+ public Q lt(Property<T, ?> property, Object val) {
|
|
|
+ return lt(true, property, val);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 小于 <
|
|
|
+ */
|
|
|
+ public Q lt(boolean condition, Property<T, ?> property, Object val) {
|
|
|
+ return addCondition(condition, property, LT, val);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 小于等于 <=
|
|
|
+ */
|
|
|
+ public Q le(Property<T, ?> property, Object val) {
|
|
|
+ return le(true, property, val);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 小于等于 <=
|
|
|
+ */
|
|
|
+ public Q le(boolean condition, Property<T, ?> property, Object val) {
|
|
|
+ return addCondition(condition, property, LE, val);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * BETWEEN 值1 AND 值2
|
|
|
+ */
|
|
|
+ public Q between(String column, Object val1, Object val2) {
|
|
|
+ return between(true, column, "val1", "val2");
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * BETWEEN 值1 AND 值2
|
|
|
+ */
|
|
|
+ public Q between(boolean condition, String column, Object val1, Object val2) {
|
|
|
+ return doIt(condition, () -> column, BETWEEN, () -> "val1", AND, () -> "val2");
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 字段 IS NULL
|
|
|
+ */
|
|
|
+ public Q isNull(String column) {
|
|
|
+ return isNull(true, column);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 字段 IS NULL
|
|
|
+ */
|
|
|
+ public Q isNull(boolean condition, String column) {
|
|
|
+ return doIt(condition, () -> column, IS_NULL);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 字段 IS NOT NULL
|
|
|
+ */
|
|
|
+ public Q isNotNull(String column) {
|
|
|
+ return isNotNull(true, column);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 字段 IS NOT NULL
|
|
|
+ */
|
|
|
+ public Q isNotNull(boolean condition, String column) {
|
|
|
+ return doIt(condition, () -> column, IS_NOT_NULL);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 分组：GROUP BY 字段, ...
|
|
|
+ */
|
|
|
+ public Q groupBy(String column) {
|
|
|
+ return doIt(true, GROUP_BY, () -> column);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 排序：ORDER BY 字段, ...
|
|
|
+ */
|
|
|
+ public Q orderBy(String column) {//todo 产生的sql有bug
|
|
|
+ return doIt(true, ORDER_BY, () -> column);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * HAVING 关键词
|
|
|
+ */
|
|
|
+ public Q having() {
|
|
|
+ return doIt(true, HAVING);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * exists ( sql 语句 )
|
|
|
+ */
|
|
|
+ public Q exists(String condition) {
|
|
|
+ return this.addNestedCondition(condition, EXISTS);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * LAST 拼接在 SQL 末尾
|
|
|
+ */
|
|
|
+ public Q last(String condition) {
|
|
|
+ return doIt(true, () -> condition);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * NOT 关键词
|
|
|
+ */
|
|
|
+ protected Q not() {//todo 待考虑
|
|
|
+ return doIt(true, NOT);
|
|
|
+ }
|
|
|
+
|
|
|
+ public Q and() {
|
|
|
+ expression.add(AND);
|
|
|
+ return typedThis();
|
|
|
+ }
|
|
|
+
|
|
|
+ public Q and(Function<Q, Q> func) {
|
|
|
+ return addNestedCondition(func, AND);
|
|
|
+ }
|
|
|
+
|
|
|
+ public Q or(Function<Q, Q> func) {
|
|
|
+ return addNestedCondition(func, OR);
|
|
|
+ }
|
|
|
+
|
|
|
+ public Q in(String condition) {//todo 待动
|
|
|
+ return addNestedCondition(condition, IN);
|
|
|
+ }
|
|
|
+
|
|
|
+ public Q or(Property<T, ?> property, Object val) {
|
|
|
+ //todo 待动
|
|
|
+ return addCondition(true, property, OR, val);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * <p>
|
|
|
+ * 嵌套查询条件
|
|
|
+ * </p>
|
|
|
+ *
|
|
|
+ * @param val 查询条件值
|
|
|
+ * @param sqlKeyword SQL 关键词
|
|
|
+ * @return this
|
|
|
+ */
|
|
|
+ protected Q addNestedCondition(Object val, SqlKeyword sqlKeyword) {
|
|
|
+ return doIt(true, sqlKeyword, () -> this.formatSql("({0})", val));
|
|
|
+ }
|
|
|
|
|
|
/**
|
|
|
* <p>
|
|
|
- * 多重嵌套查询条件
|
|
|
+ * 普通查询条件
|
|
|
* </p>
|
|
|
*
|
|
|
- * @param condition 查询条件值
|
|
|
+ * @param condition 是否执行
|
|
|
+ * @param property 属性
|
|
|
* @param sqlKeyword SQL 关键词
|
|
|
+ * @param val 条件值
|
|
|
+ * @return this
|
|
|
+ */
|
|
|
+ protected Q addCondition(boolean condition, Property<T, ?> property, SqlKeyword sqlKeyword, Object val) {
|
|
|
+ return doIt(condition, () -> TableInfoHelper.toColumn(property),
|
|
|
+ sqlKeyword, () -> this.formatSql("{0}", val));
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * <p>
|
|
|
+ * 格式化SQL
|
|
|
+ * </p>
|
|
|
+ *
|
|
|
+ * @param sqlStr SQL语句部分
|
|
|
+ * @param params 参数集
|
|
|
* @return this
|
|
|
*/
|
|
|
- protected QueryWrapper addNestedCondition(Function<QueryWrapper, QueryWrapper> condition, SqlKeyword sqlKeyword) {
|
|
|
- return doIt(true, sqlKeyword, () -> "(", condition.apply(new QueryWrapper<T>()), () -> ")");
|
|
|
+ protected String formatSql(String sqlStr, Object... params) {
|
|
|
+ return formatSqlIfNeed(true, sqlStr, params);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * <p>
|
|
|
+ * 根据需要格式化SQL<BR>
|
|
|
+ * <BR>
|
|
|
+ * Format SQL for methods: EntityQ<T>.where/and/or...("name={0}", value);
|
|
|
+ * ALL the {<b>i</b>} will be replaced with #{MPGENVAL<b>i</b>}<BR>
|
|
|
+ * <BR>
|
|
|
+ * ew.where("sample_name=<b>{0}</b>", "haha").and("sample_age ><b>{0}</b>
|
|
|
+ * and sample_age<<b>{1}</b>", 18, 30) <b>TO</b>
|
|
|
+ * sample_name=<b>#{MPGENVAL1}</b> and sample_age>#<b>{MPGENVAL2}</b> and
|
|
|
+ * sample_age<<b>#{MPGENVAL3}</b><BR>
|
|
|
+ * </p>
|
|
|
+ *
|
|
|
+ * @param need 是否需要格式化
|
|
|
+ * @param sqlStr SQL语句部分
|
|
|
+ * @param params 参数集
|
|
|
+ * @return this
|
|
|
+ */
|
|
|
+ protected String formatSqlIfNeed(boolean need, String sqlStr, Object... params) {
|
|
|
+ if (!need || StringUtils.isEmpty(sqlStr)) {
|
|
|
+ return null;
|
|
|
+ }
|
|
|
+ if (ArrayUtils.isNotEmpty(params)) {
|
|
|
+ for (int i = 0; i < params.length; ++i) {
|
|
|
+ String genParamName = MP_GENERAL_PARAMNAME + paramNameSeq.incrementAndGet();
|
|
|
+ sqlStr = sqlStr.replace(String.format(PLACE_HOLDER, i),
|
|
|
+ String.format(MYBATIS_PLUS_TOKEN, getParamAlias(), genParamName));
|
|
|
+ paramNameValuePairs.put(genParamName, params[i]);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return sqlStr;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * <p>
|
|
|
+ * 对sql片段进行组装
|
|
|
+ * </p>
|
|
|
+ *
|
|
|
+ * @param condition 是否执行
|
|
|
+ * @param sqlSegments sql片段数组
|
|
|
+ * @return this
|
|
|
+ */
|
|
|
+ protected Q doIt(boolean condition, ISqlSegment... sqlSegments) {
|
|
|
+ if (condition) {
|
|
|
+ expression.addAll(Arrays.asList(sqlSegments));
|
|
|
+ }
|
|
|
+ return typedThis();
|
|
|
+ }
|
|
|
+
|
|
|
+ public String getParamAlias() {
|
|
|
+ return StringUtils.isEmpty(paramAlias) ? DEFAULT_PARAM_ALIAS : paramAlias;
|
|
|
+ }
|
|
|
+
|
|
|
+ @SuppressWarnings("unchecked")
|
|
|
+ protected Q typedThis() {
|
|
|
+ return (Q) this;
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public String getSqlSegment() {
|
|
|
+ return String.join(" ", expression.stream()
|
|
|
+ .map(ISqlSegment::getSqlSegment)
|
|
|
+ .collect(Collectors.toList()));
|
|
|
}
|
|
|
|
|
|
}
|
hubin