【java高性能编程基础】- 实现原子操作的多种方式

x33g5p2x  于2022-02-07 转载在 Java  
字(3.7k)|赞(0)|评价(0)|浏览(543)

概述

所谓原子操作是指不会被线程调度机制打断的操作;这种操作一旦开始,就一直运行到结束,中间不会有任何 context switch (切换到另一个线程)。

下面就以i自增操作为例来实现i的原子性。

i++并不是一步完成的,它包含了三个步骤:

  1. 加载i;
  2. 计算i+1;
  3. 给i赋值。

例如,

  1. public class IaddTest {
  2. public int i = 0;
  3. public void add(){
  4. i++;
  5. }
  6. public static void main(String[] args) throws Exception {
  7. IaddTest iaddTest = new IaddTest();
  8. for(int i=0;i<2;i++){ //创建两个线程
  9. new Thread(new Runnable() {
  10. @Override
  11. public void run() {
  12. for(int i=0;i<10000;i++){
  13. iaddTest.add();
  14. }
  15. }
  16. }).start();
  17. }
  18. Thread.sleep(2000L);//sleep两秒,等待两个线程任务全部执行完成
  19. System.out.println(iaddTest.i);
  20. }
  21. }

这段代码中,创建了两个线程分别调用10000次add方法(即两万次i++),理论上最后的输出结果应该为20000,但是实际的运行结果却达不到20000,如下:

这是由于两个线程是并发执行的,当线程1读取i值时,线程2还没有将i+1的值赋给i(例如当i = 1时,线程1读取i=1,同时线程2也读取i=1,然后执行+1操作,结果两个线程都将i赋值为2)。

为了避免这种线程不安全情况的发生,保证程序的正确性,我们就需要将i++进行原子操作。

下面,就通过几种不同的方式来实现原子操作。

1、加锁

1.1、Lock接口

分别在i的前后进行加锁和解锁,实现i的原子操作。


完整示例代码如下:

  1. public class IaddTest {
  2. public int i = 0;
  3. private Lock lock = new ReentrantLock(false);
  4. public void add(){
  5. try{
  6. lock.lock(); //加锁
  7. i++;
  8. }catch (Exception e){
  9. e.printStackTrace();
  10. }finally {
  11. lock.unlock(); //解锁
  12. }
  13. }
  14. public static void main(String[] args) throws Exception {
  15. IaddTest iaddTest = new IaddTest();
  16. for(int i=0;i<2;i++){ //创建两个线程
  17. new Thread(new Runnable() {
  18. @Override
  19. public void run() {
  20. for(int i=0;i<10000;i++){
  21. iaddTest.add();
  22. }
  23. }
  24. }).start();
  25. }
  26. Thread.sleep(2000L);//sleep两秒,等待两个线程任务全部执行完成
  27. System.out.println("i:" + iaddTest.i);
  28. }
  29. }

再来看看执行结果:


正确输出20000。

1.2、synchronized关键字

  1. 给add方法加synchronized

  1. 给i++块加synchronized

完整示例代码:

  1. public class IaddTest {
  2. public int i = 0;
  3. private Lock lock = new ReentrantLock(false);
  4. // public synchronized void add() //给add方法加锁
  5. public void add(){
  6. synchronized (this){ //给i++加锁
  7. i++;
  8. }
  9. }
  10. public static void main(String[] args) throws Exception {
  11. IaddTest iaddTest = new IaddTest();
  12. for(int i=0;i<2;i++){ //创建两个线程
  13. new Thread(new Runnable() {
  14. @Override
  15. public void run() {
  16. for(int i=0;i<10000;i++){
  17. iaddTest.add();
  18. }
  19. }
  20. }).start();
  21. }
  22. Thread.sleep(2000L);//sleep两秒,等待两个线程任务全部执行完成
  23. System.out.println("i:" + iaddTest.i);
  24. }
  25. }

执行结果:


正确输出20000。

2、CAS机制

CAS是英文单词Compare and Swap的缩写,翻译过来就是比较并替换。

CAS机制中使用了3个基本操作数:内存地址V,旧的预期值A,要修改的新值B。更新一个变量的时候,只有当变量的预期值A和内存地址V当中的实际值相同时,才会将内存地址V对应的值修改为B。

sun.misc.Unsafe中提供了CAS操作的本地方法。


java中认为Unsafe类是不安全的类,所以不能直接通过构造函数构造对象,需要使用反射机制。

  1. private static Unsafe unsafe;
  2. // 通过反射获取theUnsafe属性
  3. Field field = Unsafe.class.getDeclaredField("theUnsafe");
  4. // theUnsafe是private的,设置setAccessible(true)
  5. field.setAccessible(true);
  6. // 获取Unsafe对象
  7. unsafe = (Unsafe) field.get(null);

CAS操作完整示例代码:

  1. public class IaddTest {
  2. public int i = 0;
  3. private static Unsafe unsafe;
  4. private static long fieldOffset;
  5. static {
  6. try {
  7. // 通过反射获取theUnsafe属性
  8. Field field = Unsafe.class.getDeclaredField("theUnsafe");
  9. // theUnsafe是private的,设置setAccessible(true)
  10. field.setAccessible(true);
  11. // 获取Unsafe对象
  12. unsafe = (Unsafe) field.get(null);
  13. // 获取属性i在内存中的实际地址
  14. fieldOffset = unsafe.objectFieldOffset(IaddTest.class.getField("i"));
  15. } catch (Exception e) {
  16. e.printStackTrace();
  17. }
  18. }
  19. public void add(){
  20. int current;
  21. do {
  22. current = unsafe.getIntVolatile(this,fieldOffset);
  23. }while (!unsafe.compareAndSwapInt(this,fieldOffset,i,i+1));
  24. }
  25. public static void main(String[] args) throws Exception {
  26. IaddTest iaddTest = new IaddTest();
  27. for(int i=0;i<2;i++){ //创建两个线程
  28. new Thread(new Runnable() {
  29. @Override
  30. public void run() {
  31. for(int i=0;i<10000;i++){
  32. iaddTest.add();
  33. }
  34. }
  35. }).start();
  36. }
  37. Thread.sleep(2000L);//sleep两秒,等待两个线程任务全部执行完成
  38. System.out.println("i:" + iaddTest.i);
  39. }
  40. }

执行结果:


正确输出20000。

3、原子类(Atomic)

原子类是jdk提供的一些进行原子操作的类,位于java.util.concurrent.atomic包中。

原子类的底层也是通过调用com.misc.Unsafe类中的CAS机制实现的。

使用时我们只需要创建一个原子类对象,并调用其中的方法即可保证代码的原子性。

  1. public class IaddTest {
  2. AtomicInteger i = new AtomicInteger(0);
  3. public void add(){
  4. i.incrementAndGet();
  5. }
  6. public static void main(String[] args) throws Exception {
  7. IaddTest iaddTest = new IaddTest();
  8. for(int i=0;i<2;i++){ //创建两个线程
  9. new Thread(new Runnable() {
  10. @Override
  11. public void run() {
  12. for(int i=0;i<10000;i++){
  13. iaddTest.add();
  14. }
  15. }
  16. }).start();
  17. }
  18. Thread.sleep(2000L);//sleep两秒,等待两个线程任务全部执行完成
  19. System.out.println("i:" + iaddTest.i);
  20. }
  21. }

执行结果:


正确输出20000。

相关文章

最新文章

更多