feat(findBrandInVBank): add optional parameter to invalidate cache

Agoric · Jun 25, 2024 · b0de115 · b0de115
1 parent 653bde3
commit b0de115
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 13 deletions.
diff --git a/packages/orchestration/src/exos/agoric-names-tools.js b/packages/orchestration/src/exos/agoric-names-tools.js
@@ -17,7 +17,9 @@ const { Fail } = assert;
  * Perform remote calls to agoricNames in membrane-friendly way. This is an
  * interim approach until https://github.com/Agoric/agoric-sdk/issues/9541,
  * https://github.com/Agoric/agoric-sdk/pull/9322, or
- * https://github.com/Agoric/agoric-sdk/pull/9519
+ * https://github.com/Agoric/agoric-sdk/pull/9519.
+ *
+ * XXX only works once per zone.
  *
  * XXX consider exposing `has`, `entries`, `keys`, `values` from `NameHub`
  *
@@ -33,7 +35,9 @@ export const makeResumableAgoricNamesHack = (
     {
       public: M.interface('ResumableAgoricNamesHackI', {
         lookup: M.call().rest(M.arrayOf(M.string())).returns(VowShape),
-        findBrandInVBank: M.call(BrandShape).returns(VowShape),
+        findBrandInVBank: M.call(BrandShape)
+          .optional(M.boolean())
+          .returns(VowShape),
       }),
       vbankAssetEntriesWatcher: M.interface('vbankAssetEntriesWatcher', {
         onFulfilled: M.call(M.arrayOf(M.record()))
@@ -72,12 +76,19 @@ export const makeResumableAgoricNamesHack = (
           return watch(E(agoricNames).lookup(...args));
         },
         /**
+         * Look up asset info, like denom, in agoricNames.vbankAsset using a
+         * Brand.
+         *
+         * Caches the query to agoricNames in the first call. Subsequent lookups
+         * are via cache unless a refetch is specified or a brand is not found.
+         *
          * @param {Brand<'nat'>} brand
+         * @param {boolean} [refetch] if true, will invalidate the cache
          * @returns {Vow<AssetInfo>}
          */
-        findBrandInVBank(brand) {
+        findBrandInVBank(brand, refetch) {
           const { vbankAssetsByBrand } = this.state;
-          if (vbankAssetsByBrand.has(brand)) {
+          if (vbankAssetsByBrand.has(brand) && !refetch) {
             return watch(vbankAssetsByBrand.get(brand));
           }
           const vbankAssetNameHubP = E(agoricNames).lookup('vbankAsset');
@@ -91,6 +102,7 @@ export const makeResumableAgoricNamesHack = (
       },
     },
   );
+  // XXX only works once per zone.
   return makeResumableAgoricNamesHackKit().public;
 };
 /** @typedef {ReturnType<typeof makeResumableAgoricNamesHack>} AgNamesTools */
diff --git a/packages/orchestration/test/exos/agoric-names-tools.test.ts b/packages/orchestration/test/exos/agoric-names-tools.test.ts
@@ -5,6 +5,7 @@ import { V } from '@agoric/vow/vat.js';
 import { makeHeapZone } from '@agoric/zone';
 import { withAmountUtils } from '@agoric/zoe/tools/test-utils.js';
 import { makeIssuerKit } from '@agoric/ertp';
+import { AssetInfo } from '@agoric/vats/src/vat-bank.js';
 import { makeResumableAgoricNamesHack } from '../../src/exos/agoric-names-tools.js';
 import { commonSetup } from '../supports.js';
 
@@ -32,21 +33,37 @@ test('agoric names tools', async t => {
     message: /brand(.*?)not in agoricNames.vbankAsset/,
   });
 
+  const mooToken: AssetInfo = {
+    brand: moolah.brand,
+    issuer: moolah.issuer,
+    issuerName: 'MOO',
+    denom: 'umoo',
+    proposedName: 'MOO',
+    displayInfo: { decimalPlaces: 6, assetKind: 'nat' },
+  };
+
   await E(E(agoricNamesAdmin).lookupAdmin('vbankAsset')).update(
     'umoo',
-    /** @type {AssetInfo} */ harden({
-      brand: moolah.brand,
-      issuer: moolah.issuer,
-      issuerName: 'MOO',
-      denom: 'umoo',
-      proposedName: 'MOO',
-      displayInfo: { decimals: 6, symbol: 'MOO' },
-    }),
+    harden(mooToken),
+  );
+  t.like(
+    await V.when(agNamesTools.findBrandInVBank(moolah.brand)),
+    { denom: 'umoo' },
+    'vbankAssets are refetched if brand is not found',
   );
 
+  await E(E(agoricNamesAdmin).lookupAdmin('vbankAsset')).update(
+    'umoo',
+    harden({ ...mooToken, denom: 'umoo2' }),
+  );
   t.like(
     await V.when(agNamesTools.findBrandInVBank(moolah.brand)),
     { denom: 'umoo' },
-    'refresh stale cache for new assets',
+    'old AssetInfo is cached',
+  );
+  t.like(
+    await V.when(agNamesTools.findBrandInVBank(moolah.brand, true)),
+    { denom: 'umoo2' },
+    'new AssetInfo is fetched when refetch=true',
   );
 });