Add serialization to ArrayMap

typhoon-php · Sep 2, 2024 · a65bca7 · a65bca7
1 parent 7b03676
commit a65bca7
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/src/DataStructure/Internal/ArrayMap.php b/src/DataStructure/Internal/ArrayMap.php
@@ -236,4 +236,22 @@ public function getIterator(): \Generator
             yield $kvPair->key => $kvPair->value;
         }
     }
+
+    /**
+     * @return list<KVPair<K, V>>
+     */
+    public function __serialize(): array
+    {
+        return array_values($this->kvPairs);
+    }
+
+    /**
+     * @param list<KVPair<K, V>> $kvPairs
+     */
+    public function __unserialize(array $kvPairs): void
+    {
+        foreach ($kvPairs as $kvPair) {
+            $this->kvPairs[UniqueHasher::hash($kvPair->key)] = $kvPair;
+        }
+    }
 }