多线程插入数据开启事务，要么都成功要么都失败，PlatformTransactionManager 管理事务

最新推荐文章于 2024-08-21 19:05:11 发布

暖风吹i

最新推荐文章于 2024-08-21 19:05:11 发布

阅读量1.3k

点赞数

分类专栏： Java

本文链接：https://blog.csdn.net/weixin_44912855/article/details/119153695

版权

Java 专栏收录该内容

101 篇文章 19 订阅

订阅专栏

spring为我们提供了事务相关的接口，PlatformTransactionManager

public interface PlatformTransactionManager extends TransactionManager {

   // 获取当前活动的事务或创建一个新事务
   TransactionStatus getTransaction(@Nullable TransactionDefinition definition)
         throws TransactionException;

   // 提交事务
   void commit(TransactionStatus status) throws TransactionException;

   // 回滚事务
   void rollback(TransactionStatus status) throws TransactionException;

}

项目代码(copy下来, 供参考)

Executors.newSingleThreadScheduledExecutor().scheduleWithFixedDelay(() -> {
            try {
                Map<String, Object> map = redisUtil.hmEntries(RedisKey.event);
                if (!map.isEmpty()) {
                    
                    // 先在开启多线程外面，定义一个同步集合：
                    List<TransactionStatus> transactionStatuses = Collections.synchronizedList(new ArrayList<>());

                    List<EventObject> objectList = CacheUtil.mapValueToList(map, EventObject.class);
                    List<List<EventObject>> lists = ListUtils.cut(objectList, (int) Math.ceil((double) objectList.size() / threadPoolSize));
                    // 出错标识
                    AtomicBoolean error = new AtomicBoolean(false);
                    lists.forEach(l -> {
                        executor.execute(() -> {
                            try {
                                // 开启事务
                                DefaultTransactionDefinition def = new DefaultTransactionDefinition();
                                def.setPropagationBehavior(TransactionDefinition.PROPAGATION_REQUIRES_NEW);
                                TransactionStatus status = platformTransactionManager.getTransaction(def);
                                transactionStatuses.add(status);

                                // 保存事件
                                boolean r = eventService.saveBatch(new ArrayList<>(l));
                                log.info("保存事件:[{}]", r);
                                ArrayList<TServiceEventData> eventNumList = new ArrayList<>();
                                ArrayList<EventNumDTO> eventNumDTOS = new ArrayList<>();

                                for (EventObject object : l) {
                                    TServiceEventData eventData = JSON.parseObject(JSON.toJSONString(object.getEventData()), TServiceEventData.class);
                                    eventData.setServiceEventId(object.getId());
                                    eventNumList.add(eventData);

                                    // 事件数量
                                    EventNumDTO eventNumDTO = new EventNumDTO();
                                    eventNumDTO.setAppId(Long.valueOf(object.getAppId()));
                                    eventNumDTO.setServiceId(Long.valueOf(object.getServiceId()));
                                    eventNumDTOS.add(eventNumDTO);
                                }
                                // 保存事件数量
                                boolean r2 = eventDataService.saveBatch(eventNumList);
                                log.info("保存事件数量: [{}]", r2);

                                boolean b = updateOrInsertList(eventNumDTOS);
                                log.info("更新事件结果:{}", b);
                            } catch (TransactionException e) {
                                e.printStackTrace();
                                error.set(true);
                            }
                        });
                    });
                    // 如果出错回滚事务
                    if (error.get()) {
                        transactionStatuses.forEach(status -> platformTransactionManager.rollback(status));
                    } else {
                        transactionStatuses.forEach(status -> platformTransactionManager.commit(status));
                        Set<String> keySet = map.keySet();
                        keySet.forEach(key -> redisUtil.hmDelete(RedisKey.event, key));
                    }
                }
            } catch (Exception e) {
                log.error("保存处理event定时任务异常,{}", e.getMessage());
            }
        }, 1, delay, unit);