JUC并发编程(四)----ThreadLocalRandom类原理剖析

Random类及其局限性

在JDK 7之前包括现在，java.util.Random都是使用比较广泛的随机数生成工具类，而且java.lang.Math中的随机数生成也使用的是java.util.Random的实例。 下面先看看java. util.Random的使用方法。

public class RandomTest {
    public static void main(String[] args) {
        //创建一个默认种子的随机数生成器
        Random random = new Random();

        //输出10个在0~5(包含0，不包含5)之间的随机数
        for (int i = 0; i < 5; i++) {
            System.out.println(random.nextInt(5));
        }
    }
}

随机数的生产需要一个默认的种子，这个种子其实是一个long类型的数字，可以在创建Random对象时通过构造函数指定，如果不指定则在默认构造函数内部生成一个默认的值。有了默认的种子后，如何生成随机数呢？ nextInt源码如下:

public int nextInt(int bound) {
		//参数检查
        if (bound <= 0)
            throw new IllegalArgumentException(BadBound);
		
		//根据老的种子生成新的种子
        int r = next(31);
        //根据新的种子计算随机数.
        int m = bound - 1;
        if ((bound & m) == 0)  // i.e., bound is a power of 2
            r = (int)((bound * (long)r) >> 31);
        else {
            for (int u = r;
                 u - (r = u % bound) + m < 0;
                 u = next(31))
                ;
        }
        return r;
    }

在单线程情况下每次调用nextInt都是根据老的种子计算出新的种子，这是可以保证随机数产生的随机性的。 但是在多线程下多个线程可能都拿同一个老的种子去计算新的种子，这会导致多个线程产生的新种子是一样的，由于算法是固定的，所以会导致多个线程产生相同的随机值，这并不是我们想要的。

所以需要保证原子性，也就是说当多个线程根据同一个老种子计算新种子时，第一个线程的新种子被计算出来后，第二个线程要丢弃自己老的种子，而使用第一个线程的新种子来计算自己的新种子，依此类推，只有保证了这个，才能保证在多线程下产生的随机数是随机的。

Random函数使用一个原子变量达到了这个效果，在创建Random对象时初始化的种子就被保存到了种子原子变量里面，下面看next（）的代码。

protected int next(int bits) {
        long oldseed, nextseed;
        AtomicLong seed = this.seed;
        do {
        	//（6）
            oldseed = seed.get();
            //（7）
            nextseed = (oldseed * multiplier + addend) & mask;
            //（8）
        } while (!seed.compareAndSet(oldseed, nextseed));
        //（9）
        return (int)(nextseed >>> (48 - bits));
    }

• （6）：获取当前原子变量种子的值。
• （7）：根据当前种子值计算新的种子
• （8）：使用CAS操作，它使用新的种子去更新老的种子，在多线程下可能多个线程都同时执行到了代码（6），那么可能多个线程拿到的当前种子的值是同一个，然后执行（7）计算的新种子也都是一样的，但是（8）的CAS操作会保证只有一个线程可以更新老的种子为新的，失败的线程会通过循环重新获取更新后的种子作为当前种子去计算老的种子，保证了随机数的随机性。
• （9）：使用固定算法根据新的种子计算随机数。

总结：每个Random实例里面都有一个原子性的种子变量用来记录当前的种子值，当要生成新的随机数时需要根据当前种子计算新的种子并更新回原子变量。在多线程下使用单个Random实例生成随机数时，当多个线程同时计算随机数来计算新的种子时，多个线程会竞争同一个原子变量的更新操作，由于原子变量的更新是CAS操作，同时只有一个线程会成功，所以会造成大量线程进行自旋重试，这会降低并发性能，所以ThreadLocalRandom应运而生。

ThreadLocalRandom

为了弥补多线程高并发情况下Random的缺陷，在JUC包下新增了ThreadLocalRandom类。下面首先看下如何使用它。

public class RandomTest {
    public static void main(String[] args) {
        //创建一个默认种子的随机数生成器
        ThreadLocalRandom random = ThreadLocalRandom.current();

        //输出10个在0~5(包含0，不包含5)之间的随机数
        for (int i = 0; i < 5; i++) {
            System.out.println(random.nextInt(5));
        }
    }
}

其实ThreadLocalRandom和ThreadLocal的实现原理类似，都是通过让每一个线程复制一份变量，使得在每个线程对变量进行操作时实际是操作自己本地内存里面的副本，从而避免了对共享变量进行同步。

每个线程都维护一个种子变量，则每个线程生成随机数时都根据自己老的种子计算新的种子，并使用新种子更新老的种子，再根据新种子计算随机数，就不会存在竞争问题了，这会大大提高并发性能。 原理如下图所示:

源码分析

首先看下ThreadLocalRandom的类图结构：

从图中可以看出ThreadLocalRandom类继承了Random类并重写了nextInt方法，在ThreadLocalRandom类中并没有使用继承自Random类的原子性种子变量。在ThreadLocalRandom中并没有存放具体的种子，具体的种子存放在具体的调用线程的threadLocalRandomSeed变量里面。ThreadLocalRandom类似于ThreadLocal类，就是个工具类。当线程调用ThreadLocalRandom的current方法时，ThreadLocalRandom负责初始化调用线程的threadLocalRandomSeed变量，也就是初始化种子。

