很多情况下我们只是需要简单的,高效,线程安全的递增递减方法。注意,这里有三个条件:简单,意味着程序员尽可能少的底层或者实现起来比较简单;高效,意味着耗用资源要少,程序处理速度要快; 线程安全也非常重要,这个在多线程下能保证数据的正确性。这三个条件看起来比较简单,但是实现起来却难以令人满意。
通常情况下,在Java里面, ++i或者-- i不是线程安全的,这里面有三个独立的操作:读取变量当前值,为该值+1 /-1,然后写会新的值。在没有额外资源可以利用的情况下,只能使用加锁才能保证读-改-写这三个操作证的‘"原子性"。
在J.U.C(Doug Lea)为加入jdk之前,是采用纯Java实现的,于是不可避免的采用了synchronized关键字。
public final synchronized void set(int newValue); public final synchronized int getAndSet(int newValue); public final synchronized int incrementAndGet();
同时在变量上使用了volatile(后面详述)来保证get()的时候不用加锁。尽管synchronized的代价还是很高,但是在没有JNI的手段下纯Java语言还是不能实现此操作的。
来看看java.util.cocurrent.atomic.AtomicInteger开始:
int addAndGet(int delta);
//以原子方式将给定值与当前值想家。实际上就是等于线程安全版本的 i += delta boolean compareAndSet(int expect, int update);
//如果当前值 == 预期值,则以原子方式将该值设置为给定的更新值。如果
//更新成功就返回true, 否则就返回false, 并且不修改原值 int decrementAndGet();
//以原子方式将当前值减1,相当于线程安全版本的--i操作。 int get(); //获得当前值。 int getAndAdd(int delta);
//以原子方式将给定值与当前值相加,相当于线程版本的 t = i; i += //delat; return t; int getAndDecrement();
//以原子方式将当前值减1,相当于线程安全版本的i--; int getAndIncrement();
//以原子方式将当前值加1,相当于线程安全版本的i++; int getAndSet();
//以原子方式设定给定值,并返回旧值; int incrementAndGet();
//以原子方式将当前值加1,相当于线程安全版本的++i;
看几个实现:
private static final Unsafe unsafe = Unsafe.getUnsafe();
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;
调用sun.misc.Unsafe类实现cas操作(历史版权原因,Unsafe源码看不了),获取该对象value成员的内存地址;
public final boolean compareAndSet(int expect, int update) {
return unsafe.compareAndSwapInt(this, valueOffset, expect, update); //根据unsafe的cas操作,返回true or false
}
看一下上面API的实现:
public final int getAndIncrement() {
for (;;) {
int current = get();
int next = current + 1;
if (compareAndSet(current, next))
return current;
}
} public final int getAndDecrement() {
for (;;) {
int current = get();
int next = current - 1;
if (compareAndSet(current, next))
return current;
}
} public final int getAndAdd(int delta) {
for (;;) {
int current = get();
int next = current + delta;
if (compareAndSet(current, next))
return current;
}
} public final int incrementAndGet() {
for (;;) {
int current = get();
int next = current + 1;
if (compareAndSet(current, next))
return next;
}
}
方法实现总体差不多,for(;;)死循环,读取当前值,根据相应的加减cas操作,成功从for中返回,失败不同重试;
测试用例如下:
import java.util.concurrent.atomic.AtomicInteger; import org.junit.Test; import static org.junit.Assert.*;
public class AtomicIntegerTest { @Test
public void testAll() throws InterruptedException {
final AtomicInteger value = new AtomicInteger(10);
assertEquals(value.compareAndSet(1, 2), false);
assertEquals(value.get(), 10);
assertTrue(value.compareAndSet(10, 3));
assertEquals(value.get(), 3);
value.set(0); //
assertEquals(value.incrementAndGet(), 1);
assertEquals(value.getAndAdd(2), 1);
assertEquals(value.getAndSet(5), 3);
assertEquals(value.get(), 5); // final
final int threadSize = 5;
Thread[] threads = new Thread[threadSize];
for (int i = 0; i < threadSize; i++) {
threads[i] = new Thread() {
public void run() {
value.incrementAndGet();
}
};
}
for (Thread t : threads) {
t.start();
} for (Thread t : threads) {
t.join();
}
//
assertEquals(value.get(), 5 + threadSize);
}
}
原文来至:http://www.blogjava.net/xylz/archive/2010/07/08/325587.html