题目:共享资源10000元钱,1000个并发,每个10元,结果共享资源剩余为0
使用synchronized保证线程安全
/** * Description: 使用重量级锁synchronized来保证多线程访问共享资源发生的安全问题 */ public class Test1 { public static void main(String[] args) { Account.demo(new AccountUnsafe(10000));//有锁模式 //Account.demo(new AccountCas(10000));//无锁模式 } } class AccountUnsafe implements Account { private Integer balance; public AccountUnsafe(Integer balance) { this.balance = balance; } @Override public Integer getBalance() { synchronized (this) { return balance; } } @Override public void withdraw(Integer amount) { // 通过这里加锁就可以实现线程安全,不加就会导致线程安全问题 synchronized (this) { balance -= amount; } } } interface Account { // 获取余额 Integer getBalance(); // 取款 void withdraw(Integer amount); /** * Java8之后接口新特性, 可以添加默认方法 * 方法内会启动 1000 个线程,每个线程做 -10 元 的操作 * 如果初始余额为 10000 那么正确的结果应当是 0 */ static void demo(Account account) { List<Thread> ts = new ArrayList<>(); long start = System.nanoTime(); for (int i = 0; i < 1000; i++) { ts.add(new Thread(() -> { account.withdraw(10); })); } ts.forEach(Thread::start); ts.forEach(t -> { try { t.join(); } catch (InterruptedException e) { e.printStackTrace(); } }); long end = System.nanoTime(); System.out.println(account.getBalance() + " cost: " + (end - start) / 1000_000 + " ms"); } }
使用无锁来解决此问题
class AccountCas implements Account { //使用原子整数: 底层使用CAS+重试的机制 private AtomicInteger balance; public AccountCas(int balance) { this.balance = new AtomicInteger(balance); } @Override public Integer getBalance() { //得到原子整数的值 return balance.get(); } @Override public void withdraw(Integer amount) { while(true) { //获得修改前的值 int prev = balance.get(); //获得修改后的值 int next = prev - amount; //比较并设置值 /* 此时的prev为共享变量的值, 如果prev被别的线程改了 也就是说: 自己读到的共享变量的值 和 共享变量最新值 不匹配, 就继续where(true),如果匹配上了, 将next值设置给共享变量. AtomicInteger中value属性, 被volatile修饰, 就是为了确保线程之间共享变量的可见性. */ if(balance.compareAndSet(prev, next)) { break; } } } }
前面看到的AtomicInteger的解决方法,内部并没有用锁来保护共享变量的线程安全。那么它是如何实现的呢?
@Override public void withdraw(Integer amount) { // 核心代码 // 需要不断尝试,直到成功为止 while (true){ // 比如拿到了旧值 1000 int prev = balance.get(); // 在这个基础上 1000-10 = 990 int next = prev - amount; /* compareAndSet 保证操作共享变量安全性的操作: ① 线程A首先获取balance.get(),拿到当前的balance值prev ② 根据这个prev值 - amount值 = 修改后的值next ③ 调用compareAndSet方法, 首先会判断当初拿到的prev值,是否和现在的 balance值相同; 3.1、如果相同,表示其他线程没有修改balance的值, 此时就可以将next值 设置给balance属性 3.2、如果不相同,表示其他线程也修改了balance值, 此时就设置next值失败, 然后一直重试, 重新获取balance.get()的值,计算出next值, 并判断本次的prev和balnce的值是否相同...重复上面操作 */ if (atomicInteger.compareAndSet(prev,next)){ break; } } }
流程 :
注意 :
结合 CAS 和 volatile 可以实现无锁并发,适用于线程数少、多核 CPU 的场景下
public static void main(String[] args) { 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)); // int类型加减乘除操作 // 获取并更新(i = 0, p 为 i 的当前值, 结果 i = -2, 返回 0) // 函数式编程接口,其中函数中的操作能保证原子,但函数需要无副作用 System.out.println(i.getAndUpdate(p -> p - 2)); // int类型加减乘除操作 // 更新并获取(i = -2, p 为 i 的当前值, 结果 i = 0, 返回 0) // 函数式编程接口,其中函数中的操作能保证原子,但函数需要无副作用 System.out.println(i.updateAndGet(p -> p + 2)); }
updateAndGet实现原理
public static void main(String[] args) { AtomicInteger i = new AtomicInteger(5); updateAndGet(i, new IntUnaryOperator() { @Override public int applyAsInt(int operand) { return operand / 2; } }); System.out.println(i.get()); // 2 } public static void updateAndGet(AtomicInteger i, IntUnaryOperator operator) { while (true) { int prev = i.get(); // 5 int next = operator.applyAsInt(prev); if (i.compareAndSet(prev, next)) { break; } } }
为什么需要原子引用类型?保证引用类型的共享变量是线程安全
使用原子引用实现BigDecimal存取款的线程安全
public class Test { //原子BigDecimal引用,总额10000 private static final AtomicReference<BigDecimal> balance = new AtomicReference<>(new BigDecimal("10000")); public static void main(String[] args) { List<Thread> ts = new ArrayList<>(); for (int i = 0; i < 1000; i++) { //循环创建1000个线程 ts.add(new Thread(() -> { //每个线程从总额加去10,减法失败则一直while循环重试 while (true) { BigDecimal prev = balance.get(); BigDecimal next = prev.subtract(new BigDecimal("10")); if (balance.compareAndSet(prev, next)) { break; } } })); } //启动1000个线程 ts.forEach(Thread::start); //等待1000个线程执行完,main线程再执行 ts.forEach(t -> { try { t.join(); } catch (InterruptedException e) { e.printStackTrace(); } }); //获取总额 System.out.println(balance); } }
public class Test3 { public static AtomicReference<String> ref = new AtomicReference<>("A"); public static void main(String[] args) { new Thread(() -> { String pre = ref.get(); System.out.println("change"); try { other(); } catch (InterruptedException e) { e.printStackTrace(); } try { Thread.sleep(1000); } catch (InterruptedException e) { e.printStackTrace(); } //把ref中的A改为C System.out.println("change A->C " + ref.compareAndSet(pre, "C")); }).start(); } public static void other() throws InterruptedException { new Thread(() -> { // 此时ref.get()为A,此时共享变量ref也是A,没有被改过, 此时CAS // 可以修改成功, B System.out.println("change A->B " + ref.compareAndSet(ref.get(), "B")); }).start(); Thread.sleep(500); new Thread(() -> { // 同上, 修改为A System.out.println("change B->A " + ref.compareAndSet(ref.get(), "A")); }).start(); } }
运行结果:
change change A->B true change B->A true change A->C true Process finished with exit code 0
public class Test { //指定版本号 public static AtomicStampedReference<String> ref = new AtomicStampedReference<>("A", 0); public static void main(String[] args) { new Thread(() -> { String pre = ref.getReference(); //获得版本号 int stamp = ref.getStamp(); // 此时的版本号还是第一次获取的 System.out.println("change"); try { other(); } catch (InterruptedException e) { e.printStackTrace(); } try { Thread.sleep(1000); } catch (InterruptedException e) { e.printStackTrace(); } //把ref中的A改为C,并比对版本号,如果版本号相同,就执行替换,并让版本号+1 System.out.println("change A->C stamp " + stamp + ref.compareAndSet(pre, "C", stamp, stamp + 1)); }).start(); } public static void other() throws InterruptedException { new Thread(() -> { int stamp = ref.getStamp(); System.out.println("change A->B stamp " + stamp + ref.compareAndSet(ref.getReference(), "B", stamp, stamp + 1)); }).start(); Thread.sleep(500); new Thread(() -> { int stamp = ref.getStamp(); System.out.println("change B->A stamp " + stamp + ref.compareAndSet(ref.getReference(), "A", stamp, stamp + 1)); }).start(); } }
运行结果:
change change A->B stamp 0true change B->A stamp 1true change A->C stamp 0false Process finished with exit code 0
public class Test { //指定标记,起始为true,当变为fals,则证明修改过 public static AtomicMarkableReference<String> ref = new AtomicMarkableReference<>("A", true); public static void main(String[] args) { new Thread(() -> { String pre = ref.getReference(); System.out.println("change"); try { other(); } catch (InterruptedException e) { e.printStackTrace(); } try { Thread.sleep(1000); } catch (InterruptedException e) { e.printStackTrace(); } //把str中的A改为C,并比对版本号,如果版本号相同,就执行替换,并让版本号+1 System.out.println("change A->C mark " + ref.compareAndSet(pre, "C", true, false)); }).start(); } public static void other() throws InterruptedException { new Thread(() -> { System.out.println("change A->A mark " + ref.compareAndSet( ref.getReference(), "A", true, false)); }).start(); } }
执行结果:
change change A->A mark true change A->C mark false Process finished with exit code 0
上面三个类提供的方法几乎相同,所以我们这里以 AtomicIntegerArray 为例子来介绍
普通数组内元素, 多线程访问造成安全问题
public class AtomicArrayTest { public static void main(String[] args) { demo( () -> new int[10], array -> array.length, (array, index) -> array[index]++, array -> System.out.println(Arrays.toString(array)) ); } /** * 参数1,提供数组、可以是线程不安全数组或线程安全数组 * 参数2,获取数组长度的方法 * 参数3,自增方法,回传 array, index * 参数4,打印数组的方法 */ // supplier 提供者 无中生有 ()->结果 // function 函数 一个参数一个结果 (参数)->结果 , BiFunction (参数1,参数2)->结果 // consumer 消费者 一个参数没结果 (参数)->void, BiConsumer (参数1,参数2)->void private static <T> void demo(Supplier<T> arraySupplier, Function<T, Integer> lengthFun, BiConsumer<T, Integer> putConsumer, Consumer<T> printConsumer) { List<Thread> ts = new ArrayList<>(); T array = arraySupplier.get(); int length = lengthFun.apply(array); for (int i = 0; i < length; i++) { // 创建10个线程, 每个线程对数组作 10000 次操作 ts.add(new Thread(() -> { for (int j = 0; j < 10000; j++) { putConsumer.accept(array, j % length); } })); } // 启动所有线程 ts.forEach(Thread::start); // 等所有线程结束,再执行main线程 ts.forEach(t -> { try { t.join(); } catch (InterruptedException e) { e.printStackTrace(); } }); printConsumer.accept(array); } }
执行结果:
[8374, 8170, 8091, 8031, 8029, 7981, 8276, 8405, 8434, 8504] Process finished with exit code 0
使用AtomicIntegerArray来创建安全数组
demo( ()-> new AtomicIntegerArray(10), (array) -> array.length(), (array, index) -> array.getAndIncrement(index), array -> System.out.println(array) );
执行结果:
[10000, 10000, 10000, 10000, 10000, 10000, 10000, 10000, 10000, 10000] Process finished with exit code 0
保证多线程访问同一个对象的成员变量时, 成员变量的线程安全性
注意:利用字段更新器,可以针对对象的某个域(Field)进行原子操作,只能配合 volatile 修饰的字段使用,否则会出现异常
Exception in thread "main" java.lang.IllegalArgumentException: Must be volatile type
示例
public class AtomicFieldTest { public static void main(String[] args) { Student stu = new Student(); // 获得原子更新器 // 泛型 // 参数1 持有属性的类 参数2 被更新的属性的类 // newUpdater中的参数:第三个为属性的名称 AtomicReferenceFieldUpdater updater = AtomicReferenceFieldUpdater.newUpdater(Student.class, String.class, "name"); // 期望的为null, 如果name属性没有被别的线程更改过, 默认就为null, 此时匹配, 就可以设置name为张三 System.out.println(updater.compareAndSet(stu, null, "张三")); System.out.println(updater.compareAndSet(stu, stu.name, "王五")); System.out.println(stu); } } class Student { volatile String name; @Override public String toString() { return "Student{" + "name='" + name + '\'' + '}'; } }
运行结果:
true true Student{name='王五'} Process finished with exit code 0
累加器性能比较 AtomicLong, LongAddr
public class Test { public static void main(String[] args) { System.out.println("----AtomicLong----"); demo(() -> new AtomicLong(), adder -> adder.getAndIncrement()); System.out.println("----LongAdder----"); demo(() -> new LongAdder(), adder -> adder.increment()); } private static <T> void demo(Supplier<T> adderSupplier, Consumer<T> action) { T adder = adderSupplier.get(); long start = System.nanoTime(); List<Thread> ts = new ArrayList<>(); // 40 个线程,每人累加 50 万 for (int i = 0; i < 40; i++) { ts.add(new Thread(() -> { for (int j = 0; j < 500000; j++) { action.accept(adder); } })); } ts.forEach(Thread::start); ts.forEach(t -> { try { t.join(); } catch (InterruptedException e) { e.printStackTrace(); } }); long end = System.nanoTime(); System.out.println(adder + " 耗时:" + (end - start) / 1000_000); } }
运行结果:
----AtomicLong---- 20000000 耗时:390 ----LongAdder---- 20000000 耗时:44 Process finished with exit code 0
底层的Unsafe实现原子操作
public class Test { public static void main(String[] args) throws Exception { // 通过反射获得Unsafe对象 Class<Unsafe> unsafeClass = Unsafe.class; // 获得构造函数,Unsafe的构造函数为私有的 Constructor<Unsafe> constructor = unsafeClass.getDeclaredConstructor(); // 设置为允许访问私有内容 constructor.setAccessible(true); // 创建Unsafe对象 Unsafe unsafe = constructor.newInstance(); // 创建Person对象 Person person = new Person(); // 获得其属性 name 的偏移量 long nameOffset = unsafe.objectFieldOffset(Person.class.getDeclaredField("name")); long ageOffset = unsafe.objectFieldOffset(Person.class.getDeclaredField("age")); // 通过unsafe的CAS操作改变值 unsafe.compareAndSwapObject(person, nameOffset, null, "guizy"); unsafe.compareAndSwapInt(person, ageOffset, 0, 22); System.out.println(person); } } class Person { // 配合CAS操作,必须用volatile修饰 volatile String name; volatile int age; @Override public String toString() { return "Person{" + "name='" + name + '\'' + ", age=" + age + '}'; } }
运行结果:
Person{name='guizy', age=22} Process finished with exit code 0
以AtomicInteger类incrementAndGet()方法(int+1返回更新值)为例
public class Test { public static void main(String[] args) { AtomicInteger i = new AtomicInteger(10); int i1 = i.incrementAndGet(); System.out.println(i1); } }
运行结果:
11 Process finished with exit code 0
源码分析
public class AtomicInteger extends Number implements java.io.Serializable { private static final long serialVersionUID = 6214790243416807050L; // 获取Unsafe类 private static final Unsafe unsafe = Unsafe.getUnsafe(); // 通过Unsafe类获取value偏移量 private static final long valueOffset; static { try { valueOffset = unsafe.objectFieldOffset (AtomicInteger.class.getDeclaredField("value")); } catch (Exception ex) { throw new Error(ex); } } private volatile int value; /** * 累加1,并返回累加后的int值 */ public final int incrementAndGet() { return unsafe.getAndAddInt(this, valueOffset, 1) + 1; } /** * this:本对象,即toString(),这里即为int值 */ public String toString() { return Integer.toString(get()); } /** * 获取value值 */ public int intValue() { return get(); } }
unsafe.getAndAddInt(this, valueOffset, 1)方法
public final int getAndAddInt(Object var1, long var2, int var4) { int var5; do { //通过value值的偏移量获取value的最新值 var5 = this.getIntVolatile(var1, var2); } while(!this.compareAndSwapInt(var1, var2, var5, var5 + var4)); return var5; }