orderBy支持resultType字段

This commit is contained in:
yulichang 2024-05-18 00:51:07 +08:00
parent 54f890f0de
commit da15b11833
4 changed files with 64 additions and 38 deletions

View File

@ -217,18 +217,26 @@ public abstract class JoinAbstractLambdaWrapper<T, Children extends JoinAbstract
} }
@Override @Override
protected <X> String columnToString(Integer index, String alias, X column, boolean isJoin, PrefixEnum prefixEnum) { protected <X> String columnToString(Integer index, String alias, X column, boolean isJoin, PrefixEnum prefixEnum, boolean orderBy) {
return columnToString(index, alias, (SFunction<?, ?>) column, isJoin, prefixEnum); return columnToString(index, alias, (SFunction<?, ?>) column, isJoin, prefixEnum, orderBy);
} }
@Override @Override
@SafeVarargs @SafeVarargs
protected final <X> String columnsToString(Integer index, PrefixEnum prefixEnum, String alias, X... columns) { protected final <X> String columnsToString(Integer index, PrefixEnum prefixEnum, String alias,boolean orderBy, X... columns) {
return Arrays.stream(columns).map(i -> columnToString(index, alias, (SFunction<?, ?>) i, false, prefixEnum)).collect(joining(StringPool.COMMA)); return Arrays.stream(columns).map(i ->
columnToString(index, alias, (SFunction<?, ?>) i, false, prefixEnum, orderBy))
.collect(joining(StringPool.COMMA));
} }
protected String columnToString(Integer index, String alias, SFunction<?, ?> column, boolean isJoin, PrefixEnum prefixEnum) { protected String columnToString(Integer index, String alias, SFunction<?, ?> column, boolean isJoin, PrefixEnum prefixEnum, boolean orderBy) {
Class<?> entityClass = LambdaUtils.getEntityClass(column); Class<?> entityClass = LambdaUtils.getEntityClass(column);
if (orderBy) {
TableInfo info = TableHelper.get(entityClass);
if (Objects.isNull(info)) {
return LambdaUtils.getName(column);
}
}
return (alias == null ? getDefault(index, entityClass, isJoin, prefixEnum) : alias) + StringPool.DOT + getCache(column).getColumn(); return (alias == null ? getDefault(index, entityClass, isJoin, prefixEnum) : alias) + StringPool.DOT + getCache(column).getColumn();
} }

View File

