重写Mybatis-Plus的BaseMapper的增加、删除、修改操作

最新推荐文章于 2024-05-16 23:30:33 发布

入门の小菜鸟

最新推荐文章于 2024-05-16 23:30:33 发布

阅读量5.4k

点赞数 2

分类专栏： mybatis-plus 文章标签： java mybatis

本文链接：https://blog.csdn.net/qq_43043092/article/details/123235569

版权

mybatis-plus 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

重写Mybatis-Plus的BaseMapper的增加、删除、修改操作

Mybatis-plus的deleteById()方法并不会去更新实体类中被标注的修改时间，所以通过重写deleteById()方法逻辑来实现这个功能。

1、创建自定义的Sql注入器

@Component
public class CustomSqlInjector extends DefaultSqlInjector {
 
    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass) {
        // 按照原有的对象进行复制，然后将要自定义的方法对象进行更换
        // 将deleteById方法和DeleteBatchByIds进行更换
        return Stream.of(new Insert(), new Delete(), new DeleteByMap(), new CustomDeleteById(),
                new CustomDeleteBatchByIds(), new Update(), new UpdateById(), new SelectById(),
                new SelectBatchByIds(), new SelectByMap(), new SelectOne(),
                new SelectCount(), new SelectMaps(), new SelectMapsPage(),
                new SelectObjs(), new SelectList(), new SelectPage()).collect(Collectors.toList());
    }
}

为了保证其他操作不受影响，将父类DefaultSqlInjector的getMethodList()方法复制到自定义的Sql注入器中，然后将需要自定义的方法换成自定义的对象即可

2、创建自定义的删除方法

将原有方法复制到自定义的类中，然后进行修改
public class MyDeleteById extends AbstractMethod {

    @Override
    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
        SqlMethod sqlMethod = SqlMethod.LOGIC_DELETE_BY_ID;
        String sql;
        if (tableInfo.isLogicDelete()) {
            // 获取被标注要自动填充的字段
            List<TableFieldInfo> fieldInfos = tableInfo.getFieldList().stream().filter(TableFieldInfo::isWithUpdateFill).collect(Collectors.toList());
            if (CollectionUtils.isNotEmpty(fieldInfos)) {
                String sqlSet = "SET " + fieldInfos.stream().map(i -> {
                    if (i.getPropertyType() == LocalDateTime.class) {
                        // 当为LocalDateTime类型时，添加当前时间
                        return i.getColumn() + "=now(), ";
                    }else {
                        return null;
                    }
                }).collect(Collectors.joining("")) + tableInfo.getLogicDeleteSql(false, false);
                sql = String.format(sqlMethod.getSql(), tableInfo.getTableName(), sqlSet, tableInfo.getKeyColumn(), tableInfo.getKeyProperty(), tableInfo.getLogicDeleteSql(true, true));
            } else {
                sql = String.format(sqlMethod.getSql(), tableInfo.getTableName(), this.sqlLogicSet(tableInfo), tableInfo.getKeyColumn(), tableInfo.getKeyProperty(), tableInfo.getLogicDeleteSql(true, true));
            }
        } else {
            sqlMethod = SqlMethod.DELETE_BY_ID;
            sql = String.format(sqlMethod.getSql(), tableInfo.getTableName(), tableInfo.getKeyColumn(), tableInfo.getKeyProperty());
        }

        SqlSource sqlSource = this.languageDriver.createSqlSource(this.configuration, sql, modelClass);
        return this.addUpdateMappedStatement(mapperClass, modelClass, this.getMethod(sqlMethod), sqlSource);
    }
}