Spring多线程的情况下,事务的处理需要注意以下几点:
- 不同线程对应不同的事务,不能通过**@Transactional**注解来保证线程间的事务一致性
- 需要使用TransactionTemplate或PlatformTransactionManager等编程式事务管理方式,手动控制事务的开启、提交和回滚
- 需要使用ThreadLocal或InheritableThreadLocal等线程局部变量机制,将事务对象传递给子线程,保证同一个事务对象在不同线程中使用
- 需要注意线程安全和资源释放的问题,避免出现死锁或内存泄漏的情况
CountDownLatch来保证多线程在一个事务内
大致步骤如下:
在主线程中创建一个CountDownLatch对象,初始化计数器的值为子线程的数量
在主线程中开启事务,并将事务对象保存在ThreadLocal或InheritableThreadLocal中
在主线程中创建并启动子线程,将CountDownLatch对象和事务对象传递给子线程
在子线程中获取事务对象,并执行写入数据的操作
在子线程中调用CountDownLatch的countDown()方法,将计数器的值减一
在主线程中调用CountDownLatch的await()方法,等待所有子线程执行完毕
在主线程中根据子线程的执行结果,决定提交或回滚事务,并释放资源
具体实现
// 主线程
@Service
public class MainService {
@Autowired
private PlatformTransactionManager transactionManager; // 事务管理器
@Autowired
private SubService subService; // 子线程服务类
public void mainMethod() {
int threadNum = 5; // 子线程数量
CountDownLatch latch = new CountDownLatch(threadNum); // 创建CountDownLatch对象
TransactionStatus status = transactionManager.getTransaction(new DefaultTransactionDefinition()); // 开启事务
ThreadLocal<TransactionStatus> threadLocal = new InheritableThreadLocal<>(); // 创建线程局部变量
threadLocal.set(status); // 将事务对象保存在线程局部变量中
for (int i = 0; i < threadNum; i++) {
new Thread(() -> {
subService.subMethod(latch, threadLocal.get()); // 调用子线程方法,传递CountDownLatch对象和事务对象
}).start();
}
try {
latch.await(); // 等待所有子线程执行完毕
transactionManager.commit(status); // 提交事务
System.out.println("主线程提交事务");
} catch (Exception e) {
transactionManager.rollback(status); // 回滚事务
System.out.println("主线程回滚事务");
} finally {
threadLocal.remove(); // 移除线程局部变量
}
}
}
// 子线程
@Service
public class SubService {
@Autowired
private TestDao testDao; // 数据库操作类
public void subMethod(CountDownLatch latch, TransactionStatus status) {
try {
testDao.insertData(); // 写入数据
latch.countDown(); // 计数器减一
System.out.println("子线程执行成功");
} catch (Exception e) {
status.setRollbackOnly(); // 设置回滚标志
latch.countDown(); // 计数器减一
System.out.println("子线程执行失败");
}
}
}