mybatis plus SQL注入器及 InsertBatchSomeColumn LogicDeleteByIdWithFill AlwaysUpdateSomeColumnById装载器

快来打我* 2024-04-19 16:33 1阅读 0赞

github地址[https://github.com/heng1234/mybatis\_plus][https_github.com_heng1234_mybatis_plus]

基于[https://blog.csdn.net/qq\_39313596/article/details/100943090][https_blog.csdn.net_qq_39313596_article_details_100943090]文章

参考官方文章 [https://gitee.com/baomidou/mybatis-plus-samples/tree/master/mybatis-plus-sample-deluxe][https_gitee.com_baomidou_mybatis-plus-samples_tree_master_mybatis-plus-sample-deluxe]

首先定义一个方法

DeleteAllMethod 继承 AbstractMethod

package com.hlvy.mybatis_plus.method;
    
    import com.baomidou.mybatisplus.core.injector.AbstractMethod;
    import com.baomidou.mybatisplus.core.metadata.TableInfo;
    import org.apache.ibatis.mapping.MappedStatement;
    import org.apache.ibatis.mapping.SqlSource;
    
    /**
     * DeleteAllMethod
     *
     * @author heng
     * @date 2019/9/19
     **/
    public class DeleteAllMethod extends AbstractMethod {
        @Override
        public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
            /* 执行 SQL ，动态 SQL 参考类 SqlMethod */
            String sql = "delete from " + tableInfo.getTableName();
            /* mapper 接口方法名一致 */
            String method = "deleteAll";
            SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, modelClass);
            return this.addDeleteMappedStatement(mapperClass, method, sqlSource);
        }
    }

接着写个sql配置类继承DefaultSqlInjector

package com.hlvy.mybatis_plus.injector;
    
    import com.baomidou.mybatisplus.core.injector.AbstractMethod;
    import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
    import com.baomidou.mybatisplus.extension.injector.methods.additional.AlwaysUpdateSomeColumnById;
    import com.baomidou.mybatisplus.extension.injector.methods.additional.InsertBatchSomeColumn;
    import com.baomidou.mybatisplus.extension.injector.methods.additional.LogicDeleteByIdWithFill;
    import com.hlvy.mybatis_plus.method.DeleteAllMethod;
    
    import java.util.List;
    
    /**
     * MySqlInjector
     * 自定义sql注入器
     * @author heng
     * @date 2019/9/19
     **/
    
    public class MySqlInjector extends DefaultSqlInjector {
    
        @Override
        public List<AbstractMethod> getMethodList(Class<?> mapperClass) {
            //拿到父类的getMethodList方法
            List<AbstractMethod> methodList = super.getMethodList(mapperClass);
            methodList.add(new DeleteAllMethod());
            //InsertBatchSomeColumn 选装器 谨慎用 值在mysql测试过切mysql默认值不会填写
            methodList.add(new InsertBatchSomeColumn(t->!t.isLogicDelete()&& !t.getColumn().equals("update_time")));//排除插入字段
           //根据id删除并自动填充 LogicDeleteByIdWithFill
            methodList.add(new LogicDeleteByIdWithFill());
            //根据id更新固定的那几个字段不包含逻辑删除 不更新name
            methodList.add(new AlwaysUpdateSomeColumnById(t->!t.getColumn().equals("name")));
           // methodList.add(new MyInsertAll());
           // methodList.add(new MysqlInsertAllBatch());
            return methodList;
        }
    }

然后把该类注入bean可以在该类加上@Component 或者在MybatisPlusConfig注入 我这是在MybatisPlusConfig注入的

/**
         * 自定义 SqlInjector
         * 里面包含自定义的全局方法
         */
        @Bean
        public MySqlInjector myLogicSqlInjector() {
            return new MySqlInjector();
        }

然后写个通用Mapper 写上你刚才配置的deleteAll方法

package com.hlvy.mybatis_plus.baseMapper;
    
    import com.baomidou.mybatisplus.core.mapper.BaseMapper;
    import com.baomidou.mybatisplus.core.toolkit.Constants;
    import org.apache.ibatis.annotations.Param;
    
    import java.util.List;
    
    /**
     * MyBaseMapper
     * 通用Mapper
     * @author heng
     * @date 2019/9/19
     **/
    public interface MyBaseMapper<T> extends BaseMapper<T> {
    
        /**
         * 自定义通用方法
         */
        Integer deleteAll();
    
        int myInsertAll(T entity);
    
    
        int insertBatchSomeColumn(List<T> list);
    
        int deleteByIdWithFill(T entity);
    
        int alwaysUpdateSomeColumnById(@Param(Constants.ENTITY) T entity);
    }

然后再让userMapper继承它

package com.hlvy.mybatis_plus.mapper;
    
    import com.baomidou.mybatisplus.core.conditions.Wrapper;
    import com.baomidou.mybatisplus.core.mapper.BaseMapper;
    import com.baomidou.mybatisplus.core.metadata.IPage;
    import com.baomidou.mybatisplus.core.toolkit.Constants;
    import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
    import com.hlvy.mybatis_plus.baseMapper.MyBaseMapper;
    import com.hlvy.mybatis_plus.entity.User;
    import org.apache.ibatis.annotations.Param;
    import org.apache.ibatis.annotations.Select;
    
    import java.util.List;
    
    /**
     * UserMapper
     *
     * @author heng
     **/
    
    public interface UserMapper extends MyBaseMapper<User> {
    
    
    
    }

测试

package com.hlvy.mybatis_plus.userTest;
    
    import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
    import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
    import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
    import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
    import com.baomidou.mybatisplus.core.metadata.IPage;
    import com.baomidou.mybatisplus.core.toolkit.StringUtils;
    import com.baomidou.mybatisplus.core.toolkit.Wrappers;
    import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
    import com.baomidou.mybatisplus.extension.service.additional.update.impl.LambdaUpdateChainWrapper;
    import com.hlvy.mybatis_plus.config.MybatisPlusConfig;
    import com.hlvy.mybatis_plus.entity.User;
    import com.hlvy.mybatis_plus.mapper.UserMapper;
    import org.junit.Test;
    import org.junit.runner.RunWith;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.boot.test.context.SpringBootTest;
    import org.springframework.test.context.junit4.SpringRunner;
    
    import java.util.*;
    
    /**
     * UserTest
     *
     * @author heng
     * @date 2019/4/11
     **/
    @RunWith(SpringRunner.class)
    @SpringBootTest
    public class InjectorTest {
        @Autowired
        private UserMapper userMapper;
    
        /**
         * 自定义sql注入器删除所有
         */
        @Test
        public void deleteAll() {
            System.out.println(userMapper.deleteAll());
    
        }
        /**
         * 自定义inser 选装器
         * 谨慎用 值在mysql测试过切mysql默认值不会填写
         * Consume Time：10 ms 2019-09-19 23:47:14
         *  Execute SQL：INSERT INTO User ( id, name, age, email, created_time ) VALUES ( 1174711555019276289, '恒果果', 19, 'xxx@qq.com', '2019-09-19T23:47:14.468+0800' )
         */
        @Test
        public void inserBath() {
            User user = new User();
            user.setName("恒果果");
            user.setAge(19);
            user.setEmail("xxx@qq.com");
            user.setManagerId(1L);
            User user2 = new User();
            user2.setName("恒果果2");
            user2.setAge(19);
            user2.setEmail("xxx@qq.com");
            user.setManagerId(1L);
            userMapper.insertBatchSomeColumn( Arrays.asList(user,user2));
    
        }
    
        /**
         * 根据id删除并自动填充 LogicDeleteByIdWithFill
         *  sql Consume Time：73 ms 2019-09-19 23:59:04
         *  Execute SQL：UPDATE User SET age=17,update_time='2019-09-19T23:59:04.167+0800',deleted=1 WHERE id=5 AND deleted=0
         */
        @Test
        public void deleteByIdWithFill() {
            User user = new User();
            user.setId(5L);
            user.setAge(17);//需要在删除的时候填充字段  需要在实体类加上上自动填充注解    @TableField(fill = FieldFill.UPDATE)
            System.out.println(userMapper.deleteByIdWithFill(user));
    
        }
    
    
        /**
         * 根据id更新固定的那几个字段不包含逻辑删除 不更新name 没有值会更新为空
         * UPDATE User SET age=17,update_time='2019-09-20T00:06:04.856+0800',deleted=1 WHERE id=1 AND deleted=0
         */
        @Test
        public void alwaysUpdateSomeColumnById() {
            User user = new User();
            user.setId(1L);
            user.setName("xxx");
            user.setAge(17);
            System.out.println(userMapper.deleteByIdWithFill(user));
    
        }
    }

[https_github.com_heng1234_mybatis_plus]: https://github.com/heng1234/mybatis_plus
[https_blog.csdn.net_qq_39313596_article_details_100943090]: https://blog.csdn.net/qq_39313596/article/details/100943090
[https_gitee.com_baomidou_mybatis-plus-samples_tree_master_mybatis-plus-sample-deluxe]: https://gitee.com/baomidou/mybatis-plus-samples/tree/master/mybatis-plus-sample-deluxe