Add yangtools.concepts.CheckedValue

[yangtools.git] / common / concepts / src / main / java / org / opendaylight / yangtools / concepts / AbstractRegistration.java

diff --git a/common/concepts/src/main/java/org/opendaylight/yangtools/concepts/AbstractRegistration.java b/common/concepts/src/main/java/org/opendaylight/yangtools/concepts/AbstractRegistration.java
index 50f19fc9e3ef596c56754909889b3d308a66e673..f0e6238e339a54dc52f1b576ac22e601f5509178 100644 (file)
--- a/common/concepts/src/main/java/org/opendaylight/yangtools/concepts/AbstractRegistration.java
+++ b/common/concepts/src/main/java/org/opendaylight/yangtools/concepts/AbstractRegistration.java
@@ -7,28 +7,47 @@
  */
 package org.opendaylight.yangtools.concepts;
 
-import java.util.concurrent.atomic.AtomicBoolean;
+import com.google.common.base.MoreObjects;
+import com.google.common.base.MoreObjects.ToStringHelper;
+import java.util.concurrent.atomic.AtomicIntegerFieldUpdater;
 
 /**
- * Utility registration handle. It is a convenience for register-style method
- * which can return an AutoCloseable realized by a subclass of this class.
- * Invoking the close() method triggers unregistration of the state the method
- * installed.
+ * Utility registration handle. It is a convenience for register-style method which can return an AutoCloseable realized
+ * by a subclass of this class. Invoking the close() method triggers unregistration of the state the method installed.
  */
-public abstract class AbstractRegistration implements AutoCloseable {
-    private AtomicBoolean closed = new AtomicBoolean();
-    
+public abstract class AbstractRegistration implements Registration {
+    private static final AtomicIntegerFieldUpdater<AbstractRegistration> CLOSED_UPDATER =
+            AtomicIntegerFieldUpdater.newUpdater(AbstractRegistration.class, "closed");
+    private volatile int closed = 0;
+
     /**
-     * Remove the state referenced by this registration. This method is
-     * guaranteed to be called at most once. The referenced state must be
-     * retained until this method is invoked.
+     * Remove the state referenced by this registration. This method is guaranteed to be called at most once.
+     * Referenced state must be retained until this method is invoked.
      */
     protected abstract void removeRegistration();
 
+    /**
+     * Query the state of this registration. Returns true if it was closed.
+     *
+     * @return true if the registration was closed, false otherwise.
+     */
+    public final boolean isClosed() {
+        return closed != 0;
+    }
+
     @Override
     public final void close() {
-        if (closed.compareAndSet(false, true)) {
+        if (CLOSED_UPDATER.compareAndSet(this, 0, 1)) {
             removeRegistration();
         }
     }
+
+    @Override
+    public final String toString() {
+        return addToStringAttributes(MoreObjects.toStringHelper(this).omitNullValues()).toString();
+    }
+
+    protected ToStringHelper addToStringAttributes(final ToStringHelper toStringHelper) {
+        return toStringHelper.add("closed", closed);
+    }
 }