目录

  • 一、CAS & volatile
  • 二、原子整数
  • 二、原子引用
  • 1.ABA问题
  • 2.AtomicStampedReference
  • 3.AtomicMarkableReference
  • 三、原子数组
  • 四、字段更新器
  • 五、原子累加器
  • 1.伪共享
  • 2.LongAdder源码
  • add方法
  • longAccumulate方法
  • sum方法
  • 六、Unsafe
  • 1.Unsafe CAS操作


一、CAS & volatile

JUC提供的多个类,类中的方法内部并没有用锁来保护共享变量的线程安全。
而是采用了CAS(compareAndSet,是原子操作)保证线程安全

CAS 的底层是 lock cmpxchg 指令(X86 架构),在单核 CPU 和多核 CPU 下都能够保证【比较-交换】的原子性。

在多核状态下,某个核执行到带 lock 的指令时,CPU 会让总线锁住,当这个核把此指令执行完毕,再开启总线。这个过程中不会被线程的调度机制所打断,保证了多个线程对内存操作的准确性,是原子的。

CAS 必须借助 volatile 才能读取到共享变量的最新值来实现【比较并交换】的效果

CAS的特点:结合 CAS 和 volatile 可以实现无锁并发

  • CAS 是基于乐观锁的思想:最乐观的估计,不怕别的线程来修改共享变量,就算改了也没关系,可以再重试。
  • synchronized 是基于悲观锁的思想:最悲观的估计,得防着其它线程来修改共享变量,我上了锁你们都别想改,我改完了解开锁,你们才有机会。
  • CAS 体现的是无锁并发、无阻塞并发
    因为没有使用 synchronized,所以线程不会陷入阻塞,这是效率提升的因素之一
    但如果竞争激烈,可以想到重试必然频繁发生,反而效率会受影响

二、原子整数

J.U.C 并发包提供了:

  • AtomicBoolean
  • AtomicInteger
  • AtomicLong
//以 AtomicInteger 为例
AtomicInteger i = new AtomicInteger(0);

// 获取并自增(i = 0, 结果 i = 1, 返回 0),类似于 i++
System.out.println(i.getAndIncrement());

// 自增并获取(i = 1, 结果 i = 2, 返回 2),类似于 ++i
System.out.println(i.incrementAndGet());

// 自减并获取(i = 2, 结果 i = 1, 返回 1),类似于 --i
System.out.println(i.decrementAndGet());

// 获取并自减(i = 1, 结果 i = 0, 返回 1),类似于 i--
System.out.println(i.getAndDecrement());

// 获取并加值(i = 0, 结果 i = 5, 返回 0)
System.out.println(i.getAndAdd(5));

// 加值并获取(i = 5, 结果 i = 0, 返回 0)
System.out.println(i.addAndGet(-5));

// 获取并更新(i = 0, p 为 i 的当前值, 结果 i = -2, 返回 0)
// 其中函数中的操作能保证原子,但函数需要无副作用
System.out.println(i.getAndUpdate(p -> p - 2));

// 更新并获取(i = -2, p 为 i 的当前值, 结果 i = 0, 返回 0)
// 其中函数中的操作能保证原子,但函数需要无副作用
System.out.println(i.updateAndGet(p -> p + 2));

// 获取并计算(i = 0, p 为 i 的当前值, x 为参数1, 结果 i = 10, 返回 0)
// 其中函数中的操作能保证原子,但函数需要无副作用
// getAndUpdate 如果在 lambda 中引用了外部的局部变量,要保证该局部变量是 final 的
// getAndAccumulate 可以通过 参数1 来引用外部的局部变量,但因为其不在 lambda 中因此不必是 final
System.out.println(i.getAndAccumulate(10, (p, x) -> p + x));

// 计算并获取(i = 10, p 为 i 的当前值, x 为参数1, 结果 i = 0, 返回 0)
// 其中函数中的操作能保证原子,但函数需要无副作用
System.out.println(i.accumulateAndGet(-10, (p, x) -> p + x));

二、原子引用

  • AtomicReference
  • AtomicMarkableReference
  • AtomicStampedReference
AtomicReference<BigDecimal> ref;
//compareAndSet(prev, next)用来修改BigDecimal的值为next
//修改之前会判断原来的值是不是还是prev,不是的话会返回false
//修改成功返回true
//如果修改失败会一直重试

//在while中,通过compareAndSet判断修改
while (true) {
	BigDecimal prev = ref.get();
	BigDecimal next = prev.subtract(amount);
	if (ref.compareAndSet(prev, next)) {
		break;
	}
}

1.ABA问题

compareAndSet仅能判断出在我们要修改时,共享变量的值与最初值是否相同,但并不能看出共享变量是否被修改过。

如果共享变量被其他线程从A修改为B,又被从B修改为A,这时当另一个线程调用cas,cas是发现不了的。

如何解决
再加一个版本号,作为另一个判断标准。

2.AtomicStampedReference

AtomicStampedReference 可以给原子引用加上版本号,追踪原子引用整个的变化过程。只需要在初始化操作传入一个版本号(int)。

