fix: LinkedList semantics

Consensys · Jan 17, 2024 · 4a15a36 · 4a15a36
1 parent c6bf4cf
commit 4a15a36
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 9 deletions.
diff --git a/...zation/src/main/java/net/consensys/linea/zktracer/container/stacked/list/StackedList.java b/...zation/src/main/java/net/consensys/linea/zktracer/container/stacked/list/StackedList.java
@@ -51,19 +51,20 @@ public String toString() {
 
   @Override
   public void enter() {
-    this.lists.push(new CountedList<>());
+    this.lists.addFirst(new CountedList<>());
   }
 
   public void enter(int initialCapacity) {
-    this.lists.push(new CountedList<>(initialCapacity));
+    this.lists.addFirst(new CountedList<>(initialCapacity));
   }
 
   @Override
   public void pop() {
     if (this.lists.isEmpty()) {
       throw new RuntimeException("asymmetric pop");
     }
-    this.totalSize -= this.lists.pop().size();
+
+    this.totalSize -= this.lists.removeFirst().size();
   }
 
   @Override
@@ -117,7 +118,7 @@ public <T> T[] toArray(@NotNull T[] a) {
   @Override
   public boolean add(E e) {
     this.totalSize++;
-    return this.lists.get(this.lists.size() - 1).add(e);
+    return this.lists.getFirst().add(e);
   }
 
   @Override

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 !");
+            }
+          });
     }
   }