Mybatis Plus 自定义批量更新和插入方法

定义方法

@Slf4j
public class InsertBatchMethod extends AbstractMethod {
    @Override
    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
        final String sql = "<script>insert into %s %s values %s</script>";
        final String fieldSql = prepareFieldSql(tableInfo);
        final String valueSql = prepareValuesSql(tableInfo);
        final String sqlResult = String.format(sql, tableInfo.getTableName(), fieldSql, valueSql);
        SqlSource sqlSource = languageDriver.createSqlSource(configuration, sqlResult, modelClass);
        return this.addInsertMappedStatement(mapperClass, modelClass, "insertBatch", sqlSource, new NoKeyGenerator(), null, null);
    }

    private String prepareFieldSql(TableInfo tableInfo) {
        StringBuilder fieldSql = new StringBuilder();
        fieldSql.append(tableInfo.getKeyColumn()).append(",");
        tableInfo.getFieldList().forEach(x -> fieldSql.append(x.getColumn()).append(","));
        fieldSql.delete(fieldSql.length() - 1, fieldSql.length());
        fieldSql.insert(0, "(");
        fieldSql.append(")");
        return fieldSql.toString();
    }

    private String prepareValuesSql(TableInfo tableInfo) {
        final StringBuilder valueSql = new StringBuilder();
        valueSql.append("<foreach collection=\"list\" item=\"item\" index=\"index\" open=\"(\" separator=\"),(\" close=\")\">");
        valueSql.append("#{item.").append(tableInfo.getKeyProperty()).append("},");
        tableInfo.getFieldList().forEach(x -> valueSql.append("#{item.").append(x.getProperty()).append("},"));
        valueSql.delete(valueSql.length() - 1, valueSql.length());
        valueSql.append("</foreach>");
        return valueSql.toString();
    }
}

@Slf4j
public class UpdateBatchByIdsMethod extends AbstractMethod {
    @Override
    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
        final String sql = "<script>update %s set %s where id in %s</script>";
        final String fieldSql = prepareSetSql(tableInfo);
        final String valueSql = prepareWhereSql();
        final String sqlResult = String.format(sql, tableInfo.getTableName(), fieldSql, valueSql);
        SqlSource sqlSource = languageDriver.createSqlSource(configuration, sqlResult, modelClass);
        return this.addUpdateMappedStatement(mapperClass, modelClass, "updateBatchByIds", sqlSource);
    }

    private String prepareSetSql(TableInfo tableInfo) {
        final StringBuilder valueSql = new StringBuilder();
        valueSql.append("<trim suffixOverrides=\",\">");
        for (TableFieldInfo field : tableInfo.getFieldList()) {
            valueSql.append("<if ").append("test = \"entity.").append(field.getProperty()).append(" != null \"").append(">");
            valueSql.append(field.getColumn()).append(" = ").append("#{entity.").append(field.getProperty()).append("},");
            valueSql.append("</if>");
        }
        valueSql.append("</trim>");
        return valueSql.toString();
    }

    private String prepareWhereSql() {
        return "<foreach collection=\"ids\" item=\"id\" index=\"idx\" open=\"(\" close=\")\" separator=\",\">" +
                "        #{id}" +
                "    </foreach>";
    }
}

编写并配置 sql 注入器

@Component
public class CustomizedSqlInjector extends DefaultSqlInjector {
    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass) {
        List<AbstractMethod> methodList = super.getMethodList(mapperClass);
        methodList.add(new InsertBatchMethod());
        methodList.add(new UpdateBatchByIdsMethod());
        return methodList;
    }
}

编写基础 Mapper

public interface MyBaseMapper<T> extends BaseMapper<T> {
    
    int updateBatchByIds(@Param("entity") T entity, @Param("ids") List<Integer> ids);

    int insertBatch(@Param("list") List<T> list);
}

注意点

1. 新增的方法名称必须和AbstractMethod#injectMappedStatement里新增的名称一致。
2. Spring 会自动找到注入的SqlInjector bean，但是如果是自己定义的MybatisSqlSessionFactoryBean，则需要将该SqlInjector配置到MybatisSqlSessionFactoryBean的GlobalConfig中。