refactor: 优化主键逻辑删除处理器逻辑。

This commit is contained in:
Suomm 2023-07-29 11:33:09 +08:00
parent ab87696d91
commit e8b346bdb7

View File

@ -16,7 +16,6 @@
package com.mybatisflex.core.logicdelete.impl; package com.mybatisflex.core.logicdelete.impl;
import com.mybatisflex.core.FlexGlobalConfig;
import com.mybatisflex.core.dialect.IDialect; import com.mybatisflex.core.dialect.IDialect;
import com.mybatisflex.core.exception.FlexAssert; import com.mybatisflex.core.exception.FlexAssert;
import com.mybatisflex.core.logicdelete.AbstractLogicDeleteProcessor; import com.mybatisflex.core.logicdelete.AbstractLogicDeleteProcessor;
@ -36,24 +35,25 @@ import static com.mybatisflex.core.constant.SqlConsts.EQUALS;
*/ */
public class PrimaryKeyLogicDeleteProcessor extends AbstractLogicDeleteProcessor { public class PrimaryKeyLogicDeleteProcessor extends AbstractLogicDeleteProcessor {
@Override
public String buildLogicNormalCondition(String logicColumn, TableInfo tableInfo, IDialect dialect) {
return dialect.wrap(logicColumn) + " IS NULL";
}
@Override @Override
public String buildLogicDeletedSet(String logicColumn, TableInfo tableInfo, IDialect dialect) { public String buildLogicDeletedSet(String logicColumn, TableInfo tableInfo, IDialect dialect) {
List<IdInfo> primaryKeyList = tableInfo.getPrimaryKeyList(); List<IdInfo> primaryKeyList = tableInfo.getPrimaryKeyList();
FlexAssert.notEmpty(primaryKeyList, "Must have one primary key."); FlexAssert.notEmpty(primaryKeyList, "Entity must have one primary key.");
String column = primaryKeyList.get(0).getColumn(); String column = primaryKeyList.get(0).getColumn();
return dialect.wrap(logicColumn) + EQUALS + dialect.wrap(column); return dialect.wrap(logicColumn) + EQUALS + dialect.wrap(column);
} }
/** /**
* 正常未删除的值为 0 或者可配置为其他值 * 正常未删除的值为 {@code null}兼容不同的主键类型
*/ */
@Override @Override
public Object getLogicNormalValue() { public Object getLogicNormalValue() {
Object normalValueOfLogicDelete = FlexGlobalConfig.getDefaultConfig().getNormalValueOfLogicDelete(); return null;
if (normalValueOfLogicDelete instanceof Number) {
return normalValueOfLogicDelete;
}
return 0;
} }
/** /**