前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Java并发包--ThreadlocalRandom原理解密

Java并发包--ThreadlocalRandom原理解密

作者头像
小土豆Yuki
发布2020-06-15 17:16:46
5780
发布2020-06-15 17:16:46
举报
文章被收录于专栏:洁癖是一只狗

ThreadLocalRandom是JDK1.7新增的随机生成器,我们今天来说明为什么要JUC下要新增这个类,以及解密它的原理

首先我们看看我们熟悉的Random类,我们现提出两个问题,跟着问题看文章,可能事半功倍哟

  1. 它的实现原理
  2. 它有什么缺点

随机数需要生成一个默认的种子,这个种子其实就是一个long类型的数字,你可以在创建Random对象时候通过构造函数指定,如果不指定则会默认在构造函数中生成一个默认的值,有了默认的种子,他是如何生成随机数呢?

代码语言:javascript
复制
public int nextInt(int bound) {        //1.参数校验
        if (bound <= 0)
            throw new IllegalArgumentException(BadBound);
        //2.根据老的种子生成新的种子
        int r = next(31);        //3.根据新的种子计算随机数
        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;
}

重要步骤只要两点

  1. 根据老的种子生成新的种子
  2. 根据新的种子生成随机数

在单线程中,nextInt都是根据老种子生成新的种子,这是可保证随机数产生的随机性,但是在多线程中多个线程可能都拿到同一个老种子计算新的种子,由于新的种子计算随机数是固定的函数,因此多个线程可能根据同样的种子生成相同的随机数,当然这并不是我期望的,因此Random要保证2步骤的原子性.也就是说当多个线程同时获取到同一个新的 种子,只有一个线程可以根据新种子计算出随机数,其他线程都会丢弃掉拿到的种子。这样才能保证每一个线程获取的随机数都是随机的。

代码语言:javascript
复制
protected int next(int bits) {
        long oldseed, nextseed;
        AtomicLong seed = this.seed;
        do {             //4 获取当前原子变量的种子
            oldseed = seed.get();             //5根据老的种子生成新的种子
            nextseed = (oldseed * multiplier + addend) & mask;             //6使用CAS操作
        } while (!seed.compareAndSet(oldseed, nextseed));
        return (int)(nextseed >>> (48 - bits));
    }

第6步骤是最要的,他是使用CAS操作。当多个线程获取到相同的种子,在第6步骤的时候可以保证只有一个线程更新老的种子为新的,其他线程会继续循环去重新获取种子,这样就保证随机数的随机性。

但是,有使用CAS操作,会导致多个线程的进行自旋重试,这就会降低并发性能,所以ThreadLocalRandom应运而生。

ThreadLocalRandom的实现原理

ThreadLocalRandom我们可以联想ThreadLocal,ThreadLocal的实现原理是通过每个线程复制一个变量,使得每个线程对变量进行操作时实际上是操作自己本地内存里面的副本,从而避免了对共享变量进行同步,实际上ThreadLocalRandom也是这个实现原理,Random的缺点就是多个线程使用同一个原子种子变量,从而导致了原子变量的竞争,如下图

那么,如果每个线程都维护一个种子变量,则每个线程都会根据自己的 种子生成新的种子,再根据新的种子计算出随机数,这样就避免竞争问题,这就会大大提高并发性能,实现原理如下,

源码分析

可以看出ThreadLocalRandom类继承了Random,并重写了nextint方法,在ThreadlocalRandom没有继承自Random的原子种子变量,在ThreadLocalRandom并没有具体的种子,具体的种子存放到具体调用线程的threadlocalrandomSeed变量里面,就和ThreadLocal一样就是一个工具类,当线程调用ThreadlocalRandom的current,ThreadLocalRandom负责初始化threalocalrandomseed变量,就是初始化种子

当调用ThreadLocalRandom的nextInt的时候,实际上就是获取当前线程的threadlcoalrandomseed变量作为种子来计算新的种子,然后更新threadlocalrandomseed变量,根据新的种子使用具体的算法计算随机数,需要注意的是,threadlocalrandomseed仅仅是一个普通的long类型变量,并不是原子类型,但是由于他是线程级别的,所以他就不会存在线程安全的问题,并需要原子性变量,

