Migrate to tech.pantheon.TrieMap

[yangtools.git] / common / util / src / main / java / org / opendaylight / yangtools / util / ReadOnlyTrieMap.java

diff --git a/common/util/src/main/java/org/opendaylight/yangtools/util/ReadOnlyTrieMap.java b/common/util/src/main/java/org/opendaylight/yangtools/util/ReadOnlyTrieMap.java
index 88da121c8616f44aba44d633d9e599e1396fc5a4..c0402840b7c7563816b638337c10ce06b41da7cd 100644 (file)
--- a/common/util/src/main/java/org/opendaylight/yangtools/util/ReadOnlyTrieMap.java
+++ b/common/util/src/main/java/org/opendaylight/yangtools/util/ReadOnlyTrieMap.java
@@ -7,14 +7,15 @@
  */
 package org.opendaylight.yangtools.util;
 
-import java.util.Map;
+import static java.util.Objects.requireNonNull;
 
+import com.google.common.collect.ForwardingMap;
+import java.util.Map;
+import java.util.concurrent.atomic.AtomicReferenceFieldUpdater;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
-
-import com.google.common.base.Preconditions;
-import com.google.common.collect.ForwardingMap;
-import com.romix.scala.collection.concurrent.TrieMap;
+import tech.pantheon.triemap.ImmutableTrieMap;
+import tech.pantheon.triemap.MutableTrieMap;
 
 /**
  * A read-only facade in front of a TrieMap. This is what we give out from
@@ -24,34 +25,35 @@ import com.romix.scala.collection.concurrent.TrieMap;
  * changes, we can cache it for future reference.
  */
 final class ReadOnlyTrieMap<K, V> extends ForwardingMap<K, V> {
+    @SuppressWarnings("rawtypes")
+    private static final AtomicReferenceFieldUpdater<ReadOnlyTrieMap, ImmutableTrieMap> UPDATER =
+            AtomicReferenceFieldUpdater.newUpdater(ReadOnlyTrieMap.class, ImmutableTrieMap.class, "readOnly");
     private static final Logger LOG = LoggerFactory.getLogger(ReadOnlyTrieMap.class);
-    private final TrieMap<K, V> readWrite;
+    private final MutableTrieMap<K, V> readWrite;
     private final int size;
-    private TrieMap<K, V> readOnly;
+    private volatile ImmutableTrieMap<K, V> readOnly;
 
-    ReadOnlyTrieMap(final TrieMap<K, V> map, final int size) {
-        super();
-        this.readWrite = Preconditions.checkNotNull(map);
+    ReadOnlyTrieMap(final MutableTrieMap<K, V> map, final int size) {
+        this.readWrite = requireNonNull(map);
         this.size = size;
     }
 
     Map<K, V> toReadWrite() {
-        final Map<K, V> ret = new ReadWriteTrieMap<>(readWrite.snapshot(), size);
+        final Map<K, V> ret = new ReadWriteTrieMap<>(readWrite.mutableSnapshot(), size);
         LOG.trace("Converted read-only TrieMap {} to read-write {}", this, ret);
         return ret;
     }
 
     @Override
     protected Map<K, V> delegate() {
-        if (readOnly == null) {
-            synchronized (this) {
-                if (readOnly == null) {
-                    readOnly = readWrite.readOnlySnapshot();
-                }
+        ImmutableTrieMap<K, V> ret = readOnly;
+        if (ret == null) {
+            ret = readWrite.immutableSnapshot();
+            if (!UPDATER.compareAndSet(this, null, ret)) {
+                ret = readOnly;
             }
         }
-
-        return readOnly;
+        return ret;
     }
 
     @Override