java代码示例：采用 CAS + 失败重试来保障更新操作的原子性

深碍√TFBOYSˉ_ 2024-02-17 12:19 55阅读 0赞

下面是一个使用CAS（Compare and Swap）和失败重试来保障更新操作原子性的Java代码示例：

import java.util.concurrent.atomic.AtomicInteger;
    
    public class AtomicUpdateExample {
        
        private static AtomicInteger counter = new AtomicInteger(0);
    
        public static void main(String[] args) {
        
            Thread thread1 = new Thread(new UpdateTask());
            Thread thread2 = new Thread(new UpdateTask());
    
            thread1.start();
            thread2.start();
    
            try {
        
                thread1.join();
                thread2.join();
            } catch (InterruptedException e) {
        
                e.printStackTrace();
            }
    
            System.out.println("Final value: " + counter.get());
        }
    
        static class UpdateTask implements Runnable {
        
            private static final int MAX_RETRIES = 10;
    
            @Override
            public void run() {
        
                int oldValue, newValue;
                int retries = 0;
    
                do {
        
                    oldValue = counter.get();
                    newValue = oldValue + 1;
                    retries++;
                } while (!counter.compareAndSet(oldValue, newValue) && retries < MAX_RETRIES);
    
                if (retries >= MAX_RETRIES) {
        
                    System.out.println("Update failed");
                } else {
        
                    System.out.println("Update succeeded");
                }
            }
        }
    }

在上述示例中，我们使用`AtomicInteger`作为计数器，并创建两个线程来对计数器进行更新操作。每个线程通过CAS操作（`compareAndSet`方法）尝试更新计数器的值。如果CAS操作失败，说明有其他线程已经修改了计数器的值，此时线程会进行失败重试，直到更新成功或达到最大重试次数（`MAX_RETRIES`）为止。

在主线程中，我们启动两个更新任务线程，并等待它们完成。最后输出计数器的最终值。

这种方式使用CAS和失败重试来保障更新操作的原子性，即只有一个线程能成功更新计数器的值，其他线程会进行重试直到成功。失败重试的次数可以根据实际需求进行调整。

需要注意的是，虽然CAS操作可以保障更新操作的原子性，但在高并发环境下，频繁的失败重试可能会导致性能下降。因此，在选择使用CAS和失败重试时，需要根据具体场景评估并发访问的情况和性能要求。