@ -289,13 +289,13 @@ public abstract class JoinAbstractWrapper<T, Children extends JoinAbstractWrappe
@Override @Override
public <X> Children between(boolean condition, String alias, SFunction<X, ?> column, Object val1, Object val2) { public <X> Children between(boolean condition, String alias, SFunction<X, ?> column, Object val1, Object val2) {
return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column), BETWEEN, return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column, false), BETWEEN,
() -> formatParam(null, val1), AND, () -> formatParam(null, val2))); () -> formatParam(null, val1), AND, () -> formatParam(null, val2)));
} }
@Override @Override
public <X> Children notBetween(boolean condition, String alias, SFunction<X, ?> column, Object val1, Object val2) { public <X> Children notBetween(boolean condition, String alias, SFunction<X, ?> column, Object val1, Object val2) {
return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column), NOT_BETWEEN, return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column, false), NOT_BETWEEN,
() -> formatParam(null, val1), AND, () -> formatParam(null, val2))); () -> formatParam(null, val1), AND, () -> formatParam(null, val2)));
} }
@ -374,74 +374,74 @@ public abstract class JoinAbstractWrapper<T, Children extends JoinAbstractWrappe
@Override @Override
public <X> Children isNull(boolean condition, String alias, SFunction<X, ?> column) { public <X> Children isNull(boolean condition, String alias, SFunction<X, ?> column) {
return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column), IS_NULL)); return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column, false), IS_NULL));
} }
@Override @Override
public <X> Children isNotNull(boolean condition, String alias, SFunction<X, ?> column) { public <X> Children isNotNull(boolean condition, String alias, SFunction<X, ?> column) {
return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column), IS_NOT_NULL)); return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column, false), IS_NOT_NULL));
} }
@Override @Override
public <X> Children in(boolean condition, String alias, SFunction<X, ?> column, Collection<?> coll) { public <X> Children in(boolean condition, String alias, SFunction<X, ?> column, Collection<?> coll) {
return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column), IN, inExpression(coll))); return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column, false), IN, inExpression(coll)));
} }
@Override @Override
public <X> Children in(boolean condition, String alias, SFunction<X, ?> column, Object... values) { public <X> Children in(boolean condition, String alias, SFunction<X, ?> column, Object... values) {
return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column), IN, inExpression(values))); return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column, false), IN, inExpression(values)));
} }
@Override @Override
public <X> Children notIn(boolean condition, String alias, SFunction<X, ?> column, Collection<?> coll) { public <X> Children notIn(boolean condition, String alias, SFunction<X, ?> column, Collection<?> coll) {
return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column), NOT_IN, inExpression(coll))); return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column, false), NOT_IN, inExpression(coll)));
} }
@Override @Override
public <X> Children notIn(boolean condition, String alias, SFunction<X, ?> column, Object... values) { public <X> Children notIn(boolean condition, String alias, SFunction<X, ?> column, Object... values) {
return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column), NOT_IN, inExpression(values))); return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column, false), NOT_IN, inExpression(values)));
} }
@Override @Override
public <X> Children inSql(boolean condition, String alias, SFunction<X, ?> column, String inValue) { public <X> Children inSql(boolean condition, String alias, SFunction<X, ?> column, String inValue) {
return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column), IN, return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column, false), IN,
() -> String.format("(%s)", inValue))); () -> String.format("(%s)", inValue)));
} }
@Override @Override
public <X> Children notInSql(boolean condition, String alias, SFunction<X, ?> column, String inValue) { public <X> Children notInSql(boolean condition, String alias, SFunction<X, ?> column, String inValue) {
return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column), NOT_IN, return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column, false), NOT_IN,
() -> String.format("(%s)", inValue))); () -> String.format("(%s)", inValue)));
} }
@Override @Override
public <X> Children gtSql(boolean condition, String alias, SFunction<X, ?> column, String inValue) { public <X> Children gtSql(boolean condition, String alias, SFunction<X, ?> column, String inValue) {
return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column), GT, return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column, false), GT,
() -> String.format("(%s)", inValue))); () -> String.format("(%s)", inValue)));
} }
@Override @Override
public <X> Children geSql(boolean condition, String alias, SFunction<X, ?> column, String inValue) { public <X> Children geSql(boolean condition, String alias, SFunction<X, ?> column, String inValue) {
return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column), GE, return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column, false), GE,
() -> String.format("(%s)", inValue))); () -> String.format("(%s)", inValue)));
} }
@Override @Override
public <X> Children ltSql(boolean condition, String alias, SFunction<X, ?> column, String inValue) { public <X> Children ltSql(boolean condition, String alias, SFunction<X, ?> column, String inValue) {
return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column), LT, return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column, false), LT,
() -> String.format("(%s)", inValue))); () -> String.format("(%s)", inValue)));
} }
@Override @Override
public <X> Children leSql(boolean condition, String alias, SFunction<X, ?> column, String inValue) { public <X> Children leSql(boolean condition, String alias, SFunction<X, ?> column, String inValue) {
return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column), LE, return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column, false), LE,
() -> String.format("(%s)", inValue))); () -> String.format("(%s)", inValue)));
} }
@Override @Override
public <R> Children eqSql(boolean condition, String alias, SFunction<R, ?> column, String inValue) { public <R> Children eqSql(boolean condition, String alias, SFunction<R, ?> column, String inValue) {
return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column), EQ, return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column, false), EQ,
() -> String.format("(%s)", inValue))); () -> String.format("(%s)", inValue)));
} }
@ -449,7 +449,7 @@ public abstract class JoinAbstractWrapper<T, Children extends JoinAbstractWrappe
public <R> Children groupBy(boolean condition, String alias, List<SFunction<R, ?>> columns) { public <R> Children groupBy(boolean condition, String alias, List<SFunction<R, ?>> columns) {
return maybeDo(condition, () -> { return maybeDo(condition, () -> {
if (CollectionUtils.isNotEmpty(columns)) { if (CollectionUtils.isNotEmpty(columns)) {
final String finalOne = columnsToString(index, isOn ? PrefixEnum.ON_FIRST : PrefixEnum.CD_FIRST, alias, columns); final String finalOne = columnsToString(index, isOn ? PrefixEnum.ON_FIRST : PrefixEnum.CD_FIRST, alias, false, columns);
appendSqlSegments(GROUP_BY, () -> finalOne); appendSqlSegments(GROUP_BY, () -> finalOne);
} }
}); });
@ -477,9 +477,9 @@ public abstract class JoinAbstractWrapper<T, Children extends JoinAbstractWrappe
@SafeVarargs @SafeVarargs
public final <X> Children groupBy(boolean condition, String alias, SFunction<X, ?> column, SFunction<X, ?>... columns) { public final <X> Children groupBy(boolean condition, String alias, SFunction<X, ?> column, SFunction<X, ?>... columns) {
return maybeDo(condition, () -> { return maybeDo(condition, () -> {
String one = columnToString(index, alias, column, false, isOn ? PrefixEnum.ON_FIRST : PrefixEnum.CD_FIRST); String one = columnToString(index, alias, column, false, isOn ? PrefixEnum.ON_FIRST : PrefixEnum.CD_FIRST, false);
if (ArrayUtils.isNotEmpty(columns)) { if (ArrayUtils.isNotEmpty(columns)) {
one += (StringPool.COMMA + columnsToString(index, isOn ? PrefixEnum.ON_FIRST : PrefixEnum.CD_FIRST, alias, columns)); one += (StringPool.COMMA + columnsToString(index, isOn ? PrefixEnum.ON_FIRST : PrefixEnum.CD_FIRST, alias, false, columns));
} }
final String finalOne = one; final String finalOne = one;
appendSqlSegments(GROUP_BY, () -> finalOne); appendSqlSegments(GROUP_BY, () -> finalOne);
@ -491,7 +491,7 @@ public abstract class JoinAbstractWrapper<T, Children extends JoinAbstractWrappe
return maybeDo(condition, () -> { return maybeDo(condition, () -> {
if (CollectionUtils.isNotEmpty(columns)) { if (CollectionUtils.isNotEmpty(columns)) {
columns.forEach(c -> appendSqlSegments(ORDER_BY, columns.forEach(c -> appendSqlSegments(ORDER_BY,
columnToSqlSegment(index, alias, columnSqlInjectFilter(c)), ASC)); columnToSqlSegment(index, alias, columnSqlInjectFilter(c), true), ASC));
} }
}); });
} }
@ -534,7 +534,7 @@ public abstract class JoinAbstractWrapper<T, Children extends JoinAbstractWrappe
return maybeDo(condition, () -> { return maybeDo(condition, () -> {
if (CollectionUtils.isNotEmpty(columns)) { if (CollectionUtils.isNotEmpty(columns)) {
columns.forEach(c -> appendSqlSegments(ORDER_BY, columns.forEach(c -> appendSqlSegments(ORDER_BY,
columnToSqlSegment(index, alias, columnSqlInjectFilter(c)), DESC)); columnToSqlSegment(index, alias, columnSqlInjectFilter(c), true), DESC));
} }
}); });
} }
@ -577,10 +577,10 @@ public abstract class JoinAbstractWrapper<T, Children extends JoinAbstractWrappe
public final <X> Children orderBy(boolean condition, boolean isAsc, String alias, SFunction<X, ?> column, SFunction<X, ?>... columns) { public final <X> Children orderBy(boolean condition, boolean isAsc, String alias, SFunction<X, ?> column, SFunction<X, ?>... columns) {
return maybeDo(condition, () -> { return maybeDo(condition, () -> {
final SqlKeyword mode = isAsc ? ASC : DESC; final SqlKeyword mode = isAsc ? ASC : DESC;
appendSqlSegments(ORDER_BY, columnToSqlSegment(index, alias, column), mode); appendSqlSegments(ORDER_BY, columnToSqlSegment(index, alias, column, true), mode);
if (ArrayUtils.isNotEmpty(columns)) { if (ArrayUtils.isNotEmpty(columns)) {
Arrays.stream(columns).forEach(c -> appendSqlSegments(ORDER_BY, Arrays.stream(columns).forEach(c -> appendSqlSegments(ORDER_BY,
columnToSqlSegment(index, alias, columnSqlInjectFilter(c)), mode)); columnToSqlSegment(index, alias, columnSqlInjectFilter(c), true), mode));
} }
}); });
} }
@ -632,7 +632,7 @@ public abstract class JoinAbstractWrapper<T, Children extends JoinAbstractWrappe
* <p>拼接 LIKE 以及 </p> * <p>拼接 LIKE 以及 </p>
*/ */
protected <X> Children likeValue(boolean condition, SqlKeyword keyword, String alias, SFunction<X, ?> column, Object val, SqlLike sqlLike) { protected <X> Children likeValue(boolean condition, SqlKeyword keyword, String alias, SFunction<X, ?> column, Object val, SqlLike sqlLike) {
return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column), keyword, return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column, false), keyword,
() -> formatParam(null, SqlUtils.concatLike(val, sqlLike)))); () -> formatParam(null, SqlUtils.concatLike(val, sqlLike))));
} }
@ -650,7 +650,7 @@ public abstract class JoinAbstractWrapper<T, Children extends JoinAbstractWrappe
* @param val 条件值 * @param val 条件值
*/ */
protected <X> Children addCondition(boolean condition, String alias, SFunction<X, ?> column, SqlKeyword sqlKeyword, Object val) { protected <X> Children addCondition(boolean condition, String alias, SFunction<X, ?> column, SqlKeyword sqlKeyword, Object val) {
return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column), sqlKeyword, return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column, false), sqlKeyword,
() -> formatParam(null, val))); () -> formatParam(null, val)));
} }
@ -658,7 +658,7 @@ public abstract class JoinAbstractWrapper<T, Children extends JoinAbstractWrappe
SqlKeyword sqlKeyword, String rightAlias, SFunction<S, ?> val) { SqlKeyword sqlKeyword, String rightAlias, SFunction<S, ?> val) {
Class<X> c = LambdaUtils.getEntityClass(column); Class<X> c = LambdaUtils.getEntityClass(column);
Class<S> v = LambdaUtils.getEntityClass(val); Class<S> v = LambdaUtils.getEntityClass(val);
return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column), sqlKeyword, return maybeDo(condition, () -> appendSqlSegments(columnToSqlSegment(index, alias, column, false), sqlKeyword,
isOn ? columnToSqlSegmentS(index, rightAlias, val, v == c && v == joinClass) : isOn ? columnToSqlSegmentS(index, rightAlias, val, v == c && v == joinClass) :
columnToSqlSegmentS(index, rightAlias, val, v == c) columnToSqlSegmentS(index, rightAlias, val, v == c)
)); ));
@ -844,8 +844,8 @@ public abstract class JoinAbstractWrapper<T, Children extends JoinAbstractWrappe
/** /**
* 获取 columnName * 获取 columnName
*/ */
protected final <X> ISqlSegment columnToSqlSegment(Integer index, String alias, SFunction<X, ?> column) { protected final <X> ISqlSegment columnToSqlSegment(Integer index, String alias, SFunction<X, ?> column, boolean orderBy) {
return () -> columnToString(index, alias, column, false, isOn ? PrefixEnum.ON_FIRST : PrefixEnum.CD_FIRST); return () -> columnToString(index, alias, column, false, isOn ? PrefixEnum.ON_FIRST : PrefixEnum.CD_FIRST, orderBy);
} }
protected final <X> ISqlSegment columnToSqlSegmentS(Integer index, String alias, SFunction<X, ?> column, boolean isJoin) { protected final <X> ISqlSegment columnToSqlSegmentS(Integer index, String alias, SFunction<X, ?> column, boolean isJoin) {
@ -855,7 +855,7 @@ public abstract class JoinAbstractWrapper<T, Children extends JoinAbstractWrappe
} else { } else {
prefixEnum = isOn ? PrefixEnum.ON_SECOND : PrefixEnum.CD_ON_SECOND; prefixEnum = isOn ? PrefixEnum.ON_SECOND : PrefixEnum.CD_ON_SECOND;
} }
return () -> columnToString(index, alias, column, isJoin, prefixEnum); return () -> columnToString(index, alias, column, isJoin, prefixEnum, false);
} }
protected final ISqlSegment columnToSqlSegment(String column) { protected final ISqlSegment columnToSqlSegment(String column) {
@ -865,7 +865,7 @@ public abstract class JoinAbstractWrapper<T, Children extends JoinAbstractWrappe
/** /**
* 获取 columnName * 获取 columnName
*/ */
protected <X> String columnToString(Integer index, String alias, X column, boolean isJoin, PrefixEnum prefixEnum) { protected <X> String columnToString(Integer index, String alias, X column, boolean isJoin, PrefixEnum prefixEnum, boolean orderBy) {
return (String) column; return (String) column;
} }
@ -890,10 +890,12 @@ public abstract class JoinAbstractWrapper<T, Children extends JoinAbstractWrappe
* *
* @param columns 多字段 * @param columns 多字段
*/ */
abstract <X> String columnsToString(Integer index, PrefixEnum prefixEnum, String alias, X... columns); abstract <X> String columnsToString(Integer index, PrefixEnum prefixEnum, String alias, boolean orderBy, X... columns);
public <X> String columnsToString(Integer index, PrefixEnum prefixEnum, String alias, List<X> columns) { public <X> String columnsToString(Integer index, PrefixEnum prefixEnum, String alias, boolean orderBy, List<X> columns) {
return columns.stream().map(i -> columnToString(index, alias, (SFunction<?, ?>) i, false, prefixEnum)).collect(joining(StringPool.COMMA)); return columns.stream().map(i ->
columnToString(index, alias, (SFunction<?, ?>) i, false, prefixEnum, orderBy))
.collect(joining(StringPool.COMMA));
} }
@Override @Override

View File

@ -56,7 +56,7 @@ public class MybatisPlusConfig {
@Override @Override
public boolean ignoreTable(String tableName) { public boolean ignoreTable(String tableName) {
return !"user_tenant".equals(tableName); return !tableName.startsWith("user_tenant");
} }
})); }));
interceptor.addInnerInterceptor(page); interceptor.addInnerInterceptor(page);

