partial key support in storage rune size

paritytech · Feb 27, 2023 · cf47fa9 · cf47fa9
1 parent 4915833
commit cf47fa9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/fluent/StorageRune.ts b/fluent/StorageRune.ts
@@ -19,9 +19,13 @@ export class StorageRune<in out K extends unknown[], out V, out U> extends Rune<
     this.$value = this.pallet.metadata.codec(this.into(ValueRune).access("value"))
   }
 
-  size<X>(...[blockHash]: RunicArgs<X, [HexHash?]>) {
+  size<X>(...[partialKey, blockHash]: RunicArgs<X, [partialKey?: unknown[], blockHash?: HexHash]>) {
     return this.pallet.metadata.chain.connection
-      .call("state_getStorageSize", this.$key.encoded([]).map(hex.encode), blockHash)
+      .call(
+        "state_getStorageSize",
+        this.$key.encoded(Rune.resolve(partialKey).map((x) => x ?? [])).map(hex.encode),
+        blockHash,
+      )
       .unhandle(null)
       .rehandle(null, () => Rune.constant(undefined))
   }

diff --git a/patterns/sizeTree.ts b/patterns/sizeTree.ts
@@ -2,7 +2,7 @@ import { Chain, ChainRune, HexHash, MetaRune, Rune, RunicArgs, ValueRune } from
 
 export function sizeTree<U, C extends Chain, X>(
   chain: ChainRune<U, C>,
-  ...[blockHash]: RunicArgs<X, [HexHash?]>
+  ...[blockHash]: RunicArgs<X, [blockHash?: HexHash]>
 ) {
   const metadata = chain.metadata(blockHash)
   return metadata
@@ -13,7 +13,7 @@ export function sizeTree<U, C extends Chain, X>(
         Rune.rec(Object.fromEntries(
           pallet.storage?.entries.map((entry) => [
             entry.name,
-            metadata.pallet(pallet.name).storage(entry.name).size(blockHash),
+            metadata.pallet(pallet.name).storage(entry.name).size([], blockHash),
           ]) || [],
         )),
       ])))