add zero() and clean AtomicUInt

1 files changed, 8 insertions, 4 deletions

diff --git a/bson/util/atomic_int.h b/bson/util/atomic_int.h
index f4d2749dc2d..d8169da740e 100644
--- a/bson/util/atomic_int.h
+++ b/bson/util/atomic_int.h
@@ -27,14 +27,18 @@ namespace mongo {
     struct AtomicUInt{
         AtomicUInt() : x(0) {}
         AtomicUInt(unsigned z) : x(z) { }
-        volatile unsigned x;
-        operator unsigned() const {
-            return x;
-        }
+
+        operator unsigned() const { return x; }
+        unsigned get() const { return x; }
+
         inline AtomicUInt operator++(); // ++prefix
         inline AtomicUInt operator++(int);// postfix++
         inline AtomicUInt operator--(); // --prefix
         inline AtomicUInt operator--(int); // postfix--
+        
+        inline void zero() { x = 0; } // TODO: this isn't thread safe
+        
+        volatile unsigned x;
     };
 
 #if defined(_WIN32)