View File

@ -1,6 +1,8 @@
package com.github.yulichang.test.join.m; package com.github.yulichang.test.join.m;
import com.baomidou.mybatisplus.core.toolkit.support.SFunction; import com.baomidou.mybatisplus.core.toolkit.support.SFunction;
import com.github.yulichang.test.join.dto.UserDTO;
import com.github.yulichang.test.join.entity.AddressDO;
import com.github.yulichang.test.join.entity.UserDO; import com.github.yulichang.test.join.entity.UserDO;
import com.github.yulichang.test.util.Reset; import com.github.yulichang.test.util.Reset;
import com.github.yulichang.test.util.ThreadLocalUtils; import com.github.yulichang.test.util.ThreadLocalUtils;
@ -55,4 +57,18 @@ public class OrderByTest {
list.forEach(System.out::println); list.forEach(System.out::println);
} }
@Test
void orderBy2() {
MPJLambdaWrapper<UserDO> wrapper = JoinWrappers.lambda(UserDO.class)
.selectAll(UserDO.class)
.selectAs(AddressDO::getId, UserDTO::getAddress)
.leftJoin(AddressDO.class, AddressDO::getUserId, UserDO::getId)
.orderByAsc(UserDTO::getAddress);
List<UserDO> list = wrapper.list();
list.forEach(System.out::println);
}
} }