admin 2021-04-22 17:30:07 +08:00
parent 32c333234b
commit fd6de6f50b
7 changed files with 358 additions and 281 deletions

View File

@ -19,51 +19,51 @@ public interface MPJBaseMapper<T> extends BaseMapper<T> {
/**
* 连表查询返回一条记录
*
* @param myWrapper joinWrapper
* @param wrapper joinWrapper
* @param clazz resultType
*/
<DTO> DTO selectJoinOne(@Param(Constant.CLAZZ) Class<DTO> clazz,
@Param(Constants.WRAPPER) MPJBaseJoin myWrapper);
@Param(Constants.WRAPPER) MPJBaseJoin wrapper);
/**
* 连表查询返回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
*/
<DTO> List<DTO> selectJoinList(@Param(Constant.CLAZZ) Class<DTO> clazz,
@Param(Constants.WRAPPER) MPJBaseJoin myWrapper);
@Param(Constants.WRAPPER) MPJBaseJoin wrapper);
/**
* 连表查询返回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 <DTO> 分页返回对象
*/
<DTO, P extends IPage<?>> IPage<DTO> selectJoinPage(P page,
@Param(Constant.CLAZZ) Class<DTO> clazz,
@Param(Constants.WRAPPER) MPJBaseJoin myWrapper);
@Param(Constants.WRAPPER) MPJBaseJoin wrapper);
/**
* 连表查询返回Map集合并分页
*
* @param myWrapper joinWrapper
* @param wrapper joinWrapper
*/
<P extends IPage<?>> IPage<Map<String, Object>> selectJoinMapsPage(P page,
@Param(Constants.WRAPPER) MPJBaseJoin myWrapper);
@Param(Constants.WRAPPER) MPJBaseJoin wrapper);
}

View File

@ -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");
}
}
}
}

View File

@ -0,0 +1,13 @@
package com.github.yulichang.exception;
/**
* mpj 异常
*
* @author yulichang
*/
public class MPJException extends RuntimeException {
public MPJException(String msg) {
super(msg);
}
}

View File

@ -1,8 +1,7 @@
package com.github.yulichang.toolkit;
import com.github.yulichang.query.MPJLambdaQueryWrapper;
import com.github.yulichang.query.MPJQueryWrapper;
import com.github.yulichang.wrapper.MPJJoinLambdaQueryWrapper;
import com.github.yulichang.wrapper.MPJLambdaWrapper;
/**
* Wrapper 条件构造
@ -15,12 +14,7 @@ public class Wrappers {
return new MPJQueryWrapper<>();
}
@Deprecated
public static <T> MPJLambdaQueryWrapper<T> lambdaJoin() {
return new MPJLambdaQueryWrapper<>();
}
public static <T> MPJJoinLambdaQueryWrapper<T> lambdaJoinWrapper() {
return new MPJJoinLambdaQueryWrapper<>();
public static <T> MPJLambdaWrapper<T> lambdaJoin() {
return new MPJLambdaWrapper<>();
}
}

View File

@ -1,265 +1,16 @@
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}
* <p>
* 名字太长 变更为MPJLambdaWrapper
* 推荐 使用 Wrappers.<>lambdaJoin(); 构造器
*
* @author yulichang
*/
@SuppressWarnings("all")
public class MPJJoinLambdaQueryWrapper<T> extends MPJAbstractLambdaWrapper<T, MPJJoinLambdaQueryWrapper<T>>
implements SFunctionQuery<MPJJoinLambdaQueryWrapper<T>>, LambdaJoin<MPJJoinLambdaQueryWrapper<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();
}
@Deprecated
@SuppressWarnings("DeprecatedIsStillUsed")
public class MPJJoinLambdaQueryWrapper<T> extends MPJLambdaWrapper<T> {
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;
}
}
}

View 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;
}
}
}

View File

@ -1,3 +1,4 @@
org.springframework.boot.autoconfigure.EnableAutoConfiguration=\
com.github.yulichang.interceptor.MPJInterceptor,\
com.github.yulichang.injector.MPJSqlInjector
com.github.yulichang.injector.MPJSqlInjector,\
com.github.yulichang.config.InterceptorConfig