mybatisPlus指定字段批量执行

最新推荐文章于 2024-05-01 03:28:46 发布

李子园05

最新推荐文章于 2024-05-01 03:28:46 发布

阅读量881

点赞数

分类专栏： MYSQL 文章标签： java mybatis 开发语言

本文链接：https://blog.csdn.net/qq_41028058/article/details/131437708

版权

MYSQL 专栏收录该内容

25 篇文章 0 订阅

订阅专栏

saveOrUpdateBatchByColumn(memberCompanyByBrainList, item -> new LambdaQueryWrapper<MmsMemberCompany>()
                        .eq(MmsMemberCompany::getUnifiedSocialCreditCode, item.getUnifiedSocialCreditCode())
                        .eq(MmsMemberCompany::getSite, item.getSite())
                );

public  boolean saveOrUpdateBatchByColumn(Collection<MmsMemberCompany> entityList, Function<MmsMemberCompany, LambdaQueryWrapper> function) {
        return this.saveOrUpdateBatch(this.entityClass, this.mapperClass, LogFactory.getLog(getClass()), entityList, 100, (sqlSession, entity) -> {
            Map<String, Object> param = Maps.newHashMap();
            param.put(com.baomidou.mybatisplus.core.toolkit.Constants.ENTITY, entity);
            param.put(com.baomidou.mybatisplus.core.toolkit.Constants.WRAPPER, function.apply(entity));
            return CollectionUtils.isEmpty(sqlSession.selectList(this.getSqlStatement(SqlMethod.SELECT_MAPS), param));
        }, (sqlSession, entity) -> {
            Map<String, Object> param = Maps.newHashMap();
            param.put(com.baomidou.mybatisplus.core.toolkit.Constants.ENTITY, entity);
            param.put(com.baomidou.mybatisplus.core.toolkit.Constants.WRAPPER, function.apply(entity));
            sqlSession.update(this.getSqlStatement(SqlMethod.UPDATE), param);
        });
    }

    public static <E> boolean saveOrUpdateBatch(Class<?> entityClass, Class<?> mapper, Log log, Collection<MmsMemberCompany> list, int batchSize, BiPredicate<SqlSession,MmsMemberCompany> predicate, BiConsumer<SqlSession, MmsMemberCompany> consumer) {
        String sqlStatement = SqlHelper.getSqlStatement(mapper, SqlMethod.INSERT_ONE);
        return SqlHelper.executeBatch(entityClass, log, list, batchSize, (sqlSession, entity) -> {
            if (predicate.test(sqlSession, entity)) {
                sqlSession.insert(sqlStatement, entity);
            } else {
                consumer.accept(sqlSession, entity);
            }
        });
    }