Multiple WS urls for PolkadotJS API (#273)

moonbeam-foundation · Jun 14, 2024 · d91de57 · d91de57
1 parent b303c1d
commit d91de57
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 6 deletions.
diff --git a/.changeset/quick-parrots-kick.md b/.changeset/quick-parrots-kick.md
@@ -0,0 +1,6 @@
+---
+'@moonbeam-network/xcm-types': patch
+'@moonbeam-network/xcm-utils': patch
+---
+
+Multiple WS urls for PolkadotJS API
diff --git a/packages/types/src/chain/parachain/EvmParachain.ts b/packages/types/src/chain/parachain/EvmParachain.ts
@@ -58,7 +58,7 @@ export class EvmParachain extends Parachain {
       rpcUrls: {
         default: {
           http: [this.rpc],
-          webSocket: [this.ws],
+          webSocket: Array.isArray(this.ws) ? this.ws : [this.ws],
         },
       },
     });

diff --git a/packages/types/src/chain/parachain/Parachain.ts b/packages/types/src/chain/parachain/Parachain.ts
@@ -12,7 +12,7 @@ export interface ParachainConstructorParams
   ss58Format: number;
   usesChainDecimals?: boolean;
   weight?: number;
-  ws: string;
+  ws: string | string[];
 }
 
 export class Parachain extends Chain {
@@ -28,7 +28,7 @@ export class Parachain extends Chain {
 
   readonly weight: number | undefined;
 
-  readonly ws: string;
+  readonly ws: string | string[];
 
   constructor({
     assetsData,

diff --git a/packages/utils/src/polkadot/polkadot.api.ts b/packages/utils/src/polkadot/polkadot.api.ts
@@ -22,9 +22,12 @@ const cache = new LRUCache<string, Promise<ApiPromise>>({
   },
 });
 
-export async function getPolkadotApi(ws: string): Promise<ApiPromise> {
+export async function getPolkadotApi(
+  ws: string | string[],
+): Promise<ApiPromise> {
+  const key = Array.isArray(ws) ? ws.join(';') : ws;
   const promise =
-    cache.get(ws) ||
+    cache.get(key) ||
     ApiPromise.create({
       noInitWarn: true,
       provider: new WsProvider(ws),
@@ -39,7 +42,7 @@ export async function getPolkadotApi(ws: string): Promise<ApiPromise> {
       typesBundle,
     });
 
-  cache.set(ws, promise);
+  cache.set(key, promise);
 
   const api = await promise;