default slot duration if not aura (#117)

AcalaNetwork · Dec 28, 2022 · 40172e7 · 40172e7
1 parent 1678db6
commit 40172e7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/executor.ts b/src/executor.ts
@@ -21,6 +21,7 @@ import {
   run_task,
 } from '@acala-network/chopsticks-executor'
 import { defaultLogger, truncate, truncateStorageDiff } from './logger'
+import { memoize } from 'lodash'
 
 interface JsCallback {
   getStorage: (key: HexString) => Promise<string | undefined>
@@ -125,7 +126,7 @@ export const emptyTaskHandler = {
   },
 }
 
-export const getAuraSlotDuration = async (wasm: HexString, registry: Registry): Promise<number> => {
+export const getAuraSlotDuration = memoize(async (wasm: HexString, registry: Registry): Promise<number> => {
   const result = await runTask({
     wasm,
     calls: [['AuraApi_slot_duration', '0x']],
@@ -137,4 +138,4 @@ export const getAuraSlotDuration = async (wasm: HexString, registry: Registry):
   if (!result.Call) throw new Error(result.Error)
   const slotDuration = registry.createType('u64', hexToU8a(result.Call.result)).toNumber()
   return slotDuration
-}
+})
diff --git a/src/utils/time-travel.ts b/src/utils/time-travel.ts
@@ -26,7 +26,9 @@ export const getSlotDuration = async (chain: Blockchain) => {
   const meta = await chain.head.meta
   return meta.consts.babe
     ? (meta.consts.babe.expectedBlockTime as any as BN).toNumber()
-    : getAuraSlotDuration(await chain.head.wasm, meta.registry)
+    : meta.query.aura
+    ? getAuraSlotDuration(await chain.head.wasm, meta.registry)
+    : 12_000
 }
 
 export const timeTravel = async (chain: Blockchain, timestamp: number) => {