@codex-data/sdk 2.4.12 → 2.5.2

package/dist/sdk/index.d.ts

@@ -21,7 +21,7 @@ export declare class Codex {
     private apiConfig?;
     private client;
     private wsClient;
-    private edgeWsClient;
+    private edgeWsClients;
     private disposed;
     queries: Query;
     mutations: Mutation;

package/dist/sdk/index.js

@@ -35,6 +35,7 @@ class Codex {
         var _a, _b;
         this.apiKey = apiKey;
         this.apiConfig = apiConfig;
+        this.edgeWsClients = new Map();
         this.disposed = false;
         const config = this.parseConfig((_a = this.apiConfig) !== null && _a !== void 0 ? _a : {});
         this.queries = new Query_1.Query(this);
@@ -86,9 +87,9 @@ class Codex {
     updateConfig(incomingConfig) {
         var _a, _b, _c;
         // If we have an existing websocket connection, dispose of it.
-        if (this.wsClient || this.edgeWsClient)
+        if (this.wsClient || this.edgeWsClients.size)
             void this.dispose();
-        this.edgeWsClient = undefined;
+        this.edgeWsClients.clear();
         this.disposed = false;
         this.apiConfig = Object.assign(Object.assign({}, this.apiConfig), incomingConfig);
         // Merge the incoming config with the existing config.
@@ -114,9 +115,12 @@ class Codex {
      */
     dispose() {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b;
+            var _a;
             this.disposed = true;
-            yield Promise.all([(_a = this.wsClient) === null || _a === void 0 ? void 0 : _a.dispose(), (_b = this.edgeWsClient) === null || _b === void 0 ? void 0 : _b.dispose()]);
+            yield Promise.all([
+                (_a = this.wsClient) === null || _a === void 0 ? void 0 : _a.dispose(),
+                ...Array.from(this.edgeWsClients.values(), (edge) => edge.dispose()),
+            ]);
         });
     }
     /**
@@ -163,7 +167,7 @@ class Codex {
      * @returns A cleanup function to unsubscribe from the subscription.
      */
     subscribe(doc, args, sink, options) {
-        var _a;
+        var _a, _b;
         if (!this.wsClient) {
             throw new Error("Websocket is not enabled in the config");
         }
@@ -173,15 +177,18 @@ class Codex {
         if (!(options === null || options === void 0 ? void 0 : options.useEdge))
             return this.wsClient.subscribe(payload, sink);
         // Validate before creating the edge transport or starting an operation.
-        const snapshot = (0, edge_1.prepareEdgeSubscription)(payload);
-        if (!this.edgeWsClient) {
-            const config = this.parseConfig((_a = this.apiConfig) !== null && _a !== void 0 ? _a : {});
-            this.edgeWsClient = (0, edge_1.createEdgeClient)(Object.assign(Object.assign({}, this.websocketOptions(config)), { url: config.apiEdgeUrl }));
-            this.edgeWsClient.on("error", (error) => {
+        const { payload: snapshot, networkId } = (0, edge_1.prepareEdgeSubscription)(payload);
+        const config = this.parseConfig((_a = this.apiConfig) !== null && _a !== void 0 ? _a : {});
+        const url = (_b = config.apiEdgeUrl) !== null && _b !== void 0 ? _b : (0, edge_1.edgeUrlForNetwork)(networkId);
+        let edge = this.edgeWsClients.get(url);
+        if (!edge) {
+            edge = (0, edge_1.createEdgeClient)(Object.assign(Object.assign({}, this.websocketOptions(config)), { url }));
+            edge.on("error", (error) => {
                 console.error("Websocket error: ", error);
             });
+            this.edgeWsClients.set(url, edge);
         }
-        return this.edgeWsClient.subscribe(snapshot, sink);
+        return edge.subscribe(snapshot, sink);
     }
 }
 exports.Codex = Codex;

package/dist/sdk/index.mjs

@@ -1,7 +1,7 @@
 import { GraphQLClient } from "graphql-request";
 import { createClient, } from "graphql-ws";
 import WebSocket from "isomorphic-ws";
-import { createEdgeClient, prepareEdgeSubscription, } from "./edge";
+import { createEdgeClient, edgeUrlForNetwork, prepareEdgeSubscription, } from "./edge";
 import { invariant } from "./invariant";
 import { Mutation } from "./Mutation";
 import { Query } from "./Query";
@@ -19,6 +19,7 @@ export class Codex {
     constructor(apiKey, apiConfig) {
         this.apiKey = apiKey;
         this.apiConfig = apiConfig;
+        this.edgeWsClients = new Map();
         this.disposed = false;
         const config = this.parseConfig(this.apiConfig ?? {});
         this.queries = new Query(this);
@@ -77,9 +78,9 @@ export class Codex {
      */
     updateConfig(incomingConfig) {
         // If we have an existing websocket connection, dispose of it.
-        if (this.wsClient || this.edgeWsClient)
+        if (this.wsClient || this.edgeWsClients.size)
             void this.dispose();
-        this.edgeWsClient = undefined;
+        this.edgeWsClients.clear();
         this.disposed = false;
         this.apiConfig = { ...this.apiConfig, ...incomingConfig };
         // Merge the incoming config with the existing config.
@@ -105,7 +106,10 @@ export class Codex {
      */
     async dispose() {
         this.disposed = true;
-        await Promise.all([this.wsClient?.dispose(), this.edgeWsClient?.dispose()]);
+        await Promise.all([
+            this.wsClient?.dispose(),
+            ...Array.from(this.edgeWsClients.values(), (edge) => edge.dispose()),
+        ]);
     }
     /**
      * This is the preferred way to execute queries.
@@ -154,17 +158,20 @@ export class Codex {
         if (!options?.useEdge)
             return this.wsClient.subscribe(payload, sink);
         // Validate before creating the edge transport or starting an operation.
-        const snapshot = prepareEdgeSubscription(payload);
-        if (!this.edgeWsClient) {
-            const config = this.parseConfig(this.apiConfig ?? {});
-            this.edgeWsClient = createEdgeClient({
+        const { payload: snapshot, networkId } = prepareEdgeSubscription(payload);
+        const config = this.parseConfig(this.apiConfig ?? {});
+        const url = config.apiEdgeUrl ?? edgeUrlForNetwork(networkId);
+        let edge = this.edgeWsClients.get(url);
+        if (!edge) {
+            edge = createEdgeClient({
                 ...this.websocketOptions(config),
-                url: config.apiEdgeUrl,
+                url,
             });
-            this.edgeWsClient.on("error", (error) => {
+            edge.on("error", (error) => {
                 console.error("Websocket error: ", error);
             });
+            this.edgeWsClients.set(url, edge);
         }
-        return this.edgeWsClient.subscribe(snapshot, sink);
+        return edge.subscribe(snapshot, sink);
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@codex-data/sdk",
-  "version": "2.4.12",
+  "version": "2.5.2",
   "engines": {
     "node": ">=17.5.0"
   },