org.voovan.tools.FastThreadLocal Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of voovan-framework Show documentation
Voovan is a java framwork and it not depends on any third-party framework.
There is a newer version: 4.3.8
package org.voovan.tools;

import java.util.concurrent.atomic.AtomicInteger;
import java.util.function.Supplier;

/**
 * 线程局部变量
 *
 * @author: helyho
 * Voovan Framework.
 * WebSite: https://github.com/helyho/Voovan
 * Licence: Apache v2 License
 *
 * 每个 FastThreadLocal 在各个线程中都会持有一个 index 索引位置,实际访问的时候访问的是这个 FastThreadLocal 对象在各个线程中保存的那个实例
 */
public class FastThreadLocal {
	private ThreadLocal jdkThreadLocal = new ThreadLocal();
	private static AtomicInteger indexGenerator = new AtomicInteger(0);

	volatile int index = -1;

	private Object value;

	private Supplier supplier;

	/**
	 * 构造函数
	 * 		为当前对象生成一个 id
	 */
	public FastThreadLocal(){
		this(false);
	}

	/**
	 * 构造函数
	 * 		为当前对象生成一个 id
	 */
	private FastThreadLocal(boolean isInternal){
		//分配一个索引在所有线程中都是用这个索引位置
		if(!isInternal) {
			this.index = indexGenerator.getAndIncrement();
		}
	}

	/**
	 * 基于提供起的过早函数
	 * @param supplier 线程局部变量生成器
	 * @param  范型类型
	 * @return FastThreadLocal对象
	 */
	public static  FastThreadLocal withInitial(Supplier supplier){
		FastThreadLocal fastThreadLocal = new FastThreadLocal();
		fastThreadLocal.setSupplier(supplier);
		return fastThreadLocal;
	}

	/**
	 * 获取 线程局部变量生成器
	 * @return  线程局部变量生成器
	 */
	public Supplier getSupplier() {
		return supplier;
	}

	/**
	 * 设置 线程局部变量生成器
	 * @param supplier 线程局部变量生成器
	 */
	public void setSupplier(Supplier supplier) {
		this.supplier = supplier;
	}

	/**
	 * 获取 线程局部变量
	 * @return 线程局部变量
	 */
	public T get() {
		tryCreate();
		if(FastThread.getThread() != null) {
			FastThreadLocal fastThreadLocal = FastThread.getThread().data[index];

			T t = (T) fastThreadLocal.value;
			if (t == null && supplier != null) {
				t = (T) supplier.get();
				fastThreadLocal.value = t;
			}

			return t;
		} else {
			FastThreadLocal fastThreadLocal = jdkThreadLocal.get();
			T t = (T) fastThreadLocal.value;
			if (t == null && supplier != null) {
				t = (T) supplier.get();
				fastThreadLocal.value = t;
			}

			return t;
		}
	}

	/**
	 * 根据线程的类型尝试创建不同的线程局部变量
	 */
	public void tryCreate(){
		if(FastThread.getThread() != null) {
			FastThreadLocal[] data = FastThread.getThread().data;
			FastThreadLocal fastThreadLocal = data[index];
			if (fastThreadLocal == null) {
				fastThreadLocal = new FastThreadLocal(true);
				data[index] = fastThreadLocal;
			}
		} else {
			FastThreadLocal fastThreadLocal = jdkThreadLocal.get();
			if(fastThreadLocal == null){
				jdkThreadLocal.set(new FastThreadLocal(true));
			}
		}
	}

	/**
	 * 设置线程局部变量
	 * @param t 线程局部变量
	 */
	public void set(T t){
		tryCreate();

		if(FastThread.getThread() != null) {
			FastThread.getThread().data[index].value = t;
		} else {
			jdkThreadLocal.get().value = t;
		}
	}

}