mirror of
https://gitee.com/best_handsome/mybatis-plus-join
synced 2025-07-11 00:02:22 +08:00
连表查询支持聚合函数查询
This commit is contained in:
parent
3fcd8cc869
commit
52f947b477
@ -25,8 +25,6 @@ import org.apache.ibatis.type.TypeHandler;
|
|||||||
import org.apache.ibatis.type.TypeHandlerRegistry;
|
import org.apache.ibatis.type.TypeHandlerRegistry;
|
||||||
import org.apache.ibatis.type.UnknownTypeHandler;
|
import org.apache.ibatis.type.UnknownTypeHandler;
|
||||||
|
|
||||||
import java.lang.reflect.InvocationTargetException;
|
|
||||||
import java.lang.reflect.Method;
|
|
||||||
import java.util.*;
|
import java.util.*;
|
||||||
import java.util.concurrent.ConcurrentHashMap;
|
import java.util.concurrent.ConcurrentHashMap;
|
||||||
|
|
||||||
@ -110,17 +108,7 @@ public class MPJInterceptor implements Interceptor {
|
|||||||
*/
|
*/
|
||||||
private ResultMap newResultMap(MappedStatement ms, Class<?> resultType) {
|
private ResultMap newResultMap(MappedStatement ms, Class<?> resultType) {
|
||||||
TableInfo tableInfo = TableInfoHelper.getTableInfo(resultType);
|
TableInfo tableInfo = TableInfoHelper.getTableInfo(resultType);
|
||||||
if (tableInfo != null && tableInfo.isAutoInitResultMap()) {
|
if (tableInfo != null && tableInfo.isAutoInitResultMap() && tableInfo.getEntityType() == resultType) {
|
||||||
if (tableInfo.getEntityType() != resultType) {
|
|
||||||
try {
|
|
||||||
Method info = TableInfoHelper.class.getDeclaredMethod("initTableInfo", Configuration.class, String.class, Class.class);
|
|
||||||
info.setAccessible(true);
|
|
||||||
Object invoke = info.invoke(TableInfoHelper.class, ms.getConfiguration(), ms.getId().substring(0, ms.getId().lastIndexOf(".")), resultType);
|
|
||||||
tableInfo = (TableInfo) invoke;
|
|
||||||
} catch (NoSuchMethodException | InvocationTargetException | IllegalAccessException e) {
|
|
||||||
e.printStackTrace();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return initResultMapIfNeed(tableInfo, resultType);
|
return initResultMapIfNeed(tableInfo, resultType);
|
||||||
}
|
}
|
||||||
return new ResultMap.Builder(ms.getConfiguration(), ms.getId(), resultType, EMPTY_RESULT_MAPPING).build();
|
return new ResultMap.Builder(ms.getConfiguration(), ms.getId(), resultType, EMPTY_RESULT_MAPPING).build();
|
||||||
|
@ -0,0 +1,251 @@
|
|||||||
|
package com.github.yulichang.query;
|
||||||
|
|
||||||
|
import com.baomidou.mybatisplus.core.conditions.AbstractLambdaWrapper;
|
||||||
|
import com.baomidou.mybatisplus.core.conditions.SharedString;
|
||||||
|
import com.baomidou.mybatisplus.core.conditions.query.Query;
|
||||||
|
import com.baomidou.mybatisplus.core.conditions.segments.MergeSegments;
|
||||||
|
import com.baomidou.mybatisplus.core.metadata.TableFieldInfo;
|
||||||
|
import com.baomidou.mybatisplus.core.metadata.TableInfo;
|
||||||
|
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
|
||||||
|
import com.baomidou.mybatisplus.core.toolkit.ArrayUtils;
|
||||||
|
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
||||||
|
import com.baomidou.mybatisplus.core.toolkit.StringPool;
|
||||||
|
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
|
||||||
|
import com.baomidou.mybatisplus.core.toolkit.support.SFunction;
|
||||||
|
import com.github.yulichang.query.interfaces.MPJJoin;
|
||||||
|
import com.github.yulichang.toolkit.Constant;
|
||||||
|
import org.springframework.util.Assert;
|
||||||
|
|
||||||
|
import java.util.*;
|
||||||
|
import java.util.concurrent.atomic.AtomicInteger;
|
||||||
|
import java.util.function.Predicate;
|
||||||
|
import java.util.stream.Collectors;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 不推荐使用这wrapper
|
||||||
|
* 这种既有string又有lambda对开发人员并不友好
|
||||||
|
* <p>
|
||||||
|
* 推荐使用以下两个类 :
|
||||||
|
* String -> {@link MPJQueryWrapper<T>}
|
||||||
|
* lambda -> {@link com.github.yulichang.wrapper.MPJLambdaWrapper<T>}
|
||||||
|
* <p>
|
||||||
|
*
|
||||||
|
* @author yulichang
|
||||||
|
*/
|
||||||
|
@Deprecated
|
||||||
|
@SuppressWarnings("DeprecatedIsStillUsed")
|
||||||
|
public class MPJLambdaQueryWrapper<T> extends AbstractLambdaWrapper<T, MPJLambdaQueryWrapper<T>>
|
||||||
|
implements Query<MPJLambdaQueryWrapper<T>, T, SFunction<T, ?>>, MPJJoin<MPJLambdaQueryWrapper<T>> {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 查询字段
|
||||||
|
*/
|
||||||
|
private SharedString sqlSelect = new SharedString();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 连表字段
|
||||||
|
*/
|
||||||
|
private SharedString from = SharedString.emptyString();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 主表别名
|
||||||
|
*/
|
||||||
|
private final SharedString alias = new SharedString(Constant.TABLE_ALIAS);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 查询的列
|
||||||
|
*/
|
||||||
|
private List<String> selectColumns = new ArrayList<>();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 排除的字段
|
||||||
|
*/
|
||||||
|
private List<String> ignoreColumns = new ArrayList<>();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 不建议直接 new 该实例,使用 Wrappers.lambdaQuery(entity)
|
||||||
|
*/
|
||||||
|
public MPJLambdaQueryWrapper() {
|
||||||
|
super.initNeed();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 不建议直接 new 该实例,使用 Wrappers.lambdaQuery(...)
|
||||||
|
*/
|
||||||
|
MPJLambdaQueryWrapper(T entity, Class<T> entityClass, SharedString from, SharedString sqlSelect, AtomicInteger paramNameSeq,
|
||||||
|
Map<String, Object> paramNameValuePairs, MergeSegments mergeSegments,
|
||||||
|
SharedString lastSql, SharedString sqlComment, SharedString sqlFirst,
|
||||||
|
List<String> selectColumns, List<String> ignoreColumns) {
|
||||||
|
super.setEntity(entity);
|
||||||
|
super.setEntityClass(entityClass);
|
||||||
|
this.paramNameSeq = paramNameSeq;
|
||||||
|
this.paramNameValuePairs = paramNameValuePairs;
|
||||||
|
this.expression = mergeSegments;
|
||||||
|
this.sqlSelect = sqlSelect;
|
||||||
|
this.from = from;
|
||||||
|
this.lastSql = lastSql;
|
||||||
|
this.sqlComment = sqlComment;
|
||||||
|
this.sqlFirst = sqlFirst;
|
||||||
|
this.selectColumns = selectColumns;
|
||||||
|
this.ignoreColumns = ignoreColumns;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* SELECT 部分 SQL 设置
|
||||||
|
*
|
||||||
|
* @param columns 查询字段
|
||||||
|
*/
|
||||||
|
@SafeVarargs
|
||||||
|
public final MPJLambdaQueryWrapper<T> select(SFunction<T, ?>... columns) {
|
||||||
|
if (ArrayUtils.isNotEmpty(columns)) {
|
||||||
|
for (SFunction<T, ?> s : columns) {
|
||||||
|
selectColumns.add(columnToString(s, false));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return typedThis;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 忽略查询字段
|
||||||
|
* <p>
|
||||||
|
* 用法: selectIgnore("t.id","t.sex","a.area")
|
||||||
|
*
|
||||||
|
* @since 1.1.3
|
||||||
|
*/
|
||||||
|
public MPJLambdaQueryWrapper<T> selectIgnore(String... columns) {
|
||||||
|
if (ArrayUtils.isNotEmpty(columns)) {
|
||||||
|
ignoreColumns.addAll(Arrays.asList(columns));
|
||||||
|
}
|
||||||
|
return typedThis;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 忽略查询字段
|
||||||
|
* <p>
|
||||||
|
* 用法: selectIgnore("t.id","t.sex","a.area")
|
||||||
|
*
|
||||||
|
* @since 1.1.3
|
||||||
|
*/
|
||||||
|
@SafeVarargs
|
||||||
|
public final MPJLambdaQueryWrapper<T> selectIgnore(SFunction<T, ?>... columns) {
|
||||||
|
if (ArrayUtils.isNotEmpty(columns)) {
|
||||||
|
for (SFunction<T, ?> s : columns) {
|
||||||
|
ignoreColumns.add(Constant.TABLE_ALIAS + StringPool.DOT + getColumnCache(s).getColumn());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return typedThis;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected String columnToString(SFunction<T, ?> column, boolean onlyColumn) {
|
||||||
|
return Constant.TABLE_ALIAS + StringPool.DOT + super.columnToString(column, onlyColumn);
|
||||||
|
}
|
||||||
|
|
||||||
|
public MPJLambdaQueryWrapper<T> select(String... columns) {
|
||||||
|
if (ArrayUtils.isNotEmpty(columns)) {
|
||||||
|
Collections.addAll(selectColumns, columns);
|
||||||
|
}
|
||||||
|
return typedThis;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 只针对主表
|
||||||
|
* <p>
|
||||||
|
* 过滤查询的字段信息(主键除外!)
|
||||||
|
* <p>例1: 只要 java 字段名以 "test" 开头的 -> select(i -> i.getProperty().startsWith("test"))</p>
|
||||||
|
* <p>例2: 只要 java 字段属性是 CharSequence 类型的 -> select(TableFieldInfo::isCharSequence)</p>
|
||||||
|
* <p>例3: 只要 java 字段没有填充策略的 -> select(i -> i.getFieldFill() == FieldFill.DEFAULT)</p>
|
||||||
|
* <p>例4: 要全部字段 -> select(i -> true)</p>
|
||||||
|
* <p>例5: 只要主键字段 -> select(i -> false)</p>
|
||||||
|
*
|
||||||
|
* @param predicate 过滤方式
|
||||||
|
* @return this
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
public MPJLambdaQueryWrapper<T> select(Class<T> entityClass, Predicate<TableFieldInfo> predicate) {
|
||||||
|
TableInfo info = TableInfoHelper.getTableInfo(entityClass);
|
||||||
|
Assert.notNull(info, "can not find table info");
|
||||||
|
selectColumns.addAll(info.getFieldList().stream().filter(predicate).map(c ->
|
||||||
|
Constant.TABLE_ALIAS + StringPool.DOT + c.getColumn()).collect(Collectors.toList()));
|
||||||
|
return typedThis;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 查询主表全部字段
|
||||||
|
*
|
||||||
|
* @param clazz 主表class
|
||||||
|
*/
|
||||||
|
public final MPJLambdaQueryWrapper<T> selectAll(Class<T> clazz) {
|
||||||
|
return selectAll(clazz, Constant.TABLE_ALIAS);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 查询表全部字段
|
||||||
|
*
|
||||||
|
* @param clazz 表实体
|
||||||
|
* @param as 表别名
|
||||||
|
*/
|
||||||
|
public final MPJLambdaQueryWrapper<T> selectAll(Class<?> clazz, String as) {
|
||||||
|
TableInfo info = TableInfoHelper.getTableInfo(clazz);
|
||||||
|
Assert.notNull(info, "can not find table info");
|
||||||
|
if (info.havePK()) {
|
||||||
|
selectColumns.add(as + StringPool.DOT + info.getKeyColumn());
|
||||||
|
}
|
||||||
|
selectColumns.addAll(info.getFieldList().stream().map(i ->
|
||||||
|
as + StringPool.DOT + i.getColumn()).collect(Collectors.toList()));
|
||||||
|
return typedThis;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 返回一个支持 lambda 函数写法的 wrapper
|
||||||
|
*/
|
||||||
|
public MPJQueryWrapper<T> stringQuery() {
|
||||||
|
return new MPJQueryWrapper<>(getEntity(), getEntityClass(), paramNameSeq, paramNameValuePairs,
|
||||||
|
expression, sqlSelect, from, lastSql, sqlComment, sqlFirst, selectColumns, ignoreColumns);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getSqlSelect() {
|
||||||
|
if (StringUtils.isBlank(sqlSelect.getStringValue())) {
|
||||||
|
if (CollectionUtils.isNotEmpty(ignoreColumns)) {
|
||||||
|
selectColumns.removeIf(ignoreColumns::contains);
|
||||||
|
}
|
||||||
|
sqlSelect.setStringValue(String.join(StringPool.COMMA, selectColumns));
|
||||||
|
}
|
||||||
|
return sqlSelect.getStringValue();
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getFrom() {
|
||||||
|
return from.getStringValue();
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
public String getAlias() {
|
||||||
|
return alias.getStringValue();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 用于生成嵌套 sql
|
||||||
|
* <p>故 sqlSelect selectColumn ignoreColumns from不向下传递</p>
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
protected MPJLambdaQueryWrapper<T> instance() {
|
||||||
|
return new MPJLambdaQueryWrapper<>(getEntity(), getEntityClass(), null, null, paramNameSeq, paramNameValuePairs,
|
||||||
|
new MergeSegments(), SharedString.emptyString(), SharedString.emptyString(), SharedString.emptyString(), null, null);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void clear() {
|
||||||
|
super.clear();
|
||||||
|
sqlSelect.toNull();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public MPJLambdaQueryWrapper<T> join(String keyWord, boolean condition, String joinSql) {
|
||||||
|
if (condition) {
|
||||||
|
from.setStringValue(from.getStringValue() + keyWord + joinSql);
|
||||||
|
}
|
||||||
|
return typedThis;
|
||||||
|
}
|
||||||
|
}
|
@ -172,6 +172,13 @@ public class MPJQueryWrapper<T> extends AbstractWrapper<T, String, MPJQueryWrapp
|
|||||||
return alias.getStringValue();
|
return alias.getStringValue();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 返回一个支持 lambda 函数写法的 wrapper
|
||||||
|
*/
|
||||||
|
public MPJLambdaQueryWrapper<T> lambda() {
|
||||||
|
return new MPJLambdaQueryWrapper<>(getEntity(), getEntityClass(), from, sqlSelect, paramNameSeq, paramNameValuePairs,
|
||||||
|
expression, lastSql, sqlComment, sqlFirst, selectColumns, ignoreColumns);
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 用于生成嵌套 sql
|
* 用于生成嵌套 sql
|
||||||
|
@ -14,7 +14,9 @@ import java.util.Objects;
|
|||||||
import static java.util.stream.Collectors.joining;
|
import static java.util.stream.Collectors.joining;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* copy {@link com.baomidou.mybatisplus.core.conditions.AbstractLambdaWrapper}
|
* 参考 {@link com.baomidou.mybatisplus.core.conditions.AbstractLambdaWrapper}
|
||||||
|
*
|
||||||
|
* @author yulichang
|
||||||
*/
|
*/
|
||||||
@SuppressWarnings("serial")
|
@SuppressWarnings("serial")
|
||||||
public abstract class MPJAbstractLambdaWrapper<T, Children extends MPJAbstractLambdaWrapper<T, Children>>
|
public abstract class MPJAbstractLambdaWrapper<T, Children extends MPJAbstractLambdaWrapper<T, Children>>
|
||||||
@ -49,10 +51,10 @@ public abstract class MPJAbstractLambdaWrapper<T, Children extends MPJAbstractLa
|
|||||||
Class<?> aClass = LambdaUtils.getEntityClass(fn);
|
Class<?> aClass = LambdaUtils.getEntityClass(fn);
|
||||||
Map<String, ColumnCache> cacheMap = columnMap.get(aClass);
|
Map<String, ColumnCache> cacheMap = columnMap.get(aClass);
|
||||||
if (cacheMap == null) {
|
if (cacheMap == null) {
|
||||||
cacheMap = com.baomidou.mybatisplus.core.toolkit.LambdaUtils.getColumnMap(aClass);
|
cacheMap = LambdaUtils.getColumnMap(aClass);
|
||||||
columnMap.put(aClass, cacheMap);
|
columnMap.put(aClass, cacheMap);
|
||||||
}
|
}
|
||||||
return cacheMap.get(com.baomidou.mybatisplus.core.toolkit.LambdaUtils.formatKey(LambdaUtils.getName(fn)));
|
return cacheMap.get(LambdaUtils.formatKey(LambdaUtils.getName(fn)));
|
||||||
}
|
}
|
||||||
|
|
||||||
protected String getDefault(Integer i) {
|
protected String getDefault(Integer i) {
|
||||||
|
@ -8,8 +8,10 @@ import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
|
|||||||
import com.baomidou.mybatisplus.core.toolkit.*;
|
import com.baomidou.mybatisplus.core.toolkit.*;
|
||||||
import com.baomidou.mybatisplus.core.toolkit.support.SFunction;
|
import com.baomidou.mybatisplus.core.toolkit.support.SFunction;
|
||||||
import com.github.yulichang.toolkit.Constant;
|
import com.github.yulichang.toolkit.Constant;
|
||||||
|
import com.github.yulichang.toolkit.LambdaUtils;
|
||||||
|
import com.github.yulichang.wrapper.enums.BaseFuncEnum;
|
||||||
import com.github.yulichang.wrapper.interfaces.LambdaJoin;
|
import com.github.yulichang.wrapper.interfaces.LambdaJoin;
|
||||||
import com.github.yulichang.wrapper.interfaces.SFunctionQuery;
|
import com.github.yulichang.wrapper.interfaces.Query;
|
||||||
import com.github.yulichang.wrapper.interfaces.on.OnFunction;
|
import com.github.yulichang.wrapper.interfaces.on.OnFunction;
|
||||||
import lombok.Data;
|
import lombok.Data;
|
||||||
import lombok.Getter;
|
import lombok.Getter;
|
||||||
@ -33,7 +35,7 @@ import java.util.stream.Collectors;
|
|||||||
*/
|
*/
|
||||||
@SuppressWarnings("all")
|
@SuppressWarnings("all")
|
||||||
public class MPJLambdaWrapper<T> extends MPJAbstractLambdaWrapper<T, MPJLambdaWrapper<T>>
|
public class MPJLambdaWrapper<T> extends MPJAbstractLambdaWrapper<T, MPJLambdaWrapper<T>>
|
||||||
implements SFunctionQuery<MPJLambdaWrapper<T>>, LambdaJoin<MPJLambdaWrapper<T>> {
|
implements Query<MPJLambdaWrapper<T>>, LambdaJoin<MPJLambdaWrapper<T>> {
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 查询字段 sql
|
* 查询字段 sql
|
||||||
@ -116,33 +118,40 @@ public class MPJLambdaWrapper<T> extends MPJAbstractLambdaWrapper<T, MPJLambdaWr
|
|||||||
*
|
*
|
||||||
* @param columns 查询字段
|
* @param columns 查询字段
|
||||||
*/
|
*/
|
||||||
|
@Override
|
||||||
@SafeVarargs
|
@SafeVarargs
|
||||||
public final <S> MPJLambdaWrapper<T> select(SFunction<S, ?>... columns) {
|
public final <S> MPJLambdaWrapper<T> select(SFunction<S, ?>... columns) {
|
||||||
if (ArrayUtils.isNotEmpty(columns)) {
|
if (ArrayUtils.isNotEmpty(columns)) {
|
||||||
for (SFunction<S, ?> s : columns) {
|
for (SFunction<S, ?> s : columns) {
|
||||||
selectColumns.add(new SelectColumn(com.github.yulichang.toolkit.LambdaUtils.getEntityClass(s), getCache(s).getColumn(), null));
|
selectColumns.add(SelectColumn.of(LambdaUtils.getEntityClass(s), getCache(s).getColumn()));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return typedThis;
|
return typedThis;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
public <E> MPJLambdaWrapper<T> select(Class<E> entityClass, Predicate<TableFieldInfo> predicate) {
|
public <E> MPJLambdaWrapper<T> select(Class<E> entityClass, Predicate<TableFieldInfo> predicate) {
|
||||||
TableInfo info = TableInfoHelper.getTableInfo(entityClass);
|
TableInfo info = TableInfoHelper.getTableInfo(entityClass);
|
||||||
Assert.notNull(info, "table can not be find");
|
Assert.notNull(info, "table can not be find");
|
||||||
info.getFieldList().stream().filter(predicate).collect(Collectors.toList()).forEach(
|
info.getFieldList().stream().filter(predicate).collect(Collectors.toList()).forEach(
|
||||||
i -> selectColumns.add(new SelectColumn(entityClass, i.getColumn(), null)));
|
i -> selectColumns.add(SelectColumn.of(entityClass, i.getColumn())));
|
||||||
return typedThis;
|
return typedThis;
|
||||||
}
|
}
|
||||||
|
|
||||||
public final <S, X> MPJLambdaWrapper<T> selectAs(SFunction<S, ?> columns, SFunction<X, ?> alias) {
|
@Override
|
||||||
return selectAs(columns, com.github.yulichang.toolkit.LambdaUtils.getName(alias));
|
public <S> MPJLambdaWrapper<T> selectAs(SFunction<S, ?> column, String alias) {
|
||||||
|
selectColumns.add(SelectColumn.of(LambdaUtils.getEntityClass(column), getCache(column).getColumn(), alias));
|
||||||
|
return typedThis;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
public <S> MPJLambdaWrapper<T> selectFunc(BaseFuncEnum funcEnum, SFunction<S, ?> column, String alias) {
|
||||||
* @since 1.1.3
|
selectColumns.add(SelectColumn.of(LambdaUtils.getEntityClass(column), getCache(column).getColumn(), alias, funcEnum));
|
||||||
*/
|
return typedThis;
|
||||||
public final <S, X> MPJLambdaWrapper<T> selectAs(SFunction<S, ?> columns, String alias) {
|
}
|
||||||
selectColumns.add(new SelectColumn(com.github.yulichang.toolkit.LambdaUtils.getEntityClass(columns), getCache(columns).getColumn(), alias));
|
|
||||||
|
@Override
|
||||||
|
public MPJLambdaWrapper<T> selectFunc(BaseFuncEnum funcEnum, Object column, String alias) {
|
||||||
|
selectColumns.add(SelectColumn.of(null, column.toString(), alias, funcEnum));
|
||||||
return typedThis;
|
return typedThis;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -150,10 +159,10 @@ public class MPJLambdaWrapper<T> extends MPJAbstractLambdaWrapper<T, MPJLambdaWr
|
|||||||
TableInfo info = TableInfoHelper.getTableInfo(clazz);
|
TableInfo info = TableInfoHelper.getTableInfo(clazz);
|
||||||
Assert.notNull(info, "table can not be find -> %s", clazz);
|
Assert.notNull(info, "table can not be find -> %s", clazz);
|
||||||
if (info.havePK()) {
|
if (info.havePK()) {
|
||||||
selectColumns.add(new SelectColumn(clazz, info.getKeyColumn(), null));
|
selectColumns.add(SelectColumn.of(clazz, info.getKeyColumn()));
|
||||||
}
|
}
|
||||||
info.getFieldList().forEach(c ->
|
info.getFieldList().forEach(c ->
|
||||||
selectColumns.add(new SelectColumn(clazz, c.getColumn(), null)));
|
selectColumns.add(SelectColumn.of(clazz, c.getColumn())));
|
||||||
return typedThis;
|
return typedThis;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -172,29 +181,35 @@ public class MPJLambdaWrapper<T> extends MPJAbstractLambdaWrapper<T, MPJLambdaWr
|
|||||||
public final <S> MPJLambdaWrapper<T> selectIgnore(SFunction<S, ?>... columns) {
|
public final <S> MPJLambdaWrapper<T> selectIgnore(SFunction<S, ?>... columns) {
|
||||||
if (ArrayUtils.isNotEmpty(columns)) {
|
if (ArrayUtils.isNotEmpty(columns)) {
|
||||||
for (SFunction<S, ?> s : columns) {
|
for (SFunction<S, ?> s : columns) {
|
||||||
ignoreColumns.add(new SelectColumn(com.github.yulichang.toolkit.LambdaUtils.getEntityClass(s), getCache(s).getColumn(), null));
|
ignoreColumns.add(SelectColumn.of(LambdaUtils.getEntityClass(s), getCache(s).getColumn()));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return typedThis;
|
return typedThis;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 查询条件 SQL 片段
|
||||||
|
*/
|
||||||
@Override
|
@Override
|
||||||
public String getSqlSelect() {
|
public String getSqlSelect() {
|
||||||
if (StringUtils.isBlank(sqlSelect.getStringValue())) {
|
if (StringUtils.isBlank(sqlSelect.getStringValue())) {
|
||||||
if (CollectionUtils.isNotEmpty(ignoreColumns)) {
|
if (CollectionUtils.isNotEmpty(ignoreColumns)) {
|
||||||
selectColumns.removeIf(c -> ignoreColumns.stream().anyMatch(i ->
|
selectColumns.removeIf(c -> c.getFuncEnum() == null && ignoreColumns.stream().anyMatch(i ->
|
||||||
i.getClazz() == c.getClazz() && Objects.equals(c.getColumnName(), i.getColumnName())));
|
i.getClazz() == c.getClazz() && Objects.equals(c.getColumnName(), i.getColumnName())));
|
||||||
}
|
}
|
||||||
String s = selectColumns.stream().map(i ->
|
String s = selectColumns.stream().map(i -> {
|
||||||
Constant.TABLE_ALIAS + getDefault(subTable.get(i.getClazz())) + StringPool.DOT + i.getColumnName() +
|
String str = Constant.TABLE_ALIAS + getDefault(subTable.get(i.getClazz())) + StringPool.DOT + i.getColumnName();
|
||||||
(StringUtils.isBlank(i.getAlias()) ? StringPool.EMPTY : (Constant.AS + i.getAlias())))
|
return (i.getFuncEnum() == null ? str : String.format(i.getFuncEnum().getSql(), str)) +
|
||||||
.collect(Collectors.joining(StringPool.COMMA));
|
(StringUtils.isBlank(i.getAlias()) ? StringPool.EMPTY : (Constant.AS + i.getAlias()));
|
||||||
|
}).collect(Collectors.joining(StringPool.COMMA));
|
||||||
sqlSelect.setStringValue(s);
|
sqlSelect.setStringValue(s);
|
||||||
}
|
}
|
||||||
return sqlSelect.getStringValue();
|
return sqlSelect.getStringValue();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取连表部分语句
|
||||||
|
*/
|
||||||
public String getFrom() {
|
public String getFrom() {
|
||||||
if (StringUtils.isBlank(from.getStringValue())) {
|
if (StringUtils.isBlank(from.getStringValue())) {
|
||||||
StringBuilder value = new StringBuilder();
|
StringBuilder value = new StringBuilder();
|
||||||
@ -264,10 +279,25 @@ public class MPJLambdaWrapper<T> extends MPJAbstractLambdaWrapper<T, MPJLambdaWr
|
|||||||
|
|
||||||
private String alias;
|
private String alias;
|
||||||
|
|
||||||
public SelectColumn(Class<?> clazz, String columnName, String alias) {
|
private BaseFuncEnum funcEnum;
|
||||||
|
|
||||||
|
private SelectColumn(Class<?> clazz, String columnName, String alias, BaseFuncEnum funcEnum) {
|
||||||
this.clazz = clazz;
|
this.clazz = clazz;
|
||||||
this.columnName = columnName;
|
this.columnName = columnName;
|
||||||
this.alias = alias;
|
this.alias = alias;
|
||||||
|
this.funcEnum = funcEnum;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static SelectColumn of(Class<?> clazz, String columnName) {
|
||||||
|
return of(clazz, columnName, null, null);
|
||||||
|
}
|
||||||
|
|
||||||
|
public static SelectColumn of(Class<?> clazz, String columnName, String alias) {
|
||||||
|
return of(clazz, columnName, alias, null);
|
||||||
|
}
|
||||||
|
|
||||||
|
public static SelectColumn of(Class<?> clazz, String columnName, String alias, BaseFuncEnum funcEnum) {
|
||||||
|
return new SelectColumn(clazz, columnName, alias, funcEnum);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -0,0 +1,11 @@
|
|||||||
|
package com.github.yulichang.wrapper.enums;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 函数枚举基类
|
||||||
|
*
|
||||||
|
* @author yulichang
|
||||||
|
*/
|
||||||
|
public interface BaseFuncEnum {
|
||||||
|
|
||||||
|
String getSql();
|
||||||
|
}
|
@ -0,0 +1,32 @@
|
|||||||
|
package com.github.yulichang.wrapper.enums;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 常用的sql函数枚举 默认实现
|
||||||
|
* 可以自己实现 {@link BaseFuncEnum} 自定义函数
|
||||||
|
* 目前支持一个占位符,不支持多个%s
|
||||||
|
* <p>
|
||||||
|
* 只例举几个通用的,其他函数 first() last() len() ucase() lcase() 等 或者数据库自定义函数请自行扩展
|
||||||
|
*
|
||||||
|
* @author yulichang
|
||||||
|
*/
|
||||||
|
public enum DefaultFuncEnum implements BaseFuncEnum {
|
||||||
|
|
||||||
|
SUM("SUM(%s)"),
|
||||||
|
COUNT("COUNT(%s)"),
|
||||||
|
MAX("MAX(%s)"),
|
||||||
|
MIN("MIN(%s)"),
|
||||||
|
AVG("AVG(%s)"),
|
||||||
|
LEN("LEN(%s)");
|
||||||
|
|
||||||
|
private final String sql;
|
||||||
|
|
||||||
|
DefaultFuncEnum(String sql) {
|
||||||
|
this.sql = sql;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getSql() {
|
||||||
|
return this.sql;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
@ -0,0 +1,78 @@
|
|||||||
|
package com.github.yulichang.wrapper.interfaces;
|
||||||
|
|
||||||
|
import com.baomidou.mybatisplus.core.metadata.TableFieldInfo;
|
||||||
|
import com.baomidou.mybatisplus.core.toolkit.support.SFunction;
|
||||||
|
import com.github.yulichang.toolkit.LambdaUtils;
|
||||||
|
import com.github.yulichang.wrapper.enums.BaseFuncEnum;
|
||||||
|
|
||||||
|
import java.io.Serializable;
|
||||||
|
import java.util.function.Predicate;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 参考 {@link com.baomidou.mybatisplus.core.conditions.query.Query}
|
||||||
|
*
|
||||||
|
* @author yulichang
|
||||||
|
*/
|
||||||
|
public interface Query<Children> extends Serializable {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 设置查询字段
|
||||||
|
*
|
||||||
|
* @param columns 字段数组
|
||||||
|
* @return children
|
||||||
|
*/
|
||||||
|
@SuppressWarnings("unchecked")
|
||||||
|
<E> Children select(SFunction<E, ?>... columns);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 过滤查询的字段信息(主键除外!)
|
||||||
|
* <p>例1: 只要 java 字段名以 "test" 开头的 -> select(i -> i.getProperty().startsWith("test"))</p>
|
||||||
|
* <p>例2: 只要 java 字段属性是 CharSequence 类型的 -> select(TableFieldInfo::isCharSequence)</p>
|
||||||
|
* <p>例3: 只要 java 字段没有填充策略的 -> select(i -> i.getFieldFill() == FieldFill.DEFAULT)</p>
|
||||||
|
* <p>例4: 要全部字段 -> select(i -> true)</p>
|
||||||
|
* <p>例5: 只要主键字段 -> select(i -> false)</p>
|
||||||
|
*
|
||||||
|
* @param predicate 过滤方式
|
||||||
|
* @return children
|
||||||
|
*/
|
||||||
|
<E> Children select(Class<E> entityClass, Predicate<TableFieldInfo> predicate);
|
||||||
|
|
||||||
|
default <S, X> Children selectAs(SFunction<S, ?> columns, SFunction<X, ?> alias) {
|
||||||
|
return selectAs(columns, LambdaUtils.getName(alias));
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 别名查询
|
||||||
|
*/
|
||||||
|
<S> Children selectAs(SFunction<S, ?> column, String alias);
|
||||||
|
|
||||||
|
|
||||||
|
default <S, X> Children selectFunc(BaseFuncEnum funcEnum, SFunction<S, ?> column) {
|
||||||
|
return selectFunc(funcEnum, column, column);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
default <S, X> Children selectFunc(BaseFuncEnum funcEnum, SFunction<S, ?> column, SFunction<X, ?> alias) {
|
||||||
|
return selectFunc(funcEnum, column, LambdaUtils.getName(alias));
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 聚合函数查询
|
||||||
|
*
|
||||||
|
* @param funcEnum 函数枚举 {@link com.github.yulichang.wrapper.enums.DefaultFuncEnum}
|
||||||
|
* @param column 函数作用的字段
|
||||||
|
* @param alias 别名
|
||||||
|
*/
|
||||||
|
<S> Children selectFunc(BaseFuncEnum funcEnum, SFunction<S, ?> column, String alias);
|
||||||
|
|
||||||
|
default <X> Children selectFunc(BaseFuncEnum funcEnum, Object column, SFunction<X, ?> alias) {
|
||||||
|
return selectFunc(funcEnum, column, LambdaUtils.getName(alias));
|
||||||
|
}
|
||||||
|
|
||||||
|
Children selectFunc(BaseFuncEnum funcEnum, Object column, String alias);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* select sql 片段
|
||||||
|
*/
|
||||||
|
String getSqlSelect();
|
||||||
|
}
|
@ -1,48 +0,0 @@
|
|||||||
package com.github.yulichang.wrapper.interfaces;
|
|
||||||
|
|
||||||
import com.baomidou.mybatisplus.core.metadata.TableFieldInfo;
|
|
||||||
import com.baomidou.mybatisplus.core.toolkit.support.SFunction;
|
|
||||||
|
|
||||||
import java.io.Serializable;
|
|
||||||
import java.util.function.Predicate;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* copy {@link com.baomidou.mybatisplus.core.conditions.query.Query}
|
|
||||||
*/
|
|
||||||
public interface SFunctionQuery<Children> extends Serializable {
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 设置查询字段
|
|
||||||
*
|
|
||||||
* @param columns 字段数组
|
|
||||||
* @return children
|
|
||||||
*/
|
|
||||||
@SuppressWarnings("unchecked")
|
|
||||||
<E> Children select(SFunction<E, ?>... columns);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* ignore
|
|
||||||
* <p>注意只有内部有 entity 才能使用该方法</p>
|
|
||||||
*/
|
|
||||||
default Children select(Predicate<TableFieldInfo> predicate) {
|
|
||||||
return select(null, predicate);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 过滤查询的字段信息(主键除外!)
|
|
||||||
* <p>例1: 只要 java 字段名以 "test" 开头的 -> select(i -> i.getProperty().startsWith("test"))</p>
|
|
||||||
* <p>例2: 只要 java 字段属性是 CharSequence 类型的 -> select(TableFieldInfo::isCharSequence)</p>
|
|
||||||
* <p>例3: 只要 java 字段没有填充策略的 -> select(i -> i.getFieldFill() == FieldFill.DEFAULT)</p>
|
|
||||||
* <p>例4: 要全部字段 -> select(i -> true)</p>
|
|
||||||
* <p>例5: 只要主键字段 -> select(i -> false)</p>
|
|
||||||
*
|
|
||||||
* @param predicate 过滤方式
|
|
||||||
* @return children
|
|
||||||
*/
|
|
||||||
<E>Children select(Class<E> entityClass, Predicate<TableFieldInfo> predicate);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 查询条件 SQL 片段
|
|
||||||
*/
|
|
||||||
String getSqlSelect();
|
|
||||||
}
|
|
Loading…
x
Reference in New Issue
Block a user