Merge pull request #95 from Flouse/cache-get_account_id_by_script_hash

Cache get account id by script hash

packages/api-server/src/cache/constant.ts

@@ -2,3 +2,6 @@
 export const CACHE_EXPIRED_TIME_MILSECS = 5 * 60 * 1000; // milsec, default 5 minutes
 // limit redis store filter size
 export const MAX_FILTER_TOPIC_ARRAY_LENGTH = 20;
+
+// The Cache Key Prfixs
+export const GW_RPC_KEY = "gwRPC";

packages/api-server/src/methods/modules/gw.ts

@@ -2,11 +2,22 @@ import { RPC } from "ckb-js-toolkit";
 import { parseGwRpcError } from "../gw-error";
 import { middleware } from "../validator";
 import { HexNumber } from "@ckb-lumos/base";
+import { Store } from "../../cache/store";
+import { envConfig } from "../../base/env-config";
+import { CACHE_EXPIRED_TIME_MILSECS, GW_RPC_KEY } from "../../cache/constant";
 
 export class Gw {
   private rpc: RPC;
+  private gwCache: Store;
+
   constructor() {
     this.rpc = new RPC(process.env.GODWOKEN_JSON_RPC as string);
+    this.gwCache = new Store(
+      envConfig.redisUrl,
+      true,
+      CACHE_EXPIRED_TIME_MILSECS
+    );
+    this.gwCache.init();
 
     this.ping = middleware(this.ping.bind(this), 0);
     this.get_tip_block_hash = middleware(this.get_tip_block_hash.bind(this), 0);
@@ -154,7 +165,18 @@ export class Gw {
    */
   async get_account_id_by_script_hash(args: any[]) {
     try {
-      const result = await this.rpc.gw_get_account_id_by_script_hash(...args);
+      const scriptHash = args[0];
+      let result = await this.gwCache.get(`${GW_RPC_KEY}_${scriptHash}`);
+      if (result != null) {
+        console.debug(`using cache: ${scriptHash} -> ${result}`);
+        return result;
+      }
+
+      result = await this.rpc.gw_get_account_id_by_script_hash(...args);
+      if (result != null) {
+        console.debug(`update cache: ${scriptHash} -> ${result}`);
+        this.gwCache.insert(`${GW_RPC_KEY}_${scriptHash}`, result);
+      }
       return result;
     } catch (error) {
       parseGwRpcError(error);