Java并发编程之ThreadLocal类

ThreadLocal类可以理解为ThreadLocalVariable(线程局部变量)，提供了get与set等访问接口或方法，这些方法为每个使用该变量的线程都存有一份独立的副本，因此get总是返回当前执行线程在调用set时设置的最新值。可以将ThreadLocal<T>视为 包含了Map<Thread,T>对象，保存了特定于该线程的值。

概括起来说，对于多线程资源共享的问题，同步机制采用了“以时间换空间”的方式，而ThreadLocal采用了“以空间换时间”的方式。前者仅提供一份变量，让不同的线程排队访问，而后者为每一个线程都提供了一份变量，因此可以同时访问而互不影响。

模拟ThreadLocal

import java.util.Collections;

import java.util.HashMap;

import java.util.Map;

 

public class SimpleThreadLocal<T> {

	private Map<Thread, T> valueMap = Collections

			.synchronizedMap(new HashMap<Thread, T>());

 

	public void set(T newValue) {

		valueMap.put(Thread.currentThread(), newValue); // ①键为线程对象，值为本线程的变量副本

	}

 

	public T get() {

		Thread currentThread = Thread.currentThread();

		T o = valueMap.get(currentThread); // ②返回本线程对应的变量

		if (o == null && !valueMap.containsKey(currentThread)) { // ③如果在Map中不存在，放到Map中保存起来。

			o = initialValue();

			valueMap.put(currentThread, o);

		}

		return o;

	}

 

	public void remove() {

		valueMap.remove(Thread.currentThread());

	}

 

	protected T initialValue() {

		return null;

	}

}

实用ThreadLocal

class Count {

	private SimpleThreadLocal<Integer> count = new SimpleThreadLocal<Integer>() {

		@Override

		protected Integer initialValue() {

			return 0;

		}

	};

 

	public Integer increase() {

		count.set(count.get() + 1);

		return count.get();

	}

 

}

 

class TestThread implements Runnable {

	private Count count;

 

	public TestThread(Count count) {

		this.count = count;

	}

 

	@Override

	public void run() {

		// TODO Auto-generated method stub

		for (int i = 1; i <= 3; i++) {

			System.out.println(Thread.currentThread().getName() + "\t" + i

					+ "th\t" + count.increase());

		}

	}

}

 

public class TestThreadLocal {

	public static void main(String[] args) {

		Count count = new Count();

		Thread t1 = new Thread(new TestThread(count));

		Thread t2 = new Thread(new TestThread(count));

		Thread t3 = new Thread(new TestThread(count));

		Thread t4 = new Thread(new TestThread(count));

		t1.start();

		t2.start();

		t3.start();

		t4.start();

	}

}

输出

Thread-0    1th    1

Thread-0    2th    2

Thread-0    3th    3

Thread-3    1th    1

Thread-1    1th    1

Thread-1    2th    2

Thread-2    1th    1

Thread-1    3th    3

Thread-3    2th    2

Thread-3    3th    3

Thread-2    2th    2

Thread-2    3th    3

来自为知笔记(Wiz)