MybatisPlus如何实现批量插入

这篇文章将为大家详细讲解有关MybatisPlus如何实现批量插入，小编觉得挺实用的，因此分享给大家做个参考，希望大家阅读完这篇文章后可以有所收获。

一、背景

1. 再数据同步或者幂等场景下，常常需要设置唯一索引来避免重复请求，select and update效率低，且并发时还是会报错，并不友好，那么可以用Mysql的Insert ignore语法来优化。

2. MybatisPlus官方并没有针此处场景进行支持

二、环境

        com.baomidou        mybatis-plus-boot-starter        3.2.0                                                com.baomidou                        mybatis-plus-generator                        

三、注入自定义批量插入sql

因为只需要改造insertBatchSomeColumn方法，那直接CV就好

insertBatchSomeColumn方法属于mybatis plus官方扩展包中

1. sql模板

public class InsertIgnoreBatchAllColumn extends AbstractMethod {    /**     * mapper 对应的方法名     */    private static final String MAPPER_METHOD = "insertIgnoreBatchAllColumn";    /**     * 字段筛选条件     */    @Setter    @Accessors(chain = true)    private Predicate predicate;    @SuppressWarnings("Duplicates")    @Override    public MappedStatement injectMappedStatement(Class mapperClass, Class modelClass, TableInfo tableInfo) {        KeyGenerator keyGenerator = new NoKeyGenerator();        SqlMethod sqlMethod = SqlMethod.INSERT_ONE;        String sqlTemplate = "";        List fieldList = tableInfo.getFieldList();        String insertSqlColumn = tableInfo.getKeyInsertSqlColumn(false) +            this.filterTableFieldInfo(fieldList, predicate, TableFieldInfo::getInsertSqlColumn, EMPTY);        String columnScript = LEFT_BRACKET + insertSqlColumn.substring(0, insertSqlColumn.length() - 1) + RIGHT_BRACKET;        String insertSqlProperty = tableInfo.getKeyInsertSqlProperty(ENTITY_DOT, false) +            this.filterTableFieldInfo(fieldList, predicate, i -> i.getInsertSqlProperty(ENTITY_DOT), EMPTY);        insertSqlProperty = LEFT_BRACKET + insertSqlProperty.substring(0, insertSqlProperty.length() - 1) + RIGHT_BRACKET;        String valuesScript = SqlScriptUtils.convertForeach(insertSqlProperty, "list", null, ENTITY, COMMA);        String keyProperty = null;        String keyColumn = null;        // 表包含主键处理逻辑,如果不包含主键当普通字段处理        if (StringUtils.isNotEmpty(tableInfo.getKeyProperty())) {            if (tableInfo.getIdType() == IdType.AUTO) {                /* 自增主键 */                keyGenerator = new Jdbc3KeyGenerator();                keyProperty = tableInfo.getKeyProperty();                keyColumn = tableInfo.getKeyColumn();            } else {                if (null != tableInfo.getKeySequence()) {                    keyGenerator = TableInfoHelper.genKeyGenerator(tableInfo, builderAssistant, sqlMethod.getMethod(), languageDriver);                    keyProperty = tableInfo.getKeyProperty();                    keyColumn = tableInfo.getKeyColumn();                }            }        }        String sql = String.format(sqlTemplate, tableInfo.getTableName(), columnScript, valuesScript);        SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, modelClass);        return this.addInsertMappedStatement(mapperClass, modelClass, MAPPER_METHOD, sqlSource, keyGenerator, keyProperty, keyColumn);    }}

2. 注入sql

public class CustomerSqlInjector extends DefaultSqlInjector {    @Override    public List getMethodList(Class mapperClass) {        List methodList = super.getMethodList(mapperClass);        methodList.add(new InsertIgnoreBatchAllColumn());        return methodList;    }}

3. 通用mapper

public interface CommonMapper extends BaseMapper {    /**     * 全量插入,等价于insert，忽略唯一索引冲突的行     * {@link InsertIgnoreBatchAllColumn}     *     * @param entityList     * @return     */    int insertIgnoreBatchAllColumn(List entityList);}

4. 通用Service

public class CommonServiceImpl, T> extends ServiceImpl {    @Transactional(rollbackFor = Exception.class)    public boolean fastSaveIgnoreBatch(List list, int batchSize) {        if(CollectionUtils.isEmpty(list)) {            return true;        }        batchSize = batchSize < 1 ? BATCH_SIZE : batchSize;        if(list.size() <= batchSize) {            return retBool(baseMapper.insertIgnoreBatchAllColumn(list));        }        for (int fromIdx = 0 , endIdx = batchSize ; ; fromIdx += batchSize, endIdx += batchSize) {            if(endIdx > list.size()) {                endIdx = list.size();            }            baseMapper.insertIgnoreBatchAllColumn(list.subList(fromIdx, endIdx));            if(endIdx == list.size()) {                return true;            }        }    }                @Transactional(rollbackFor = Exception.class)    public boolean fastSaveIgnoreBatch(List list) {        return fastSaveIgnoreBatch(list, BATCH_SIZE);    }}

关于"MybatisPlus如何实现批量插入"这篇文章就分享到这里了，希望以上内容可以对大家有一定的帮助，使各位可以学到更多知识，如果觉得文章不错，请把它分享出去让更多的人看到。