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
32c333234b
commit
fd6de6f50b
@ -19,51 +19,51 @@ public interface MPJBaseMapper<T> extends BaseMapper<T> {
|
|||||||
/**
|
/**
|
||||||
* 连表查询返回一条记录
|
* 连表查询返回一条记录
|
||||||
*
|
*
|
||||||
* @param myWrapper joinWrapper
|
* @param wrapper joinWrapper
|
||||||
* @param clazz resultType
|
* @param clazz resultType
|
||||||
*/
|
*/
|
||||||
<DTO> DTO selectJoinOne(@Param(Constant.CLAZZ) Class<DTO> clazz,
|
<DTO> DTO selectJoinOne(@Param(Constant.CLAZZ) Class<DTO> clazz,
|
||||||
@Param(Constants.WRAPPER) MPJBaseJoin myWrapper);
|
@Param(Constants.WRAPPER) MPJBaseJoin wrapper);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 连表查询返回Map
|
* 连表查询返回Map
|
||||||
*
|
*
|
||||||
* @param myWrapper joinWrapper
|
* @param wrapper joinWrapper
|
||||||
*/
|
*/
|
||||||
Map<String, Object> selectJoinMap(@Param(Constants.WRAPPER) MPJBaseJoin myWrapper);
|
Map<String, Object> selectJoinMap(@Param(Constants.WRAPPER) MPJBaseJoin wrapper);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 连表查询返回记录集合
|
* 连表查询返回记录集合
|
||||||
*
|
*
|
||||||
* @param myWrapper joinWrapper
|
* @param wrapper joinWrapper
|
||||||
* @param clazz resultType
|
* @param clazz resultType
|
||||||
*/
|
*/
|
||||||
<DTO> List<DTO> selectJoinList(@Param(Constant.CLAZZ) Class<DTO> clazz,
|
<DTO> List<DTO> selectJoinList(@Param(Constant.CLAZZ) Class<DTO> clazz,
|
||||||
@Param(Constants.WRAPPER) MPJBaseJoin myWrapper);
|
@Param(Constants.WRAPPER) MPJBaseJoin wrapper);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 连表查询返回Map集合
|
* 连表查询返回Map集合
|
||||||
*
|
*
|
||||||
* @param myWrapper joinWrapper
|
* @param wrapper joinWrapper
|
||||||
*/
|
*/
|
||||||
List<Map<String, Object>> selectJoinMaps(@Param(Constants.WRAPPER) MPJBaseJoin myWrapper);
|
List<Map<String, Object>> selectJoinMaps(@Param(Constants.WRAPPER) MPJBaseJoin wrapper);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 连表查询返回记录集合并分页
|
* 连表查询返回记录集合并分页
|
||||||
*
|
*
|
||||||
* @param myWrapper joinWrapper
|
* @param wrapper joinWrapper
|
||||||
* @param clazz resultType
|
* @param clazz resultType
|
||||||
* @param <DTO> 分页返回对象
|
* @param <DTO> 分页返回对象
|
||||||
*/
|
*/
|
||||||
<DTO, P extends IPage<?>> IPage<DTO> selectJoinPage(P page,
|
<DTO, P extends IPage<?>> IPage<DTO> selectJoinPage(P page,
|
||||||
@Param(Constant.CLAZZ) Class<DTO> clazz,
|
@Param(Constant.CLAZZ) Class<DTO> clazz,
|
||||||
@Param(Constants.WRAPPER) MPJBaseJoin myWrapper);
|
@Param(Constants.WRAPPER) MPJBaseJoin wrapper);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 连表查询返回Map集合并分页
|
* 连表查询返回Map集合并分页
|
||||||
*
|
*
|
||||||
* @param myWrapper joinWrapper
|
* @param wrapper joinWrapper
|
||||||
*/
|
*/
|
||||||
<P extends IPage<?>> IPage<Map<String, Object>> selectJoinMapsPage(P page,
|
<P extends IPage<?>> IPage<Map<String, Object>> selectJoinMapsPage(P page,
|
||||||
@Param(Constants.WRAPPER) MPJBaseJoin myWrapper);
|
@Param(Constants.WRAPPER) MPJBaseJoin wrapper);
|
||||||
}
|
}
|
||||||
|
@ -0,0 +1,52 @@
|
|||||||
|
package com.github.yulichang.config;
|
||||||
|
|
||||||
|
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
||||||
|
import com.github.yulichang.exception.MPJException;
|
||||||
|
import com.github.yulichang.interceptor.MPJInterceptor;
|
||||||
|
import org.apache.ibatis.plugin.Interceptor;
|
||||||
|
import org.apache.ibatis.plugin.InterceptorChain;
|
||||||
|
import org.apache.ibatis.session.Configuration;
|
||||||
|
import org.apache.ibatis.session.SqlSessionFactory;
|
||||||
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
import org.springframework.boot.context.event.ApplicationStartedEvent;
|
||||||
|
import org.springframework.context.ApplicationListener;
|
||||||
|
|
||||||
|
import java.lang.reflect.Field;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 拦截器配置类 如果配置了分页插件,可能会使拦截器失效
|
||||||
|
* 此类的作用就是校验拦截器顺序,保证连表插件在其他拦截器之前执行
|
||||||
|
*
|
||||||
|
* @author yulichang
|
||||||
|
*/
|
||||||
|
public class InterceptorConfig implements ApplicationListener<ApplicationStartedEvent> {
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private List<SqlSessionFactory> sqlSessionFactoryList;
|
||||||
|
@Autowired
|
||||||
|
private MPJInterceptor mpjInterceptor;
|
||||||
|
|
||||||
|
@Override
|
||||||
|
@SuppressWarnings("unchecked")
|
||||||
|
public void onApplicationEvent(ApplicationStartedEvent event) {
|
||||||
|
if (CollectionUtils.isNotEmpty(sqlSessionFactoryList)) {
|
||||||
|
try {
|
||||||
|
for (SqlSessionFactory factory : sqlSessionFactoryList) {
|
||||||
|
Field interceptorChain = Configuration.class.getDeclaredField("interceptorChain");
|
||||||
|
interceptorChain.setAccessible(true);
|
||||||
|
InterceptorChain chain = (InterceptorChain) interceptorChain.get(factory.getConfiguration());
|
||||||
|
Field interceptors = InterceptorChain.class.getDeclaredField("interceptors");
|
||||||
|
interceptors.setAccessible(true);
|
||||||
|
List<Interceptor> list = (List<Interceptor>) interceptors.get(chain);
|
||||||
|
if (CollectionUtils.isNotEmpty(list) && list.get(list.size() - 1) != mpjInterceptor) {
|
||||||
|
list.removeIf(i -> i == mpjInterceptor);
|
||||||
|
list.add(mpjInterceptor);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (Exception ignored) {
|
||||||
|
throw new MPJException("mpjInterceptor exception");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,13 @@
|
|||||||
|
package com.github.yulichang.exception;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* mpj 异常
|
||||||
|
*
|
||||||
|
* @author yulichang
|
||||||
|
*/
|
||||||
|
public class MPJException extends RuntimeException {
|
||||||
|
|
||||||
|
public MPJException(String msg) {
|
||||||
|
super(msg);
|
||||||
|
}
|
||||||
|
}
|
@ -1,8 +1,7 @@
|
|||||||
package com.github.yulichang.toolkit;
|
package com.github.yulichang.toolkit;
|
||||||
|
|
||||||
import com.github.yulichang.query.MPJLambdaQueryWrapper;
|
|
||||||
import com.github.yulichang.query.MPJQueryWrapper;
|
import com.github.yulichang.query.MPJQueryWrapper;
|
||||||
import com.github.yulichang.wrapper.MPJJoinLambdaQueryWrapper;
|
import com.github.yulichang.wrapper.MPJLambdaWrapper;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Wrapper 条件构造
|
* Wrapper 条件构造
|
||||||
@ -15,12 +14,7 @@ public class Wrappers {
|
|||||||
return new MPJQueryWrapper<>();
|
return new MPJQueryWrapper<>();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Deprecated
|
public static <T> MPJLambdaWrapper<T> lambdaJoin() {
|
||||||
public static <T> MPJLambdaQueryWrapper<T> lambdaJoin() {
|
return new MPJLambdaWrapper<>();
|
||||||
return new MPJLambdaQueryWrapper<>();
|
|
||||||
}
|
|
||||||
|
|
||||||
public static <T> MPJJoinLambdaQueryWrapper<T> lambdaJoinWrapper() {
|
|
||||||
return new MPJJoinLambdaQueryWrapper<>();
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -1,265 +1,16 @@
|
|||||||
package com.github.yulichang.wrapper;
|
package com.github.yulichang.wrapper;
|
||||||
|
|
||||||
import com.baomidou.mybatisplus.core.conditions.SharedString;
|
|
||||||
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.*;
|
|
||||||
import com.baomidou.mybatisplus.core.toolkit.support.SFunction;
|
|
||||||
import com.github.yulichang.toolkit.Constant;
|
|
||||||
import com.github.yulichang.toolkit.LambdaUtils;
|
|
||||||
import com.github.yulichang.wrapper.interfaces.LambdaJoin;
|
|
||||||
import com.github.yulichang.wrapper.interfaces.SFunctionQuery;
|
|
||||||
|
|
||||||
import java.util.ArrayList;
|
|
||||||
import java.util.List;
|
|
||||||
import java.util.Map;
|
|
||||||
import java.util.Objects;
|
|
||||||
import java.util.concurrent.atomic.AtomicInteger;
|
|
||||||
import java.util.function.Predicate;
|
|
||||||
import java.util.stream.Collectors;
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* copy {@link com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper}
|
* copy {@link com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper}
|
||||||
|
* <p>
|
||||||
|
* 名字太长 变更为MPJLambdaWrapper
|
||||||
|
* 推荐 使用 Wrappers.<>lambdaJoin(); 构造器
|
||||||
*
|
*
|
||||||
* @author yulichang
|
* @author yulichang
|
||||||
*/
|
*/
|
||||||
@SuppressWarnings("all")
|
@Deprecated
|
||||||
public class MPJJoinLambdaQueryWrapper<T> extends MPJAbstractLambdaWrapper<T, MPJJoinLambdaQueryWrapper<T>>
|
@SuppressWarnings("DeprecatedIsStillUsed")
|
||||||
implements SFunctionQuery<MPJJoinLambdaQueryWrapper<T>>, LambdaJoin<MPJJoinLambdaQueryWrapper<T>> {
|
public class MPJJoinLambdaQueryWrapper<T> extends MPJLambdaWrapper<T> {
|
||||||
|
|
||||||
/**
|
|
||||||
* 查询字段 sql
|
|
||||||
*/
|
|
||||||
private SharedString sqlSelect = new SharedString();
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 查询表
|
|
||||||
*/
|
|
||||||
private final SharedString from = new SharedString();
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 主表别名
|
|
||||||
*/
|
|
||||||
private final SharedString alias = new SharedString(Constant.TABLE_ALIAS);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 查询的字段
|
|
||||||
*/
|
|
||||||
private final List<SelectColumn> selectColumns = new ArrayList<>();
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 忽略查询的字段
|
|
||||||
*/
|
|
||||||
private final List<SelectColumn> ignoreColumns = new ArrayList<>();
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 表序号
|
|
||||||
*/
|
|
||||||
private int tableIndex = 1;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 不建议直接 new 该实例,使用 Wrappers.lambdaQuery()
|
|
||||||
*/
|
|
||||||
public MPJJoinLambdaQueryWrapper() {
|
|
||||||
super.initNeed();
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 不建议直接 new 该实例,使用 Wrappers.lambdaQuery(...)
|
|
||||||
*/
|
|
||||||
MPJJoinLambdaQueryWrapper(T entity, Class<T> entityClass, SharedString sqlSelect, AtomicInteger paramNameSeq,
|
|
||||||
Map<String, Object> paramNameValuePairs, MergeSegments mergeSegments,
|
|
||||||
SharedString lastSql, SharedString sqlComment, SharedString sqlFirst,
|
|
||||||
Map<Class<?>, Integer> subTable) {
|
|
||||||
super.setEntity(entity);
|
|
||||||
super.setEntityClass(entityClass);
|
|
||||||
this.paramNameSeq = paramNameSeq;
|
|
||||||
this.paramNameValuePairs = paramNameValuePairs;
|
|
||||||
this.expression = mergeSegments;
|
|
||||||
this.sqlSelect = sqlSelect;
|
|
||||||
this.lastSql = lastSql;
|
|
||||||
this.sqlComment = sqlComment;
|
|
||||||
this.sqlFirst = sqlFirst;
|
|
||||||
this.subTable = subTable;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* SELECT 部分 SQL 设置
|
|
||||||
*
|
|
||||||
* @param columns 查询字段
|
|
||||||
*/
|
|
||||||
@SafeVarargs
|
|
||||||
public final <S> MPJJoinLambdaQueryWrapper<T> select(SFunction<S, ?>... columns) {
|
|
||||||
if (ArrayUtils.isNotEmpty(columns)) {
|
|
||||||
for (SFunction<S, ?> s : columns) {
|
|
||||||
selectColumns.add(new SelectColumn(LambdaUtils.getEntityClass(s), LambdaUtils.getColumn(s), null));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return typedThis;
|
|
||||||
}
|
|
||||||
|
|
||||||
public <E> MPJJoinLambdaQueryWrapper<T> select(Class<E> entityClass, Predicate<TableFieldInfo> predicate) {
|
|
||||||
TableInfo info = TableInfoHelper.getTableInfo(entityClass);
|
|
||||||
Assert.notNull(info, "table can not be find");
|
|
||||||
info.getFieldList().stream().filter(predicate).collect(Collectors.toList()).forEach(
|
|
||||||
i -> selectColumns.add(new SelectColumn(entityClass, i.getColumn(), null)));
|
|
||||||
return typedThis;
|
|
||||||
}
|
|
||||||
|
|
||||||
public final <S, X> MPJJoinLambdaQueryWrapper<T> selectAs(SFunction<S, ?> columns, SFunction<X, ?> alias) {
|
|
||||||
return selectAs(columns, LambdaUtils.getName(alias));
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @since 1.1.3
|
|
||||||
*/
|
|
||||||
public final <S, X> MPJJoinLambdaQueryWrapper<T> selectAs(SFunction<S, ?> columns, String alias) {
|
|
||||||
selectColumns.add(new SelectColumn(LambdaUtils.getEntityClass(columns), LambdaUtils.getColumn(columns), alias));
|
|
||||||
return typedThis;
|
|
||||||
}
|
|
||||||
|
|
||||||
public final MPJJoinLambdaQueryWrapper<T> selectAll(Class<?> clazz) {
|
|
||||||
TableInfo info = TableInfoHelper.getTableInfo(clazz);
|
|
||||||
Assert.notNull(info, "table can not be find -> %s", clazz);
|
|
||||||
if (info.havePK()) {
|
|
||||||
selectColumns.add(new SelectColumn(clazz, info.getKeyColumn(), null));
|
|
||||||
}
|
|
||||||
info.getFieldList().forEach(c ->
|
|
||||||
selectColumns.add(new SelectColumn(clazz, c.getColumn(), null)));
|
|
||||||
return typedThis;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 忽略查询字段
|
|
||||||
* <p>
|
|
||||||
* 用法: selectIgnore(UserDO::getId,UserDO::getSex)
|
|
||||||
* 注意: 一个selectIgnore只支持一个对象 如果要忽略多个实体的字段,请调用多次
|
|
||||||
* <p>
|
|
||||||
* .selectIgnore(UserDO::getId,UserDO::getSex)
|
|
||||||
* .selectIgnore(UserAddressDO::getArea,UserAddressDO::getCity)
|
|
||||||
*
|
|
||||||
* @since 1.1.3
|
|
||||||
*/
|
|
||||||
@SafeVarargs
|
|
||||||
public final <S> MPJJoinLambdaQueryWrapper<T> selectIgnore(SFunction<S, ?>... columns) {
|
|
||||||
if (ArrayUtils.isNotEmpty(columns)) {
|
|
||||||
for (SFunction<S, ?> s : columns) {
|
|
||||||
ignoreColumns.add(new SelectColumn(LambdaUtils.getEntityClass(s), LambdaUtils.getColumn(s), null));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return typedThis;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public String getSqlSelect() {
|
|
||||||
if (StringUtils.isBlank(sqlSelect.getStringValue())) {
|
|
||||||
if (CollectionUtils.isNotEmpty(ignoreColumns)) {
|
|
||||||
selectColumns.removeIf(c -> ignoreColumns.stream().anyMatch(i ->
|
|
||||||
i.getClazz() == c.getClazz() && Objects.equals(c.getColumnName(), i.getColumnName())));
|
|
||||||
}
|
|
||||||
String s = selectColumns.stream().map(i ->
|
|
||||||
Constant.TABLE_ALIAS + getDefault(subTable.get(i.getClazz())) + StringPool.DOT + i.getColumnName() +
|
|
||||||
(StringUtils.isBlank(i.getAlias()) ? StringPool.EMPTY : (Constant.AS + i.getAlias())))
|
|
||||||
.collect(Collectors.joining(StringPool.COMMA));
|
|
||||||
sqlSelect.setStringValue(s);
|
|
||||||
}
|
|
||||||
return sqlSelect.getStringValue();
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
public String getFrom() {
|
|
||||||
return from.getStringValue();
|
|
||||||
}
|
|
||||||
|
|
||||||
public String getAlias() {
|
|
||||||
return alias.getStringValue();
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 用于生成嵌套 sql
|
|
||||||
* <p>故 sqlSelect 不向下传递</p>
|
|
||||||
*/
|
|
||||||
@Override
|
|
||||||
protected MPJJoinLambdaQueryWrapper<T> instance() {
|
|
||||||
return new MPJJoinLambdaQueryWrapper<>(getEntity(), getEntityClass(), null, paramNameSeq, paramNameValuePairs,
|
|
||||||
new MergeSegments(), SharedString.emptyString(), SharedString.emptyString(), SharedString.emptyString(),
|
|
||||||
this.subTable);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void clear() {
|
|
||||||
super.clear();
|
|
||||||
sqlSelect.toNull();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public <L, X> MPJJoinLambdaQueryWrapper<T> join(String keyWord, boolean condition, Class<L> clazz, SFunction<L, ?> left, SFunction<X, ?> right) {
|
|
||||||
if (condition) {
|
|
||||||
subTable.put(clazz, tableIndex);
|
|
||||||
TableInfo leftInfo = TableInfoHelper.getTableInfo(clazz);
|
|
||||||
StringBuilder sb = new StringBuilder(keyWord)
|
|
||||||
.append(leftInfo.getTableName())
|
|
||||||
.append(Constant.SPACE_TABLE_ALIAS)
|
|
||||||
.append(tableIndex)
|
|
||||||
.append(Constant.ON_TABLE_ALIAS)
|
|
||||||
.append(tableIndex)
|
|
||||||
.append(StringPool.DOT)
|
|
||||||
.append(LambdaUtils.getColumn(left))
|
|
||||||
.append(Constant.EQUALS_TABLE_ALIAS)
|
|
||||||
.append(getDefault(subTable.get(LambdaUtils.getEntityClass(right))))
|
|
||||||
.append(StringPool.DOT)
|
|
||||||
.append(LambdaUtils.getColumn(right));
|
|
||||||
tableIndex++;
|
|
||||||
if (StringUtils.isBlank(from.getStringValue())) {
|
|
||||||
from.setStringValue(sb.toString());
|
|
||||||
} else {
|
|
||||||
from.setStringValue(from.getStringValue() + sb.toString());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return typedThis;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* select字段
|
|
||||||
*/
|
|
||||||
public static class SelectColumn {
|
|
||||||
|
|
||||||
private Class<?> clazz;
|
|
||||||
|
|
||||||
private String columnName;
|
|
||||||
|
|
||||||
private String alias;
|
|
||||||
|
|
||||||
public SelectColumn(Class<?> clazz, String columnName, String alias) {
|
|
||||||
this.clazz = clazz;
|
|
||||||
this.columnName = columnName;
|
|
||||||
this.alias = alias;
|
|
||||||
}
|
|
||||||
|
|
||||||
public Class<?> getClazz() {
|
|
||||||
return clazz;
|
|
||||||
}
|
|
||||||
|
|
||||||
public void setClazz(Class<?> clazz) {
|
|
||||||
this.clazz = clazz;
|
|
||||||
}
|
|
||||||
|
|
||||||
public String getColumnName() {
|
|
||||||
return columnName;
|
|
||||||
}
|
|
||||||
|
|
||||||
public void setColumnName(String columnName) {
|
|
||||||
this.columnName = columnName;
|
|
||||||
}
|
|
||||||
|
|
||||||
public String getAlias() {
|
|
||||||
return alias;
|
|
||||||
}
|
|
||||||
|
|
||||||
public void setAlias(String alias) {
|
|
||||||
this.alias = alias;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
266
src/main/java/com/github/yulichang/wrapper/MPJLambdaWrapper.java
Normal file
266
src/main/java/com/github/yulichang/wrapper/MPJLambdaWrapper.java
Normal file
@ -0,0 +1,266 @@
|
|||||||
|
package com.github.yulichang.wrapper;
|
||||||
|
|
||||||
|
import com.baomidou.mybatisplus.core.conditions.SharedString;
|
||||||
|
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.*;
|
||||||
|
import com.baomidou.mybatisplus.core.toolkit.support.SFunction;
|
||||||
|
import com.github.yulichang.toolkit.Constant;
|
||||||
|
import com.github.yulichang.toolkit.LambdaUtils;
|
||||||
|
import com.github.yulichang.wrapper.interfaces.LambdaJoin;
|
||||||
|
import com.github.yulichang.wrapper.interfaces.SFunctionQuery;
|
||||||
|
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.Objects;
|
||||||
|
import java.util.concurrent.atomic.AtomicInteger;
|
||||||
|
import java.util.function.Predicate;
|
||||||
|
import java.util.stream.Collectors;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* copy {@link com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper}
|
||||||
|
* Lambda 语法使用 Wrapper
|
||||||
|
*
|
||||||
|
* @author yulichang
|
||||||
|
*/
|
||||||
|
@SuppressWarnings("all")
|
||||||
|
public class MPJLambdaWrapper<T> extends MPJAbstractLambdaWrapper<T, MPJLambdaWrapper<T>>
|
||||||
|
implements SFunctionQuery<MPJLambdaWrapper<T>>, LambdaJoin<MPJLambdaWrapper<T>> {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 查询字段 sql
|
||||||
|
*/
|
||||||
|
private SharedString sqlSelect = new SharedString();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 查询表
|
||||||
|
*/
|
||||||
|
private final SharedString from = new SharedString();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 主表别名
|
||||||
|
*/
|
||||||
|
private final SharedString alias = new SharedString(Constant.TABLE_ALIAS);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 查询的字段
|
||||||
|
*/
|
||||||
|
private final List<SelectColumn> selectColumns = new ArrayList<>();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 忽略查询的字段
|
||||||
|
*/
|
||||||
|
private final List<SelectColumn> ignoreColumns = new ArrayList<>();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 表序号
|
||||||
|
*/
|
||||||
|
private int tableIndex = 1;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 不建议直接 new 该实例,使用 Wrappers.lambdaQuery()
|
||||||
|
*/
|
||||||
|
public MPJLambdaWrapper() {
|
||||||
|
super.initNeed();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 不建议直接 new 该实例,使用 Wrappers.lambdaQuery(...)
|
||||||
|
*/
|
||||||
|
MPJLambdaWrapper(T entity, Class<T> entityClass, SharedString sqlSelect, AtomicInteger paramNameSeq,
|
||||||
|
Map<String, Object> paramNameValuePairs, MergeSegments mergeSegments,
|
||||||
|
SharedString lastSql, SharedString sqlComment, SharedString sqlFirst,
|
||||||
|
Map<Class<?>, Integer> subTable) {
|
||||||
|
super.setEntity(entity);
|
||||||
|
super.setEntityClass(entityClass);
|
||||||
|
this.paramNameSeq = paramNameSeq;
|
||||||
|
this.paramNameValuePairs = paramNameValuePairs;
|
||||||
|
this.expression = mergeSegments;
|
||||||
|
this.sqlSelect = sqlSelect;
|
||||||
|
this.lastSql = lastSql;
|
||||||
|
this.sqlComment = sqlComment;
|
||||||
|
this.sqlFirst = sqlFirst;
|
||||||
|
this.subTable = subTable;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* SELECT 部分 SQL 设置
|
||||||
|
*
|
||||||
|
* @param columns 查询字段
|
||||||
|
*/
|
||||||
|
@SafeVarargs
|
||||||
|
public final <S> MPJLambdaWrapper<T> select(SFunction<S, ?>... columns) {
|
||||||
|
if (ArrayUtils.isNotEmpty(columns)) {
|
||||||
|
for (SFunction<S, ?> s : columns) {
|
||||||
|
selectColumns.add(new SelectColumn(LambdaUtils.getEntityClass(s), LambdaUtils.getColumn(s), null));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return typedThis;
|
||||||
|
}
|
||||||
|
|
||||||
|
public <E> MPJLambdaWrapper<T> select(Class<E> entityClass, Predicate<TableFieldInfo> predicate) {
|
||||||
|
TableInfo info = TableInfoHelper.getTableInfo(entityClass);
|
||||||
|
Assert.notNull(info, "table can not be find");
|
||||||
|
info.getFieldList().stream().filter(predicate).collect(Collectors.toList()).forEach(
|
||||||
|
i -> selectColumns.add(new SelectColumn(entityClass, i.getColumn(), null)));
|
||||||
|
return typedThis;
|
||||||
|
}
|
||||||
|
|
||||||
|
public final <S, X> MPJLambdaWrapper<T> selectAs(SFunction<S, ?> columns, SFunction<X, ?> alias) {
|
||||||
|
return selectAs(columns, LambdaUtils.getName(alias));
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @since 1.1.3
|
||||||
|
*/
|
||||||
|
public final <S, X> MPJLambdaWrapper<T> selectAs(SFunction<S, ?> columns, String alias) {
|
||||||
|
selectColumns.add(new SelectColumn(LambdaUtils.getEntityClass(columns), LambdaUtils.getColumn(columns), alias));
|
||||||
|
return typedThis;
|
||||||
|
}
|
||||||
|
|
||||||
|
public final MPJLambdaWrapper<T> selectAll(Class<?> clazz) {
|
||||||
|
TableInfo info = TableInfoHelper.getTableInfo(clazz);
|
||||||
|
Assert.notNull(info, "table can not be find -> %s", clazz);
|
||||||
|
if (info.havePK()) {
|
||||||
|
selectColumns.add(new SelectColumn(clazz, info.getKeyColumn(), null));
|
||||||
|
}
|
||||||
|
info.getFieldList().forEach(c ->
|
||||||
|
selectColumns.add(new SelectColumn(clazz, c.getColumn(), null)));
|
||||||
|
return typedThis;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 忽略查询字段
|
||||||
|
* <p>
|
||||||
|
* 用法: selectIgnore(UserDO::getId,UserDO::getSex)
|
||||||
|
* 注意: 一个selectIgnore只支持一个对象 如果要忽略多个实体的字段,请调用多次
|
||||||
|
* <p>
|
||||||
|
* .selectIgnore(UserDO::getId,UserDO::getSex)
|
||||||
|
* .selectIgnore(UserAddressDO::getArea,UserAddressDO::getCity)
|
||||||
|
*
|
||||||
|
* @since 1.1.3
|
||||||
|
*/
|
||||||
|
@SafeVarargs
|
||||||
|
public final <S> MPJLambdaWrapper<T> selectIgnore(SFunction<S, ?>... columns) {
|
||||||
|
if (ArrayUtils.isNotEmpty(columns)) {
|
||||||
|
for (SFunction<S, ?> s : columns) {
|
||||||
|
ignoreColumns.add(new SelectColumn(LambdaUtils.getEntityClass(s), LambdaUtils.getColumn(s), null));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return typedThis;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getSqlSelect() {
|
||||||
|
if (StringUtils.isBlank(sqlSelect.getStringValue())) {
|
||||||
|
if (CollectionUtils.isNotEmpty(ignoreColumns)) {
|
||||||
|
selectColumns.removeIf(c -> ignoreColumns.stream().anyMatch(i ->
|
||||||
|
i.getClazz() == c.getClazz() && Objects.equals(c.getColumnName(), i.getColumnName())));
|
||||||
|
}
|
||||||
|
String s = selectColumns.stream().map(i ->
|
||||||
|
Constant.TABLE_ALIAS + getDefault(subTable.get(i.getClazz())) + StringPool.DOT + i.getColumnName() +
|
||||||
|
(StringUtils.isBlank(i.getAlias()) ? StringPool.EMPTY : (Constant.AS + i.getAlias())))
|
||||||
|
.collect(Collectors.joining(StringPool.COMMA));
|
||||||
|
sqlSelect.setStringValue(s);
|
||||||
|
}
|
||||||
|
return sqlSelect.getStringValue();
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
public String getFrom() {
|
||||||
|
return from.getStringValue();
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getAlias() {
|
||||||
|
return alias.getStringValue();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 用于生成嵌套 sql
|
||||||
|
* <p>故 sqlSelect 不向下传递</p>
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
protected MPJLambdaWrapper<T> instance() {
|
||||||
|
return new MPJLambdaWrapper<>(getEntity(), getEntityClass(), null, paramNameSeq, paramNameValuePairs,
|
||||||
|
new MergeSegments(), SharedString.emptyString(), SharedString.emptyString(), SharedString.emptyString(),
|
||||||
|
this.subTable);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void clear() {
|
||||||
|
super.clear();
|
||||||
|
sqlSelect.toNull();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public <L, X> MPJLambdaWrapper<T> join(String keyWord, boolean condition, Class<L> clazz, SFunction<L, ?> left, SFunction<X, ?> right) {
|
||||||
|
if (condition) {
|
||||||
|
subTable.put(clazz, tableIndex);
|
||||||
|
TableInfo leftInfo = TableInfoHelper.getTableInfo(clazz);
|
||||||
|
StringBuilder sb = new StringBuilder(keyWord)
|
||||||
|
.append(leftInfo.getTableName())
|
||||||
|
.append(Constant.SPACE_TABLE_ALIAS)
|
||||||
|
.append(tableIndex)
|
||||||
|
.append(Constant.ON_TABLE_ALIAS)
|
||||||
|
.append(tableIndex)
|
||||||
|
.append(StringPool.DOT)
|
||||||
|
.append(LambdaUtils.getColumn(left))
|
||||||
|
.append(Constant.EQUALS_TABLE_ALIAS)
|
||||||
|
.append(getDefault(subTable.get(LambdaUtils.getEntityClass(right))))
|
||||||
|
.append(StringPool.DOT)
|
||||||
|
.append(LambdaUtils.getColumn(right));
|
||||||
|
tableIndex++;
|
||||||
|
if (StringUtils.isBlank(from.getStringValue())) {
|
||||||
|
from.setStringValue(sb.toString());
|
||||||
|
} else {
|
||||||
|
from.setStringValue(from.getStringValue() + sb.toString());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return typedThis;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* select字段
|
||||||
|
*/
|
||||||
|
public static class SelectColumn {
|
||||||
|
|
||||||
|
private Class<?> clazz;
|
||||||
|
|
||||||
|
private String columnName;
|
||||||
|
|
||||||
|
private String alias;
|
||||||
|
|
||||||
|
public SelectColumn(Class<?> clazz, String columnName, String alias) {
|
||||||
|
this.clazz = clazz;
|
||||||
|
this.columnName = columnName;
|
||||||
|
this.alias = alias;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Class<?> getClazz() {
|
||||||
|
return clazz;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setClazz(Class<?> clazz) {
|
||||||
|
this.clazz = clazz;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getColumnName() {
|
||||||
|
return columnName;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setColumnName(String columnName) {
|
||||||
|
this.columnName = columnName;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getAlias() {
|
||||||
|
return alias;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setAlias(String alias) {
|
||||||
|
this.alias = alias;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -1,3 +1,4 @@
|
|||||||
org.springframework.boot.autoconfigure.EnableAutoConfiguration=\
|
org.springframework.boot.autoconfigure.EnableAutoConfiguration=\
|
||||||
com.github.yulichang.interceptor.MPJInterceptor,\
|
com.github.yulichang.interceptor.MPJInterceptor,\
|
||||||
com.github.yulichang.injector.MPJSqlInjector
|
com.github.yulichang.injector.MPJSqlInjector,\
|
||||||
|
com.github.yulichang.config.InterceptorConfig
|
||||||
|
Loading…
x
Reference in New Issue
Block a user