1 /*
   2  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   3  *
   4  * This code is free software; you can redistribute it and/or modify it
   5  * under the terms of the GNU General Public License version 2 only, as
   6  * published by the Free Software Foundation.  Oracle designates this
   7  * particular file as subject to the "Classpath" exception as provided
   8  * by Oracle in the LICENSE file that accompanied this code.
   9  *
  10  * This code is distributed in the hope that it will be useful, but WITHOUT
  11  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  12  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  13  * version 2 for more details (a copy is included in the LICENSE file that
  14  * accompanied this code).
  15  *
  16  * You should have received a copy of the GNU General Public License version
  17  * 2 along with this work; if not, write to the Free Software Foundation,
  18  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  19  *
  20  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  21  * or visit www.oracle.com if you need additional information or have any
  22  * questions.
  23  */
  24 
  25 /*
  26  * This file is available under and governed by the GNU General Public
  27  * License version 2 only, as published by the Free Software Foundation.
  28  * However, the following notice accompanied the original version of this
  29  * file:
  30  *
  31  * Written by Doug Lea with assistance from members of JCP JSR-166
  32  * Expert Group and released to the public domain, as explained at
  33  * http://creativecommons.org/publicdomain/zero/1.0/
  34  */
  35 
  36 package java.util.concurrent.atomic;
  37 
  38 import java.io.Serializable;
  39 
  40 /**
  41  * One or more variables that together maintain an initially zero
  42  * {@code long} sum.  When updates (method {@link #add}) are contended
  43  * across threads, the set of variables may grow dynamically to reduce
  44  * contention. Method {@link #sum} (or, equivalently, {@link
  45  * #longValue}) returns the current total combined across the
  46  * variables maintaining the sum.
  47  *
  48  * <p>This class is usually preferable to {@link AtomicLong} when
  49  * multiple threads update a common sum that is used for purposes such
  50  * as collecting statistics, not for fine-grained synchronization
  51  * control.  Under low update contention, the two classes have similar
  52  * characteristics. But under high contention, expected throughput of
  53  * this class is significantly higher, at the expense of higher space
  54  * consumption.
  55  *
  56  * <p>LongAdders can be used with a {@link
  57  * java.util.concurrent.ConcurrentHashMap} to maintain a scalable
  58  * frequency map (a form of histogram or multiset). For example, to
  59  * add a count to a {@code ConcurrentHashMap<String,LongAdder> freqs},
  60  * initializing if not already present, you can use {@code
  61  * freqs.computeIfAbsent(key, k -> new LongAdder()).increment();}
  62  *
  63  * <p>This class extends {@link Number}, but does <em>not</em> define
  64  * methods such as {@code equals}, {@code hashCode} and {@code
  65  * compareTo} because instances are expected to be mutated, and so are
  66  * not useful as collection keys.
  67  *
  68  * @since 1.8
  69  * @author Doug Lea
  70  */
  71 public class LongAdder extends Striped64 implements Serializable {
  72     private static final long serialVersionUID = 7249069246863182397L;
  73 
  74     /**
  75      * Creates a new adder with initial sum of zero.
  76      */
  77     public LongAdder() {
  78     }
  79 
  80     /**
  81      * Adds the given value.
  82      *
  83      * @param x the value to add
  84      */
  85     public void add(long x) {
  86         Cell[] cs; long b, v; int m; Cell c;
  87         if ((cs = cells) != null || !casBase(b = base, b + x)) {
  88             boolean uncontended = true;
  89             if (cs == null || (m = cs.length - 1) < 0 ||
  90                 (c = cs[getProbe() & m]) == null ||
  91                 !(uncontended = c.cas(v = c.value, v + x)))
  92                 longAccumulate(x, null, uncontended);
  93         }
  94     }
  95 
  96     /**
  97      * Equivalent to {@code add(1)}.
  98      */
  99     public void increment() {
 100         add(1L);
 101     }
 102 
 103     /**
 104      * Equivalent to {@code add(-1)}.
 105      */
 106     public void decrement() {
 107         add(-1L);
 108     }
 109 
 110     /**
 111      * Returns the current sum.  The returned value is <em>NOT</em> an
 112      * atomic snapshot; invocation in the absence of concurrent
 113      * updates returns an accurate result, but concurrent updates that
 114      * occur while the sum is being calculated might not be
 115      * incorporated.
 116      *
 117      * @return the sum
 118      */
 119     public long sum() {
 120         Cell[] cs = cells;
 121         long sum = base;
 122         if (cs != null) {
 123             for (Cell c : cs)
 124                 if (c != null)
 125                     sum += c.value;
 126         }
 127         return sum;
 128     }
 129 
 130     /**
 131      * Resets variables maintaining the sum to zero.  This method may
 132      * be a useful alternative to creating a new adder, but is only
 133      * effective if there are no concurrent updates.  Because this
 134      * method is intrinsically racy, it should only be used when it is
 135      * known that no threads are concurrently updating.
 136      */
 137     public void reset() {
 138         Cell[] cs = cells;
 139         base = 0L;
 140         if (cs != null) {
 141             for (Cell c : cs)
 142                 if (c != null)
 143                     c.reset();
 144         }
 145     }
 146 
 147     /**
 148      * Equivalent in effect to {@link #sum} followed by {@link
 149      * #reset}. This method may apply for example during quiescent
 150      * points between multithreaded computations.  If there are
 151      * updates concurrent with this method, the returned value is
 152      * <em>not</em> guaranteed to be the final value occurring before
 153      * the reset.
 154      *
 155      * @return the sum
 156      */
 157     public long sumThenReset() {
 158         Cell[] cs = cells;
 159         long sum = getAndSetBase(0L);
 160         if (cs != null) {
 161             for (Cell c : cs) {
 162                 if (c != null)
 163                     sum += c.getAndSet(0L);
 164             }
 165         }
 166         return sum;
 167     }
 168 
 169     /**
 170      * Returns the String representation of the {@link #sum}.
 171      * @return the String representation of the {@link #sum}
 172      */
 173     public String toString() {
 174         return Long.toString(sum());
 175     }
 176 
 177     /**
 178      * Equivalent to {@link #sum}.
 179      *
 180      * @return the sum
 181      */
 182     public long longValue() {
 183         return sum();
 184     }
 185 
 186     /**
 187      * Returns the {@link #sum} as an {@code int} after a narrowing
 188      * primitive conversion.
 189      */
 190     public int intValue() {
 191         return (int)sum();
 192     }
 193 
 194     /**
 195      * Returns the {@link #sum} as a {@code float}
 196      * after a widening primitive conversion.
 197      */
 198     public float floatValue() {
 199         return (float)sum();
 200     }
 201 
 202     /**
 203      * Returns the {@link #sum} as a {@code double} after a widening
 204      * primitive conversion.
 205      */
 206     public double doubleValue() {
 207         return (double)sum();
 208     }
 209 
 210     /**
 211      * Serialization proxy, used to avoid reference to the non-public
 212      * Striped64 superclass in serialized forms.
 213      * @serial include
 214      */
 215     private static class SerializationProxy implements Serializable {
 216         private static final long serialVersionUID = 7249069246863182397L;
 217 
 218         /**
 219          * The current value returned by sum().
 220          * @serial
 221          */
 222         private final long value;
 223 
 224         SerializationProxy(LongAdder a) {
 225             value = a.sum();
 226         }
 227 
 228         /**
 229          * Returns a {@code LongAdder} object with initial state
 230          * held by this proxy.
 231          *
 232          * @return a {@code LongAdder} object with initial state
 233          * held by this proxy
 234          */
 235         private Object readResolve() {
 236             LongAdder a = new LongAdder();
 237             a.base = value;
 238             return a;
 239         }
 240     }
 241 
 242     /**
 243      * Returns a
 244      * <a href="{@docRoot}/serialized-form.html#java.util.concurrent.atomic.LongAdder.SerializationProxy">
 245      * SerializationProxy</a>
 246      * representing the state of this instance.
 247      *
 248      * @return a {@link SerializationProxy}
 249      * representing the state of this instance
 250      */
 251     private Object writeReplace() {
 252         return new SerializationProxy(this);
 253     }
 254 
 255     /**
 256      * @param s the stream
 257      * @throws java.io.InvalidObjectException always
 258      */
 259     private void readObject(java.io.ObjectInputStream s)
 260         throws java.io.InvalidObjectException {
 261         throw new java.io.InvalidObjectException("Proxy required");
 262     }
 263 
 264 }