admin/mybatis-flex
1
0
Fork 0
mirror of https://gitee.com/mybatis-flex/mybatis-flex.git synced 2025-12-07 00:58:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

add logic delete support

Browse Source
This commit is contained in:
开源海哥 2023-03-02 16:32:13 +08:00
parent ca180e18a4
commit 0afe2e5401
4 changed files with 168 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mybatis-flex-core/src/main/java/com/mybatisflex/core/BaseMapper.java
View File

@ -15,6 +15,7 @@
*/
package com.mybatisflex.core;
import com.mybatisflex.core.exception.FlexExceptions;
import com.mybatisflex.core.paginate.Page;
import com.mybatisflex.core.provider.EntitySqlProvider;
import com.mybatisflex.core.querywrapper.QueryColumn;
@ -82,6 +83,9 @@ public interface BaseMapper<T> {
* @return 返回影响的行数
*/
default int deleteByMap(Map<String, Object> whereConditions) {
if (whereConditions == null || whereConditions.isEmpty()) {
throw FlexExceptions.wrap("deleteByMap is not allow empty map.");
}
return deleteByQuery(QueryWrapper.create().where(whereConditions));
}

135
mybatis-flex-core/src/main/java/com/mybatisflex/core/dialect/CommonsDialectImpl.java
View File

@ -402,14 +402,111 @@ public class CommonsDialectImpl implements IDialect {
@Override
public String forDeleteEntityById(TableInfo tableInfo) {
String logicDeleteColumn = tableInfo.getLogicDeleteColumn();
//正常删除
if (StringUtil.isBlank(logicDeleteColumn)) {
return forDeleteById(tableInfo.getTableName(), tableInfo.getPrimaryKeys());
}
//逻辑删除
StringBuilder sql = new StringBuilder();
String[] primaryKeys = tableInfo.getPrimaryKeys();
sql.append("UPDATE ").append(wrap(tableInfo.getTableName())).append(" SET ");
sql.append(wrap(logicDeleteColumn)).append(" = 1");
sql.append(" WHERE ");
for (int i = 0; i < primaryKeys.length; i++) {
if (i > 0) {
sql.append(" AND ");
}
sql.append(wrap(primaryKeys[i])).append(" = ?");
}
sql.append(" AND ").append(wrap(logicDeleteColumn)).append(" = 0 ");
return sql.toString();
}
@Override
public String forDeleteEntityBatchById(TableInfo tableInfo, Object[] primaryValues) {
public String forDeleteEntityBatchByIds(TableInfo tableInfo, Object[] primaryValues) {
String logicDeleteColumn = tableInfo.getLogicDeleteColumn();
//正常删除
if (StringUtil.isBlank(logicDeleteColumn)) {
return forDeleteBatchByIds(tableInfo.getTableName(), tableInfo.getPrimaryKeys(), primaryValues);
}
StringBuilder sql = new StringBuilder();
sql.append("UPDATE");
sql.append(wrap(tableInfo.getTableName()));
sql.append("SET ").append(wrap(logicDeleteColumn)).append(" = 1");
sql.append(" WHERE ");
String[] primaryKeys = tableInfo.getPrimaryKeys();
//多主键的场景
if (primaryKeys.length > 1) {
for (int i = 0; i < primaryValues.length / primaryKeys.length; i++) {
if (i > 0) {
sql.append(" OR ");
}
sql.append("(");
for (int j = 0; j < primaryKeys.length; j++) {
if (j > 0) {
sql.append(" AND ");
}
sql.append(wrap(primaryKeys[j])).append(" = ?");
}
sql.append(")");
}
}
// 单主键
else {
for (int i = 0; i < primaryValues.length; i++) {
if (i > 0) {
sql.append(" OR ");
}
sql.append(wrap(primaryKeys[0])).append(" = ?");
}
}
return sql.toString();
}
@Override
public String forDeleteEntityBatchByQuery(TableInfo tableInfo, QueryWrapper queryWrapper) {
String logicDeleteColumn = tableInfo.getLogicDeleteColumn();
//正常删除
if (StringUtil.isBlank(logicDeleteColumn)) {
return forDeleteByQuery(queryWrapper);
}
//逻辑删除
List<QueryTable> queryTables = CPI.getQueryTables(queryWrapper);
List<QueryTable> joinTables = CPI.getJoinTables(queryWrapper);
List<QueryTable> allTables = CollectionUtil.merge(queryTables, joinTables);
//ignore selectColumns
StringBuilder sqlBuilder = new StringBuilder("UPDATE ");
sqlBuilder.append(wrap(tableInfo.getTableName()));
sqlBuilder.append(" SET ").append(wrap(logicDeleteColumn)).append(" = 1 ");
buildJoinSql(sqlBuilder, queryWrapper, allTables);
buildWhereSql(sqlBuilder, queryWrapper, allTables);
buildGroupBySql(sqlBuilder, queryWrapper, allTables);
buildHavingSql(sqlBuilder, queryWrapper, allTables);
//ignore orderBy and limit
//buildOrderBySql(sqlBuilder, queryWrapper);
//buildLimitSql(sqlBuilder, queryWrapper);
return sqlBuilder.toString();
}
@Override
public String forUpdateEntity(TableInfo tableInfo, Object entity, boolean ignoreNulls) {
@ -419,22 +516,20 @@ public class CommonsDialectImpl implements IDialect {
String[] primaryKeys = tableInfo.getPrimaryKeys();
sql.append("UPDATE ").append(wrap(tableInfo.getTableName())).append(" SET ");
int index = 0;
StringJoiner stringJoiner = new StringJoiner(", ");
for (String modifyAttr : modifyAttrs) {
if (index > 0) {
sql.append(", ");
}
sql.append(wrap(modifyAttr)).append(" = ? ");
index++;
stringJoiner.add(wrap(modifyAttr) + " = ?");
}
Map<String, String> onUpdateColumns = tableInfo.getOnUpdateColumns();
if (onUpdateColumns != null && !onUpdateColumns.isEmpty()) {
StringJoiner stringJoiner = new StringJoiner(", ");
onUpdateColumns.forEach((column, value) -> stringJoiner.add(wrap(column) + " = " + value));
sql.append(", ").append(stringJoiner);
}
sql.append(stringJoiner);
sql.append(" WHERE ");
for (int i = 0; i < primaryKeys.length; i++) {
if (i > 0) {
@ -443,6 +538,11 @@ public class CommonsDialectImpl implements IDialect {
sql.append(wrap(primaryKeys[i])).append(" = ?");
}
String logicDeleteColumn = tableInfo.getLogicDeleteColumn();
if (StringUtil.isNotBlank(logicDeleteColumn)) {
sql.append(" AND ").append(wrap(logicDeleteColumn)).append(" = 0 ");
}
return sql.toString();
}
@ -487,6 +587,13 @@ public class CommonsDialectImpl implements IDialect {
}
sql.append(wrap(pKeys[i])).append(" = ?");
}
//逻辑删除的情况下，需要添加逻辑删除的条件
String logicDeleteColumn = tableInfo.getLogicDeleteColumn();
if (StringUtil.isNotBlank(logicDeleteColumn)) {
sql.append(" AND ").append(wrap(logicDeleteColumn)).append(" = 0 ");
}
return sql.toString();
}
@ -498,6 +605,11 @@ public class CommonsDialectImpl implements IDialect {
sql.append(" WHERE ");
String[] primaryKeys = tableInfo.getPrimaryKeys();
String logicDeleteColumn = tableInfo.getLogicDeleteColumn();
if (StringUtil.isNotBlank(logicDeleteColumn)) {
sql.append(wrap(logicDeleteColumn)).append(" = 0 AND (");
}
//多主键的场景
if (primaryKeys.length > 1) {
for (int i = 0; i < primaryValues.length / primaryKeys.length; i++) {
@ -523,6 +635,11 @@ public class CommonsDialectImpl implements IDialect {
sql.append(wrap(primaryKeys[0])).append(" = ?");
}
}
if (StringUtil.isNotBlank(logicDeleteColumn)) {
sql.append(wrap(logicDeleteColumn)).append(" )");
}
return sql.toString();
}

4
mybatis-flex-core/src/main/java/com/mybatisflex/core/dialect/IDialect.java
View File

@ -65,7 +65,9 @@ public interface IDialect {
String forDeleteEntityById(TableInfo tableInfo);
String forDeleteEntityBatchById(TableInfo tableInfo, Object[] primaryValues);
String forDeleteEntityBatchByIds(TableInfo tableInfo, Object[] primaryValues);
String forDeleteEntityBatchByQuery(TableInfo tableInfo, QueryWrapper queryWrapper);
String forUpdateEntity(TableInfo tableInfo, Object entity, boolean ignoreNulls);

34
mybatis-flex-core/src/main/java/com/mybatisflex/core/provider/EntitySqlProvider.java
View File

@ -17,11 +17,14 @@ package com.mybatisflex.core.provider;
import com.mybatisflex.core.dialect.DialectFactory;
import com.mybatisflex.core.exception.FlexExceptions;
import com.mybatisflex.core.querywrapper.QueryColumn;
import com.mybatisflex.core.querywrapper.QueryCondition;
import com.mybatisflex.core.querywrapper.QueryWrapper;
import com.mybatisflex.core.querywrapper.CPI;
import com.mybatisflex.core.table.TableInfo;
import com.mybatisflex.core.util.ArrayUtil;
import com.mybatisflex.core.util.CollectionUtil;
import com.mybatisflex.core.util.StringUtil;
import org.apache.ibatis.builder.annotation.ProviderContext;
import java.io.Serializable;
@ -127,7 +130,7 @@ public class EntitySqlProvider {
TableInfo tableInfo = ProviderUtil.getTableInfo(context);
ProviderUtil.setSqlArgs(params, primaryValues);
return DialectFactory.getDialect().forDeleteEntityBatchById(tableInfo, primaryValues);
return DialectFactory.getDialect().forDeleteEntityBatchByIds(tableInfo, primaryValues);
}
@ -149,7 +152,8 @@ public class EntitySqlProvider {
TableInfo tableInfo = ProviderUtil.getTableInfo(context);
queryWrapper.from(tableInfo.getTableName());
return DialectFactory.getDialect().forDeleteByQuery(queryWrapper);
return DialectFactory.getDialect().forDeleteEntityBatchByQuery(tableInfo, queryWrapper);
}
@ -194,8 +198,13 @@ public class EntitySqlProvider {
boolean ignoreNulls = ProviderUtil.isIgnoreNulls(params);
QueryWrapper queryWrapper = ProviderUtil.getQueryWrapper(params);
TableInfo tableInfo = ProviderUtil.getTableInfo(context);
String logicDeleteColumn = tableInfo.getLogicDeleteColumn();
if (StringUtil.isNotBlank(logicDeleteColumn)) {
queryWrapper.and(QueryCondition.create(new QueryColumn(tableInfo.getTableName(), logicDeleteColumn), 0));
}
Object[] values = tableInfo.obtainUpdateValues(entity, ignoreNulls, true);
ProviderUtil.setSqlArgs(params, ArrayUtil.concat(values, CPI.getValueArray(queryWrapper)));
@ -259,11 +268,19 @@ public class EntitySqlProvider {
if (queryWrapper == null) {
throw FlexExceptions.wrap("queryWrapper can not be null.");
}
TableInfo tableInfo = ProviderUtil.getTableInfo(context);
String logicDeleteColumn = tableInfo.getLogicDeleteColumn();
if (StringUtil.isNotBlank(logicDeleteColumn)) {
queryWrapper.and(QueryCondition.create(new QueryColumn(tableInfo.getTableName(), logicDeleteColumn), 0));
}
Object[] values = CPI.getValueArray(queryWrapper);
ProviderUtil.setSqlArgs(params, values);
TableInfo tableInfo = ProviderUtil.getTableInfo(context);
queryWrapper.from(tableInfo.getTableName());
return DialectFactory.getDialect().forSelectListByQuery(queryWrapper);
}
@ -281,10 +298,17 @@ public class EntitySqlProvider {
throw FlexExceptions.wrap("queryWrapper can not be null.");
}
TableInfo tableInfo = ProviderUtil.getTableInfo(context);
//逻辑删除
String logicDeleteColumn = tableInfo.getLogicDeleteColumn();
if (StringUtil.isNotBlank(logicDeleteColumn)) {
queryWrapper.and(QueryCondition.create(new QueryColumn(tableInfo.getTableName(), logicDeleteColumn), 0));
}
Object[] values = CPI.getValueArray(queryWrapper);
ProviderUtil.setSqlArgs(params, values);
TableInfo tableInfo = ProviderUtil.getTableInfo(context);
queryWrapper.from(tableInfo.getTableName());
return DialectFactory.getDialect().forSelectCountByQuery(queryWrapper);
}