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

举报
1+1=王 发表于 2022/12/21 16:22:49 2022/12/21
【摘要】 【java高性能编程基础】- 实现原子操作的多种方式

@[TOC]

概述

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

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

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

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

例如,

public class IaddTest {

    public int i = 0;

    public void add(){
        i++;
    }

    public static void main(String[] args) throws Exception {
        IaddTest iaddTest = new IaddTest();

        for(int i=0;i<2;i++){       //创建两个线程
            new Thread(new Runnable() {
                @Override
                public void run() {
                    for(int i=0;i<10000;i++){
                        iaddTest.add();
                    }
                }
            }).start();
        }

        Thread.sleep(2000L);//sleep两秒,等待两个线程任务全部执行完成
        System.out.println(iaddTest.i);
    }
}

这段代码中,创建了两个线程分别调用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++的原子操作。
在这里插入图片描述完整示例代码如下:

public class IaddTest {

    public int i = 0;
    private Lock lock = new ReentrantLock(false);

    public void add(){
        try{
            lock.lock();	//加锁
            i++;
        }catch (Exception e){
            e.printStackTrace();
        }finally {
            lock.unlock();	//解锁
        }

    }

    public static void main(String[] args) throws Exception {
        IaddTest iaddTest = new IaddTest();

        for(int i=0;i<2;i++){       //创建两个线程
            new Thread(new Runnable() {
                @Override
                public void run() {
                    for(int i=0;i<10000;i++){
                        iaddTest.add();
                    }
                }
            }).start();
        }

        Thread.sleep(2000L);//sleep两秒,等待两个线程任务全部执行完成
        System.out.println("i:" + iaddTest.i);
    }
}

再来看看执行结果:
在这里插入图片描述正确输出20000。

1.2、synchronized关键字

  1. 给add方法加synchronized
    在这里插入图片描述

  2. 给i++块加synchronized
    在这里插入图片描述
    完整示例代码:

public class IaddTest {

    public int i = 0;
    private Lock lock = new ReentrantLock(false);

//    public synchronized void add()    //给add方法加锁
    public void add(){
        synchronized (this){            //给i++加锁
            i++;
        }
    }

    public static void main(String[] args) throws Exception {
        IaddTest iaddTest = new IaddTest();

        for(int i=0;i<2;i++){       //创建两个线程
            new Thread(new Runnable() {
                @Override
                public void run() {
                    for(int i=0;i<10000;i++){
                        iaddTest.add();
                    }
                }
            }).start();
        }

        Thread.sleep(2000L);//sleep两秒,等待两个线程任务全部执行完成
        System.out.println("i:" + iaddTest.i);
    }
}

执行结果:
在这里插入图片描述正确输出20000。

2、CAS机制

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

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

sun.misc.Unsafe中提供了CAS操作的本地方法。
在这里插入图片描述java中认为Unsafe类是不安全的类,所以不能直接通过构造函数构造对象,需要使用==反射机制==。
在这里插入图片描述

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

CAS操作完整示例代码:

public class IaddTest {

    public int i = 0;
    private static Unsafe unsafe;
    private static long fieldOffset;

    static {
        try {
            // 通过反射获取theUnsafe属性
            Field field = Unsafe.class.getDeclaredField("theUnsafe");
            // theUnsafe是private的,设置setAccessible(true)
            field.setAccessible(true);
            // 获取Unsafe对象
            unsafe = (Unsafe) field.get(null);

            // 获取属性i在内存中的实际地址
            fieldOffset = unsafe.objectFieldOffset(IaddTest.class.getField("i"));
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public void add(){
        int current;
        do {
            current = unsafe.getIntVolatile(this,fieldOffset);
        }while (!unsafe.compareAndSwapInt(this,fieldOffset,i,i+1));
    }

    public static void main(String[] args) throws Exception {
        IaddTest iaddTest = new IaddTest();

        for(int i=0;i<2;i++){       //创建两个线程
            new Thread(new Runnable() {
                @Override
                public void run() {
                    for(int i=0;i<10000;i++){
                        iaddTest.add();
                    }
                }
            }).start();
        }

        Thread.sleep(2000L);//sleep两秒,等待两个线程任务全部执行完成
        System.out.println("i:" + iaddTest.i);
    }
}

执行结果:
在这里插入图片描述正确输出20000。

3、原子类(Atomic)

原子类是jdk提供的一些进行原子操作的类,位于java.util.concurrent.atomic包中。
在这里插入图片描述
原子类的底层也是通过调用com.misc.Unsafe类中的CAS机制实现的。

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

public class IaddTest {

    AtomicInteger i = new AtomicInteger(0);

    public void add(){
        i.incrementAndGet();
    }

    public static void main(String[] args) throws Exception {
        IaddTest iaddTest = new IaddTest();

        for(int i=0;i<2;i++){       //创建两个线程
            new Thread(new Runnable() {
                @Override
                public void run() {
                    for(int i=0;i<10000;i++){
                        iaddTest.add();
                    }
                }
            }).start();
        }

        Thread.sleep(2000L);//sleep两秒,等待两个线程任务全部执行完成
        System.out.println("i:" + iaddTest.i);
    }
}

执行结果:
在这里插入图片描述正确输出20000。

【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。