手動實現 spring 事務

手動實現 spring 事務

手動實現 spring 事務替換 @Transactional

## 自定義註解
@Target({ ElementType.METHOD })
@Retention(RetentionPolicy.RUNTIME)
public @interface ExtTransaction {

}

TransactionUtils

// 編程事務（需要手動begin 手動回滾 手都提交）
@Component
@Scope("prototype") // 每個事務都是新的實例 目的解決線程安全問題 多例子
public class TransactionUtils {

	// 全局接受事務狀態
	private TransactionStatus transactionStatus;
	// 獲取事務源
	@Autowired
	private DataSourceTransactionManager dataSourceTransactionManager;

	// 開啓事務
	public TransactionStatus begin() {
		System.out.println("開啓事務");
		transactionStatus = dataSourceTransactionManager.getTransaction(new DefaultTransactionAttribute());
		return transactionStatus;
	}

	// 提交事務
	public void commit(TransactionStatus transaction) {
		System.out.println("提交事務");
		dataSourceTransactionManager.commit(transaction);
	}

	// 回滾事務
	public void rollback() {
		System.out.println("回滾事務...");
		dataSourceTransactionManager.rollback(transactionStatus);
	}
}

aop 實現

//  自定義事務註解具體實現
@Aspect
@Component
public class AopExtTransaction {
	// 一個事務實例子 針對一個事務
	@Autowired
	private TransactionUtils transactionUtils;

	// 使用異常通知進行 回滾事務
	@AfterThrowing("execution(* com.itat.service.*.*.*(..))")
	public void afterThrowing() {
		// 獲取當前事務進行回滾
		// TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();
		transactionUtils.rollback();
	}

	// 環繞通知 在方法之前和之後處理事情
	@Around("execution(* com.itat.service.*.*.*(..))")
	public void around(ProceedingJoinPoint pjp) throws Throwable {

		// 1.獲取該方法上是否加上註解
		ExtTransaction extTransaction = getMethodExtTransaction(pjp);
		TransactionStatus transactionStatus = begin(extTransaction);
		// 2.調用目標代理對象方法
		pjp.proceed();
		// 3.判斷該方法上是否就上註解
		commit(transactionStatus);
	}

	private TransactionStatus begin(ExtTransaction extTransaction) {
		if (extTransaction == null) {
			return null;
		}
		// 2.如果存在事務註解,開啓事務
		return transactionUtils.begin();
	}

	private void commit(TransactionStatus transactionStatus) {
		if (transactionStatus != null) {
			// 5.如果存在註解,提交事務
			transactionUtils.commit(transactionStatus);
		}

	}

	// 獲取方法上是否存在事務註解
	private ExtTransaction getMethodExtTransaction(ProceedingJoinPoint pjp)
			throws NoSuchMethodException, SecurityException {
		String methodName = pjp.getSignature().getName();
		// 獲取目標對象
		Class<?> classTarget = pjp.getTarget().getClass();
		// 獲取目標對象類型
		Class<?>[] par = ((MethodSignature) pjp.getSignature()).getParameterTypes();
		// 獲取目標對象方法
		Method objMethod = classTarget.getMethod(methodName, par);
		ExtTransaction extTransaction = objMethod.getDeclaredAnnotation(ExtTransaction.class);
		return extTransaction;
	}

}

調用過程

// @ExtTransaction
	@Transactional
	public void add() {
		// 調用接口的時候 接口失敗 需要回滾，但是日誌記錄不需要回滾。
		logService.addLog(); // 後面程序發生錯誤，不能影響到我的回滾### 正常當addLog方法執行完畢，就應該提交事務
		userDao.add("test001", 20);
		// int i = 1 / 0;
		System.out.println("################");
		userDao.add("test002", 21);

	}