Use a concurrent set for registrations

[bgpcep.git] / bgp / rib-impl / src / main / java / org / opendaylight / protocol / bgp / rib / impl / state / BGPStateCollector.java

diff --git a/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/state/BGPStateCollector.java b/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/state/BGPStateCollector.java
index 2b77720b4f0e2756e27792f47e097c83a94b2096..728f974eeec26573781a41ddb0ecfd1fc5eed461 100644 (file)
--- a/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/state/BGPStateCollector.java
+++ b/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/state/BGPStateCollector.java
@@ -10,7 +10,8 @@ package org.opendaylight.protocol.bgp.rib.impl.state;
 import com.google.common.collect.ImmutableList;
 import java.util.List;
 import java.util.Objects;
-import java.util.concurrent.CopyOnWriteArrayList;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
 import javax.inject.Inject;
 import javax.inject.Singleton;
 import org.eclipse.jdt.annotation.NonNull;
@@ -51,8 +52,8 @@ public final class BGPStateCollector implements BGPStateProviderRegistry, BGPSta
         }
     }
 
-    private final List<BGPPeerStateProviderRegistration> bgpPeerStates = new CopyOnWriteArrayList<>();
-    private final List<BGPRibStateProviderRegistration> bgpRibStates = new CopyOnWriteArrayList<>();
+    private final Set<BGPPeerStateProviderRegistration> bgpPeerStates = ConcurrentHashMap.newKeySet();
+    private final Set<BGPRibStateProviderRegistration> bgpRibStates = ConcurrentHashMap.newKeySet();
 
     @Inject
     public BGPStateCollector() {