其中seed和probeGenerator是原子变量,他是在初始化种子的时候使用,每个线程只会调用一次,另外变量instance是ThreadLocalRandom的一个实例,该变量是static,多个线程使用的实例是同一个,但是由于具体的种子存在在线程里面的,所以在ThreadlocalRandom的实例里面只包含线程无关的的通用算法,因此他是线程安全的。

代码语言:javascript
复制
private static final sun.misc.Unsafe UNSAFE;
    private static final long SEED;
    private static final long PROBE;
    private static final long SECONDARY;
    static {
        try {
            //获取unsafe实例
            UNSAFE = sun.misc.Unsafe.getUnsafe();
            Class<?> tk = Thread.class;
            //获取Thread类里面threadLocalRandomSeed变量在Thread实例里面偏移量
            SEED = UNSAFE.objectFieldOffset
                (tk.getDeclaredField("threadLocalRandomSeed"));
            //获取Thread类里面threadLocalRandomProbe变量在Thread实例里面偏移量
            PROBE = UNSAFE.objectFieldOffset
                (tk.getDeclaredField("threadLocalRandomProbe"));
            //获取Thread类里面threadLocalRandomProbe变量在Thread实例里面偏移量
            SECONDARY = UNSAFE.objectFieldOffset
                (tk.getDeclaredField("threadLocalRandomSecondarySeed"));
        } catch (Exception e) {
            throw new Error(e);
        }
    }

ThreadLocalRandom的current方法

该方法获取到ThreadlocalRandom的实例,并且初始化调用线程的thredLocalRandomSeed和threadlocalRandomProbe变量。

代码语言:javascript
复制
static final ThreadLocalRandom instance = new ThreadLocalRandom();
    public static ThreadLocalRandom current() {
        //(7)
        if (UNSAFE.getInt(Thread.currentThread(), PROBE) == 0)
            //(8)
            localInit();
        //(9)
        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();
        UNSAFE.putLong(t, SEED, seed);
        UNSAFE.putInt(t, PROBE, probe);
    }

如上面代码(7),如果线程的threadlocalRandomProbe的变量值为0时(默认他的值就是0),则说明他是第一次调用current,那么就会使用localinit()方法计算当前线程的初始化种子,这里是为了延迟初始化,在不使用生成随机数功能时,就不会初始化线程的种子变量,这也是一种优化,

代码(8)首先根据probeGenerator计算当前线程中ThreadLocalRandomProbe的初始值,然后根据seeder计算当前线程的初始化种子,而后把这两个变量放到当前线程,代码9返回ThreadLocalRandom的实例,需要注意的是这个方法是静态方法,多个线程返回的是同一个实例,

nextInt(int bound)

代码语言:javascript
复制
public int nextInt(int bound) {
        //(10)参数校验
        if (bound <= 0)
            throw new IllegalArgumentException(BadBound);
        //(11) 根据当前线程中种子计算新种子
        int r = mix32(nextSeed());
        //(12)根据新种子和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;
    }

如上代码的逻辑步骤和random相似,重点看一下,

代码语言:javascript
复制
final long nextSeed() {
        Thread t; long r; // 
        UNSAFE.putLong(t = Thread.currentThread(), SEED,
                       r = UNSAFE.getLong(t, SEED) + GAMMA);
        return r;
    }

如上代码中,首先使用r=UNSAFE.getLong(t,SEND)获得当前threadlocalrandomSeed的值,然后在种子的基础上加上GAMMA作为新种子,在使用UNSAFE的putLong方法把新的种子放入到当前线程的threadLocalRandomSeed变量中。

由于Random的缺点,从而引出了ThreadlcoalRandom类,ThreadLocalRandom使用Threadlocall的原理,让每个线程都持有一个本地的种子变量,该种子变量只有在使用随机数时才会被初始化,在多个线程中每个线程根据自己的种子进行更新,从而避免了并发竞争。

本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2020-04-02,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 洁癖是一只狗 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档