多线程环境下,如果操作共享资源,需要做额外的并发控制。共享资源不仅包含自定义的Java对象,也包含基本类型的数据。
而多个线程同时操作一个不加控制的共享资源时,会得到意想不到结果。
必要性
以int为例,我们定义一个共享资源类
public class IntegerResouce {
private int i;
public int getAndIncrement() {
int pre = i;
i = i + 1;
return pre;
}
public int getAndDecrement() {
int pre = i;
i = i - 1;
return pre;
}
public int getI() {
return i;
}
}
然后起两个线程同时访问i,一个线程做1000次递增操作,一个线程做1000次递减操作,理论上最后 i 的值应该是0。
public class IntegerTest {
public static void main(String[] args) {
IntegerResouce resouce = new IntegerResouce();
Thread thread1 = new Thread(new Runnable() {
@Override
public void run() {
for (int i = 0; i < 1000; i++) {
resouce.getAndIncrement();
}
}
});
Thread thread2 = new Thread(new Runnable() {
@Override
public void run() {
for (int i = 0; i < 1000; i++) {
resouce.getAndDecrement();
}
}
});
thread1.start();
thread2.start();
try {
thread1.join();
thread2.join();
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println(resouce.getI());
}
}
运行多次后发现结果各不相同,正数、负数、0都有可能。
这是因为 i = i + 1虽然是一个简单的计算表达式,但对于线程来说需要先获取i,再做运算,最后把值赋给i,所以说这里的i并不是线程安全的。(即便是更短小的i++也不是线程安全的)
下面我们对共享资源做一些访问控制,对 i 的所有访问都加上锁:
public class IntegerResouce {
private int i;
public synchronized int getAndIncrement() {
int pre = i;
i = i + 1;
return pre;
}
public synchronized int getAndDecrement() {
int pre = i;
i = i - 1;
return pre;
}
public synchronized int getI() {
return i;
}
}
这样不管运行多少次,结果都是0了。但上述的解决方案并不完美,synchronized太重了,效率偏低。
这时候就可以用JUC包下的AtomicInteger了
public class IntegerResouce {
private AtomicInteger i = new AtomicInteger(0);
public int getAndIncrement() {
return i.getAndIncrement();
}
public int getAndDecrement() {
return i.getAndDecrement();
}
public int getI() {
return i.get();
}
}
多次运行,结果也是正确的。
实现原理
为啥说AtomicInteger比我们直接用synchronized对 i 做访问控制更高效呢?
因为AtomicInteger通过乐观锁(compare and swap)来保证原子性。
AtomicInteger继承自Number,用int类型的value保存数值,并持有Unsafe对象。
Unsafe类如其名,存放着一写不太安全的方法,可以直接访问系统内存资源、自主管理内资源、直接调用CPU指令等等。
Atomic相关类主要用到这几个方法:
以getAndIncrement为例, 直接调用unsafe的getAndSet方法
Unsafe通过失败重试的方式完成CAS操作。
使用场景距离
在使用websocket来做后台推送时,可能需要一个 静态 int变量来保存在线人数,这里的onlineCount就是属于共享变量,可以用AtomicInteger保存。