compareAndSet也会增加两个变量,一个是之前的版本号,一个是下一个版本号,在修改时,除了比较之前的值外,还会比较版本号,两个都没变才能修改。

3.AtomicMarkableReference

有时候,并不关心引用变量更改了几次,只是单纯的关心是否更改过(即true/false即可),所以就有了AtomicMarkableReference,使用与AtomicStampedReference相似。

三、原子数组

AtomicIntegerArray
AtomicLongArray
AtomicReferenceArray
使用与前面的都类似

四、字段更新器

AtomicReferenceFieldUpdater // 域 字段
AtomicIntegerFieldUpdater
AtomicLongFieldUpdater

利用字段更新器,可以针对对象的某个域(Field)进行原子操作,只能配合 volatile 修饰的字段使用,否则会出现异常。

public class Test {
	private volatile int field;
	
	public static void main(String[] args) {
	AtomicIntegerFieldUpdater fieldUpdater =
			AtomicIntegerFieldUpdater.newUpdater(Test.class, "field");
	
	Test test = new Test();
	
	fieldUpdater.compareAndSet(test, 0, 10);
	// 修改成功 field = 10
	System.out.println(test.field);
	
	// 修改成功 field = 20
	fieldUpdater.compareAndSet(test, 10, 20);
	System.out.println(test.field);
	
	// 修改失败 field = 20,原来的已经是20了,传入10肯定失败
	fieldUpdater.compareAndSet(test, 10, 30);
	System.out.println(test.field);
	}
}

五、原子累加器

LongAdder,原子累加器,increment()和decrement()分别自增和自减。
但效率比AtomicLong的getAndIncrement()高很多。

性能提升的原因很简单,就是在有竞争时,设置多个累加单元,Therad-0 累加 Cell[0],而 Thread-1 累加Cell[1]… 最后将结果汇总。这样它们在累加时操作的不同的 Cell 变量,因此减少了 CAS 重试失败,从而提高性能。

1.伪共享

LongAdder中有Cell[]数组

// 这个注解,用来防止缓存行伪共享
@sun.misc.Contended
static final class Cell {
	volatile long value;
	Cell(long x) { value = x; }
	
	// 最重要的方法, 用来 cas 方式进行累加, prev 表示旧值, next 表示新值
	final boolean cas(long prev, long next) {
		return UNSAFE.compareAndSwapLong(this, valueOffset, prev, next);
	}
	// 省略不重要代码
}

java 双buffer Map 无锁化_java


因为 Cell 是数组形式,在内存中是连续存储的,一个 Cell 为 24 字节(16 字节的对象头和 8 字节的 value),因此缓存行可以存下 2 个的 Cell 对象。

无论谁修改成功,都会导致对方 Core 的缓存行失效,比如 Core-0 中 Cell[0]=6000, Cell[1]=8000 要累加Cell[0]=6001, Cell[1]=8000 ,这时会让 Core-1 的缓存行失效。

@sun.misc.Contended 用来解决这个问题,它的原理是在使用此注解的对象或字段的前后各增加 128 字节大小的padding,从而让 CPU 将对象预读至缓存时占用不同的缓存行,这样,不会造成对方缓存行的失效。

java 双buffer Map 无锁化_多线程_02

2.LongAdder源码

//关键三个属性
// 累加单元数组, 懒惰初始化
transient volatile Cell[] cells;
// 基础值, 如果没有竞争, 则用 cas 累加这个域
transient volatile long base;
// 在 cells 创建或扩容时, 置为 1, 表示加锁
transient volatile int cellsBusy;

add方法

java 双buffer Map 无锁化_并发编程_03

public void add(long x) {
        // as 为累加单元数组
        // b 为基础值
        // x 为累加值
        Cell[] as; long b, v; int m; Cell a;

        // 进入 if 的两个条件
        // 1. as 有值, 表示已经发生过竞争, 进入 if
        // 2. cas 给 base 累加时失败了, 表示 base 发生了竞争, 进入 if  (casBase是对base累加的cas方法
        if ((as = cells) != null || !casBase(b = base, b + x)) {
            // uncontended 表示 cell 没有竞争
            boolean uncontended = true;
            
            if (
                    // as 还没有创建
                    as == null || (m = as.length - 1) < 0 ||
                            // 当前线程对应的 cell 还没有
                            (a = as[getProbe() & m]) == null ||
                            // cas 给当前线程的 cell 累加失败 uncontended=false ( a 为当前线程的 cell )
                            !(uncontended = a.cas(v = a.value, v + x))
            ) {
                
                // 进入 cell 数组创建、cell 创建的流程
                longAccumulate(x, null, uncontended);
            }
        }
    }

longAccumulate方法

java 双buffer Map 无锁化_java_04

