From b34b9dc2c789aa839df11869038571a096d1631c Mon Sep 17 00:00:00 2001
From: Andrei Koval <andrei.koval@neo4j.com>
Date: Wed, 2 May 2018 11:13:37 +0200
Subject: [PATCH] Use primitive long diff sets

---
 .../api/index/updater/UniquePropertyIndexUpdater.java    | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/community/kernel/src/main/java/org/neo4j/kernel/impl/api/index/updater/UniquePropertyIndexUpdater.java b/community/kernel/src/main/java/org/neo4j/kernel/impl/api/index/updater/UniquePropertyIndexUpdater.java
index 73eb95ef90e46..eac48eafd08ed 100644
--- a/community/kernel/src/main/java/org/neo4j/kernel/impl/api/index/updater/UniquePropertyIndexUpdater.java
+++ b/community/kernel/src/main/java/org/neo4j/kernel/impl/api/index/updater/UniquePropertyIndexUpdater.java
@@ -25,7 +25,8 @@
 
 import org.neo4j.kernel.api.index.IndexEntryUpdate;
 import org.neo4j.kernel.api.index.IndexUpdater;
-import org.neo4j.kernel.impl.util.diffsets.DiffSets;
+import org.neo4j.kernel.impl.util.diffsets.MutableLongDiffSets;
+import org.neo4j.kernel.impl.util.diffsets.MutableLongDiffSetsImpl;
 
 /**
  * This IndexUpdater ensures that updated properties abide by uniqueness constraints. Updates are grouped up in
@@ -34,7 +35,7 @@
  */
 public abstract class UniquePropertyIndexUpdater implements IndexUpdater
 {
-    private final Map<Object, DiffSets<Long>> referenceCount = new HashMap<>();
+    private final Map<Object, MutableLongDiffSets> referenceCount = new HashMap<>();
     private final ArrayList<IndexEntryUpdate<?>> updates = new ArrayList<>();
 
     @Override
@@ -70,8 +71,8 @@ public void close()
 
     protected abstract void flushUpdates( Iterable<IndexEntryUpdate<?>> updates );
 
-    private DiffSets<Long> propertyValueDiffSet( Object value )
+    private MutableLongDiffSets propertyValueDiffSet( Object value )
     {
-        return referenceCount.computeIfAbsent( value, k -> new DiffSets<>() );
+        return referenceCount.computeIfAbsent( value, k -> new MutableLongDiffSetsImpl() );
     }
 }