com.cjy.fat.data.TransactionContent Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of fat-common Show documentation

fat , the resolver of soa transaction

There is a newer version: 1.0.6-RELEASE

package com.cjy.fat.data;

import java.util.HashMap;
import java.util.Map;

public class TransactionContent {
	
	/**
	 * 所属的远程分布式事务key
	 */
	public static final String STR_REMOTE_TX_KEY = "remoteTxKey";
	
	/**
	 * 根txKey
	 */
	public static final String STR_ROOT_TX_KEY = "rootTxKey";
	
	private static ThreadLocal container = new ThreadLocal<>();
	
	/**
	 * 获取事务信息提供给自定义拦截器
	 * @return
	 */
	public static final Map buildRemoteData(){
		//返回新的对象，不开放修改入口，避免被客户端串改
		Map map = new HashMap<>();
		map.put(STR_REMOTE_TX_KEY, getLocalTxKey());
		map.put(STR_ROOT_TX_KEY, getRootTxKey());
		return map;
	}
	
	/**
	 * 设置远程分布式key
	 * @param txKey
	 * @return
	 */
	public static void setRemoteTxKey(String remoteTxKey) {
		container.get().setRemoteTxKey(remoteTxKey);
	}
	
	/**
	 * 获取远程分布式key
	 */
	public static String getRemoteTxKey() {
		return container.get().getRemoteTxKey();
	}
	
	/**
	 * 设置本地分布式key
	 * @param txKey
	 * @return
	 */
	public static void setLocalTxKey(String localTxKey) {
		container.get().setLocalTxKey(localTxKey);
	}
	
	/**
	 * 获取本地分布式key
	 */
	public static String getLocalTxKey() {
		return  container.get().getLocalTxKey();
	}
	
	public static void setRootTxKey(String rootTxKey) {
		container.get().setRootTxKey(rootTxKey);
	}

	public static String getRootTxKey() {
		return container.get().getRootTxKey();
	}
	
	/**
	 * 本地事务组加入元素
	 */
	public static void pushLocalTxQueue(String localTxMark){
		container.get().getLocalTxQueue().add(localTxMark);
	}
	
	/**
	 * 本地事务组取出元素
	 */
	public static String pollLocalTxQueue(){
		return container.get().getLocalTxQueue().poll();
	}
	
	/**
	 * 本地事务组数量
	 */
	public static int localTxQueueSize(){
		return container.get().getLocalTxQueue().size();
	}
	
	/**
	 * 获取服务标识
	 */
	public static String getServiceId(){
		return container.get().getServiceId();
	}
	
	/**
	 * 设置服务标识
	 */
	public static void setServiceId(String serviceId){
		container.get().setServiceId(serviceId);
	}
	
	/**
	 * 由于使用了线程池，当线程复用的时候，TheadLocal依然存在，需要在请求入口清空ThreadLocal
	 */
	public static final void initContainer(){
		container.remove();
		container.set(new TransactionThreadLocal());
	}
	
}