题目:共享资源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;
}