当调用ThreadLocalRandom的nextInt方法时，实际上是获取当前线程的threadLocalRandomSeed变量作为当前种子来计算新的种子，然后更新新的种子到当前线程的threadLocalRandomSeed变量，而后再根据新种子并使用具体算法计算随机数。

这里需要注意的是，threadLocalRandomSeed变量就是Thread类里面的一个普通long变量，它并不是原子性变量。其实道理很简单，因为这个变量是线程级别的，所以根本不需要使用原子性变量。

其中seeder和probeGenerator是两个原子性变量，在初始化调用线程的种子和探针变量时会用到它们，每个线程只会使用一次。

另外，变量instance是ThreadLocalRandom的一个实例，该变量是static的。当多线程通过ThreadLocalRandom的current方法获取ThreadLocalRandom的实例时，其实获取的是同一个实例。但是由于具体的种子是存放在线程里面的，所以在ThreadLocalRandom的实例里面只包含与线程无关的通用算法，所以它是线程安全的。

ThreadLocalRandom主要代码实现逻辑

1.Unsafe机制

	//获取Unsafe实例
 private static final Unsafe U = Unsafe.getUnsafe();
    private static final long SEED;
    private static final long PROBE;
    private static final long SECONDARY;
    static {
        try {
        	//获取Thread类里面threadLocalRandomSeed变量再Thread实例里面的偏移量
            SEED = U.objectFieldOffset
                (Thread.class.getDeclaredField("threadLocalRandomSeed"));
            //获取Thread类里面threadLocalRandomProbe变量再Thread实例里面的偏移量
            PROBE = U.objectFieldOffset
                (Thread.class.getDeclaredField("threadLocalRandomProbe"));
            //获取Thread类里面threadLocalRandomSecond变量再Thread实例里面的偏移量
            SECONDARY = U.objectFieldOffset
                (Thread.class.getDeclaredField("threadLocalRandomSecondarySeed"));

        } catch (ReflectiveOperationException e) {
            throw new Error(e);
        }
    }

ThreadLocalRandom current()方法

该方法获取ThreadLocalRandom实例，并初始化调用线程中的threadLocalRandomSeed和threadLocalRandomProbe变量

public static ThreadLocalRandom current() {
		//（1）
        if (U.getInt(Thread.currentThread(), PROBE) == 0)
            //（2）
            localInit();
        //（3）
        return instance;
    }
     static final void localInit() {
        int p = probeGenerator.addAndGet(PROBE_INCREMENT);
        int probe = (p == 0) ? 1 : p; // skip 0
        long seed = mix64(seeder.getAndAdd(SEEDER_INCREMENT));
        Thread t = Thread.currentThread();
        U.putLong(t, SEED, seed);
        U.putInt(t, PROBE, probe);
    }

在上面代码（1）中，如果当前线程中的threadLocalRandomProbe的变量值为0(默认情况下线程的这个变量值为0)，则说明当前线程是第一次调用ThreadLocalRandom的current方法，那么就需要调用localInit方法计算当前线程的初始化种子变量。这里为了延迟初始化，在不需要使用随机数功能时就不初始化Thread类中的种子变量，这是一种优化。

代码（2）首先根据probeGenerator计算当前线程中threadLocalRandomProbe的初始化值，然后根据seeder计算当前线程的初始化种子，而后把这两个变量设置到当前线程。

代码（3）返回ThreadLocalRandom的实例。需要注意的是，这个方法是静态方法，多个线程返回的是同一个ThreadLocalRandom实例。

int nextInt(int bound)方法

计算当前线程的下一个随机数

public int nextInt(int bound) {
		//参数校验
        if (bound <= 0)
            throw new IllegalArgumentException(BAD_BOUND);
        //根据当前线程中的种子计算新种子
        int r = mix32(nextSeed());
        //根据新种子和bound计算随机数
        int m = bound - 1;
        if ((bound & m) == 0) // power of two
            r &= m;
        else { // reject over-represented candidates
            for (int u = r >>> 1;
                 u + m - (r = u % bound) < 0;
                 u = mix32(nextSeed()) >>> 1)
                ;
        }
        return r;
    }

nextSeed()方法：

final long nextSeed() {
        Thread t; long r; // read and update per-thread seed
        U.putLong(t = Thread.currentThread(), SEED,
                  r = U.getLong(t, SEED) + GAMMA);
        return r;
    }

在如上代码中，首先使用r = UNSAFE.getLong（t, SEED）获取当前线程中threadLocalRandomSeed变量的值，然后在种子的基础上累加GAMMA值作为新种子，而后使用UNSAFE的putLong方法把新种子放入当前线程的threadLocalRandomSeed变量中。

总结：

ThreadLocalRandom使用ThreadLocal的原理，让每个线程都持有一个本地的种子变量，该种子变量只有在使用随机数时才会被初始化。在多线程下计算新种子时是根据自己线程内维护的种子变量进行更新，从而避免了竞争。