LongAdder 类是 LongAccumulator 的一个特例,LongAccumulator 比 LongAdder 的功能更强大。例如下面的构造函数,其中 accumulatorFunction 是一个双目运算器接口,其根据输入的两个参数返回一个计算值,identity 则是 LongAccumulator 累加器的初始值。
public LongAccumulator(LongBinaryOperator accumulatorFunction,long identity) {this.function = accumulatorFunction;base = this.identity = identity;}public interface LongBinaryOperator {//根据两个参数计算并返回一个值long applyAsLong(long left, long right);}
上面提到,LongAdder 其实是 LongAccumulator 的一个特例,调用 LongAdder 就相当于使用下面的方式调用 LongAccumulator:
LongAdder adder = new LongAdder();LongAccumulator accumulator = new LongAccumulator(new LongBinaryOperator() {@Overridepublic long applyAsLong(long left, long right) {return left + right;}}, 0);
LongAccumulator 相比于 LongAdder,可以为累加器提供非 0 的初始值,后者只能提供默认的 0 值。另外,前者还可以指定累加规则,比如不进行累加而进行相乘,只需要在构造 LongAccumulator 时传入自定义的双目运算器即可,后者则内置累加的规则。
从下面代码我们可以知道,LongAccumulator 相比于 LongAdder 的不同在于,在调用 casBase 时后者传递的是 b+x,前者则使用了 r = function.applyAsLong(b = base, x)来计算。
//LongAdder 的 addpublic void add(long x) {Cell[] as; long b, v; int m; Cell a;if ((as = cells) ! = null || ! casBase(b = base, b + x)) {boolean uncontended = true;if (as == null || (m = as.length -1) < 0 ||(a = as[getProbe() & m]) == null ||!(uncontended = a.cas(v = a.value, v + x)))longAccumulate(x, null, uncontended);}}//LongAccumulator 的 accumulatepublic void accumulate(long x) {Cell[] as; long b, v, r; int m; Cell a;if ((as = cells) ! = null ||(r = function.applyAsLong(b = base, x)) ! = b && ! casBase(b, r)) {boolean uncontended = true;if (as == null || (m = as.length -1) < 0 ||(a = as[getProbe() & m]) == null ||!(uncontended =(r = function.applyAsLong(v = a.value, x)) == v ||a.cas(v, r)))longAccumulate(x, function, uncontended);}}
另外,前者在调用 longAccumulate 时传递的是 function,而后者是 null。从下面的代码可知,当 fn 为 null 时就使用 v+x 加法运算,这时候就等价于 LongAdder,当 fn 不为 null 时则使用传递的 fn 函数计算。
else if (casBase(v = base, ((fn == null) ? v + x :fn.applyAsLong(v, x))))break; // Fall back on using base
总结:本节简单介绍了 LongAccumulator 的原理。LongAdder 类是 LongAccumulator 的一个特例,只是后者提供了更加强大的功能,可以让用户自定义累加规则。
