fix: ArrayDeque semantics

Consensys · Jan 16, 2024 · 22b6529 · 22b6529
1 parent d6f4682
commit 22b6529
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/...tization/src/main/java/net/consensys/linea/zktracer/container/stacked/set/StackedSet.java b/...tization/src/main/java/net/consensys/linea/zktracer/container/stacked/set/StackedSet.java
@@ -42,16 +42,22 @@ public class StackedSet<E extends ModuleOperation> implements StackedContainer,
 
   @Override
   public void enter() {
-    this.sets.push(new HashSet<>());
+    this.sets.addLast(new HashSet<>());
   }
 
   @Override
   public void pop() {
-    Set<E> set = this.sets.pop();
+    Set<E> set = this.sets.removeLast();
     for (E e : set) {
-      Integer count = occurrences.get(e);
-      if (count > 0) occurrences.put(e, count - 1);
-      else throw new IllegalStateException("asymmetric element removal !");
+      occurrences.computeIfPresent(
+          e,
+          (k, count) -> {
+            if (count > 0) {
+              return count - 1;
+            } else {
+              throw new IllegalStateException("asymmetric element removal !");
+            }
+          });
     }
   }