@typeberry/lib 0.3.0 → 0.3.1

package/index.cjs

@@ -11350,15 +11350,19 @@ class InMemorySerializedStates {
     async updateAndSetState(header, state, update) {
         const blake2b = this.blake2b;
         const updatedValues = serializeStateUpdate(this.spec, blake2b, update);
-        const { values, leafs } = updateLeafs(state.backend.leafs, blake2b, updatedValues);
+        // make sure to clone the leafs before writing, since the collection is re-used.
+        const newLeafs = SortedSet.fromSortedArray(leafComparator, state.backend.leafs.array);
+        const { values, leafs } = updateLeafs(newLeafs, blake2b, updatedValues);
+        // make sure to reset the cache and re-create leafsdb lookup
+        state.updateBackend(LeafDb.fromLeaves(leafs, state.backend.db));
         // insert values to the db
         // valuesdb can be shared between all states because it's just
         // <valuehash> -> <value> mapping and existence is managed by trie leafs.
         for (const val of values) {
             this.valuesDb.set(val[0], val[1]);
         }
-        // make sure to clone the leafs before writing, since the collection is re-used.
-        this.db.set(header, SortedSet.fromSortedArray(leafComparator, leafs.slice()));
+        // store new set of leaves
+        this.db.set(header, leafs);
         return Result$1.ok(OK);
     }
     getState(header) {

package/index.d.ts

@@ -13131,7 +13131,11 @@ declare class InMemorySerializedStates implements StatesDb<SerializedState<LeafD
   ): Promise<Result$2<OK, StateUpdateError>> {
     const blake2b = this.blake2b;
     const updatedValues = serializeStateUpdate(this.spec, blake2b, update);
-    const { values, leafs } = updateLeafs(state.backend.leafs, blake2b, updatedValues);
+    // make sure to clone the leafs before writing, since the collection is re-used.
+    const newLeafs = SortedSet.fromSortedArray(leafComparator, state.backend.leafs.array);
+    const { values, leafs } = updateLeafs(newLeafs, blake2b, updatedValues);
+    // make sure to reset the cache and re-create leafsdb lookup
+    state.updateBackend(LeafDb.fromLeaves(leafs, state.backend.db));
 
     // insert values to the db
     // valuesdb can be shared between all states because it's just
@@ -13140,8 +13144,8 @@ declare class InMemorySerializedStates implements StatesDb<SerializedState<LeafD
       this.valuesDb.set(val[0], val[1]);
     }
 
-    // make sure to clone the leafs before writing, since the collection is re-used.
-    this.db.set(header, SortedSet.fromSortedArray(leafComparator, leafs.slice()));
+    // store new set of leaves
+    this.db.set(header, leafs);
 
     return Result.ok(OK);
   }

package/index.js

@@ -11347,15 +11347,19 @@ class InMemorySerializedStates {
     async updateAndSetState(header, state, update) {
         const blake2b = this.blake2b;
         const updatedValues = serializeStateUpdate(this.spec, blake2b, update);
-        const { values, leafs } = updateLeafs(state.backend.leafs, blake2b, updatedValues);
+        // make sure to clone the leafs before writing, since the collection is re-used.
+        const newLeafs = SortedSet.fromSortedArray(leafComparator, state.backend.leafs.array);
+        const { values, leafs } = updateLeafs(newLeafs, blake2b, updatedValues);
+        // make sure to reset the cache and re-create leafsdb lookup
+        state.updateBackend(LeafDb.fromLeaves(leafs, state.backend.db));
         // insert values to the db
         // valuesdb can be shared between all states because it's just
         // <valuehash> -> <value> mapping and existence is managed by trie leafs.
         for (const val of values) {
             this.valuesDb.set(val[0], val[1]);
         }
-        // make sure to clone the leafs before writing, since the collection is re-used.
-        this.db.set(header, SortedSet.fromSortedArray(leafComparator, leafs.slice()));
+        // store new set of leaves
+        this.db.set(header, leafs);
         return Result$1.ok(OK);
     }
     getState(header) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@typeberry/lib",
-  "version": "0.3.0",
+  "version": "0.3.1",
   "main": "index.js",
   "author": "Fluffy Labs",
   "license": "MPL-2.0",