final void longAccumulate(long x, LongBinaryOperator fn,
                              boolean wasUncontended) {
        int h;
        // 当前线程还没有对应的 cell, 需要随机生成一个 h 值用来将当前线程绑定到 cell
        if ((h = getProbe()) == 0) {
            // 初始化 probe
            ThreadLocalRandom.current();
            // h 对应新的 probe 值, 用来对应 cell
            h = getProbe();
            wasUncontended = true;
        }

        // collide 为 true 表示需要扩容
        boolean collide = false;
        for (;;) {
            Cell[] as; Cell a; int n; long v;

            // 1.已经有了 cells
            if ((as = cells) != null && (n = as.length) > 0) {

                // 还没有 cell
                if ((a = as[(n - 1) & h]) == null) {
                    // 为 cellsBusy 加锁, 创建 cell, cell 的初始累加值为 x
                    // 成功则 break, 否则继续 continue 循环
                    if (cellsBusy == 0) {       // Try to attach new Cell
                        Striped64.Cell r = new Striped64.Cell(x);   // Optimistically create
                        if (cellsBusy == 0 && casCellsBusy()) {
                            boolean created = false;
                            try {               // Recheck under lock
                                Striped64.Cell[] rs; int m, j;
                                if ((rs = cells) != null &&
                                        (m = rs.length) > 0 &&
                                        rs[j = (m - 1) & h] == null) {
                                    rs[j] = r;
                                    created = true;
                                }
                            } finally {
                                cellsBusy = 0;
                            }
                            if (created)
                                break;
                            continue;           // Slot is now non-empty
                        }
                    }
                    collide = false;
                }

                // 有竞争, 改变线程对应的 cell 来重试 cas
                //会返回for循环,重新竞争
                else if (!wasUncontended)
                    wasUncontended = true;

                // cas 尝试累加, fn 配合 LongAccumulator 不为 null, 配合 LongAdder 为 null
                else if (a.cas(v = a.value, ((fn == null) ? v + x : fn.applyAsLong(v, x))))
                    break;

                // 如果 cells 长度已经超过了最大长度, 或者已经扩容, 改变线程对应的 cell 来重试 cas
                else if (n >= NCPU || cells != as)
                    collide = false;
                // 确保 collide 为 false 进入此分支, 就不会进入下面的 else if 进行扩容了
                else if (!collide)
                    collide = true;

                // 加锁
                else if (cellsBusy == 0 && casCellsBusy()) {
                // 加锁成功, 扩容
                    //此处省略扩容代码,核心原理和第一个if类似
                    continue;
                }
                // 改变线程对应的 cell
                h = advanceProbe(h);
            }


            // 2.还没有 cells, 尝试给 cellsBusy 加锁
            else if (cellsBusy == 0 && cells == as && casCellsBusy()) {
                // 加锁成功, 初始化 cells, 最开始长度为 2, 并填充一个 cell
                // 成功则 break;
                boolean init = false;
                try {                           // Initialize table
                    if (cells == as) {
                        //懒加载,只加载第一个元素
                        Cell[] rs = new Cell[2];
                        rs[h & 1] = new Cell(x);
                        cells = rs;
                        init = true;
                    }
                } finally {
                    cellsBusy = 0;
                }
                if (init)
                    break;
            }
            
            // 3.上两种情况失败, 尝试给 base 累加
            else if (casBase(v = base, ((fn == null) ? v + x : fn.applyAsLong(v, x))))
                break;
        }
    }

java 双buffer Map 无锁化_多线程_05


java 双buffer Map 无锁化_并发编程_06

sum方法

//累加在一起
public long sum() {
	Cell[] as = cells; Cell a;
	long sum = base;
	if (as != null) {
		for (int i = 0; i < as.length; ++i) {
			if ((a = as[i]) != null)
				sum += a.value;
			}
	}
	return sum;
}

六、Unsafe

Unsafe 对象提供了非常底层的,操作内存、线程的方法,Unsafe 对象不能直接调用,只能通过反射获得。unsafe不是说这个类不安全,而是说我们不要使用它,不规范的使用会不安全。

1.Unsafe CAS操作

public class UnsafeAccessor {
	static Unsafe unsafe;
	static {
		try {
		//通过反射获取theUnsafe
			Field theUnsafe = Unsafe.class.getDeclaredField("theUnsafe");
			theUnsafe.setAccessible(true);
		//由于是静态类,不需要实例对象,传入null即可
			unsafe = (Unsafe) theUnsafe.get(null);
		} catch (NoSuchFieldException | IllegalAccessException e) {
			throw new Error(e);
		}
	}
	static Unsafe getUnsafe() {
		return unsafe;
	}
}
@Data
	class Student {
	volatile int id;
	volatile String name;
}
Unsafe unsafe = UnsafeAccessor.getUnsafe();
Field id = Student.class.getDeclaredField("id");
Field name = Student.class.getDeclaredField("name");

// 获得成员变量的偏移量
long idOffset = UnsafeAccessor.unsafe.objectFieldOffset(id);
long nameOffset = UnsafeAccessor.unsafe.objectFieldOffset(name);
Student student = new Student();

// 使用 cas 方法替换成员变量的值
UnsafeAccessor.unsafe.compareAndSwapInt(student, idOffset, 0, 20); // 成功返回 true
UnsafeAccessor.unsafe.compareAndSwapObject(student, nameOffset, null, "张三"); // 返回 true
System.out.println(student);