com.cjy.fat.resolve.handler.ServiceRunningHandler Maven / Gradle / Ivy

Go to download
package com.cjy.fat.resolve.handler;

import java.util.Map;

import org.apache.commons.lang3.StringUtils;
import org.aspectj.lang.ProceedingJoinPoint;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
import org.springframework.context.ApplicationContext;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;
import org.springframework.transaction.TransactionStatus;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.transaction.support.AbstractPlatformTransactionManager;
import org.springframework.transaction.support.DefaultTransactionDefinition;

import com.alibaba.fastjson.JSONObject;
import com.cjy.fat.data.TransactionContent;
import com.cjy.fat.data.TransactionResolveParam;
import com.cjy.fat.redis.RedisHelper;
import com.cjy.fat.resolve.CommitResolver;

@Service
@ConditionalOnClass(value= {DataSourceTransactionManager.class})
public class ServiceRunningHandler {
	
	@Autowired
	ApplicationContext context;
	
	private static final Logger Logger = LoggerFactory.getLogger(ServiceRunningHandler.class);

	@Autowired
	RedisHelper redisHelper;
	
	@Autowired
	CommitResolver commitResolver;
	
	@Async("transactionResolveExecutor")
	public void proceed(ProceedingJoinPoint joinPoint,Transactional transactional ,TransactionResolveParam param ,Map txData) throws Throwable {
		this.bulidCunrrentThreadTxAttributesContent(txData);
		AbstractPlatformTransactionManager transactionManager = null ;
		String transactionManagerName = transactional.transactionManager();
		if(StringUtils.isNotBlank(transactionManagerName)) {
			transactionManager = (AbstractPlatformTransactionManager) context.getBean(transactionManagerName);
		}else {
			transactionManager = context.getBean(DataSourceTransactionManager.class);
		}
		DefaultTransactionDefinition transDefinition = new DefaultTransactionDefinition();
		transDefinition.setIsolationLevel(transactional.isolation().value());
		transDefinition.setPropagationBehavior(transactional.propagation().value());
		transDefinition.setReadOnly(transactional.readOnly());
		transDefinition.setTimeout(transactional.timeout());
		TransactionStatus transStatus = transactionManager.getTransaction(transDefinition);
		try {
			Logger.info("{}-transaction start"  ,param.getLocalTxMark());
			redisHelper.opsForServiceError().isServiceError(param.getTxKey());
			Object result = joinPoint.proceed();
			// 写入执行结果返回主线程 ,改用本地阻塞式队列
			param.offerToLocalResultQueue(result);
			Logger.info("{}-service is finished , transaction is waiting for commit,service result:{}", param.getLocalTxMark(), JSONObject.toJSONString(result));
			// 交给事务提交处理器处理可提交逻辑
			commitResolver.blockProceed(param);
			transactionManager.commit(transStatus);
			Logger.info("{}-transaction commit" , param.getLocalTxMark());
		} catch (Exception e) {
			redisHelper.opsForServiceError().serviceError(param.getTxKey());
			if(param.needToNotifyRootTxKey()) {
				redisHelper.opsForServiceError().serviceError(param.getTxKey());
			}
			param.setLocalRunningException(e);
			transactionManager.rollback(transStatus);
			Logger.error("{}-transaction rollback ,error:{}", param.getLocalTxMark() , e.getMessage());
		}
	}
	
	/**
	 * 由于在服务调服务的场景中，可能出现在service方法调用远程服务，处于不同线程中，所以此时需要再一次初始化当前线程的container
	 * @param txData
	 */
	private void bulidCunrrentThreadTxAttributesContent(Map txData) {
		if(null != txData) {
			TransactionContent.initContainer();
			String rootTxKey = txData.get(TransactionContent.STR_ROOT_TX_KEY);
			String localTxKey = txData.get(TransactionContent.STR_REMOTE_TX_KEY);
			if(StringUtils.isNotBlank(rootTxKey)) {
				TransactionContent.setRootTxKey(rootTxKey);
			}
			if(StringUtils.isNotBlank(localTxKey)) {
				TransactionContent.setLocalTxKey(localTxKey);
			}
		}
	}

}