mybatis-plus默认字段填充以及批量数据插入优化

日常开发中，我们需要设置一些数据库的默认字段填充，比如创建时间、创建人、更新时间、更新人等等，那么mybatis-plus给我们提供了一个这样的接口去做这件事情

MetaObjectHandler。

1、首先可以创建一个实现类来实现MetaObjectHandler接口

@Configuration
public class FillObjectHandler implements MetaObjectHandler {

    @Override
    public void insertFill(MetaObject metaObject) {
        log.info("start insert fill ....");
        this.setFieldValByName("createTime", DateUtils.getNowDate(), metaObject);
        this.setFieldValByName("createBy", SecurityContextHolder.getUserName(), metaObject);
    }

    @Override
    public void updateFill(MetaObject metaObject) {
        log.info("start update fill ....");
        this.setFieldValByName("updateTime", DateUtils.getNowDate(), metaObject);
        this.setFieldValByName("updateBy", SecurityContextHolder.getUserName(), metaObject);
    }
}

这里重写的两个方法，一个是再插入时调用，一个是更新时调用。

这里设置创建人的时候，有的时候，你是再子线程中进行获取上下文中用户名，但是子线程并不能获取到主线程的数据，那么这个时候，需要使用到ThreadLocal，将主线程的变量保存在ThreadLocal中，然后通过SecurityContextHolder来获取。

2、

public class SecurityContextHolder {
    private static final TransmittableThreadLocal<Map<String, Object>> THREAD_LOCAL = new TransmittableThreadLocal<>();

    public static void set(String key, Object value) {
        Map<String, Object> map = getLocalMap();
        map.put(key, value == null ? StringUtils.EMPTY : value);
    }

    public static String get(String key) {
        Map<String, Object> map = getLocalMap();
        return Convert.toStr(map.getOrDefault(key, StringUtils.EMPTY));
    }

    public static <T> T get(String key, Class<T> clazz) {
        Map<String, Object> map = getLocalMap();
        return StringUtils.cast(map.getOrDefault(key, null));
    }

    public static Map<String, Object> getLocalMap() {
        Map<String, Object> map = THREAD_LOCAL.get();
        if (map == null) {
            map = new ConcurrentHashMap<String, Object>();
            THREAD_LOCAL.set(map);
        }
        return map;
    }

    public static void setLocalMap(Map<String, Object> threadLocalMap) {
        THREAD_LOCAL.set(threadLocalMap);
    }

    public static String getUserName() {
        return get("username");
    }

    public static void setUserName(String username) {
        set("username", username);
    }

    public static void remove() {
        THREAD_LOCAL.remove();
    }
}

3、需要一个拦截器，在接受到请求时，将变量写入到Threadlocal中

@Component
public class SecurityInterceptor implements HandlerInterceptor {
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        //通过request获取到当前访问用户信息 在设置进线程变量中
        SecurityContextHolder.setUserName(SecurityUtils.getUsername());
        return true;
    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex)
            throws Exception {
        SecurityContextHolder.remove();
    }
}

做完以上步骤后，就可以完成默认字段设置了。

mybtais-plus是提供了批量插入的方法的，只需要继承ServiceImpl类就可以使用，但是貌似插入时也是单条进行插入。

他这块是批量提交到数据库，然后单条插入。

我们可以使用mybatis的sql注入器，mybatis-plus其实已经提供了这个方法

insertBatchSomeColumn

1、首先第一步需要自定义一个sql注入器

public class BatchSqlInjector extends DefaultSqlInjector {

    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass, TableInfo tableInfo) {
        List<AbstractMethod> methodList = super.getMethodList(mapperClass,tableInfo);
        //更新时自动填充的字段，不用插入值
        methodList.add(new InsertBatchSomeColumn(i -> i.getFieldFill() != FieldFill.UPDATE));
        return methodList;
    }

}

2、创建mybatis-plus的配置类

@Configuration
public class MyBatisPlusConfig {

    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        //分页
        interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));
        //乐观锁
        interceptor.addInnerInterceptor(new OptimisticLockerInnerInterceptor());
        return interceptor;
    }

    @Bean
    public BatchSqlInjector sqlInjector() {
        return new BatchSqlInjector();
    }

}

3、insertBatchSomeColumn方法名称需要保持一致

public interface BatchInsertMapper<T> extends BaseMapper<T> {

    int insertBatchSomeColumn(List<T> entityList);

}

4、然后使用我们的mapper调用insertBatchSomeColumn就可以使用一条sql语句去执行批量插入了。如果不是大批量数据需要插入，也没有时限要求，也用不上这个方法，适用于对大批量插入有性能要求的可以使用

5、另外一个可以在数据库连接池上进行调整。设置参数

rewriteBatchedStatements=true有很大批量插入性能提升