@moneypot/hub 1.22.3 → 1.22.4

package/dist/src/process-transfers/websocket-processor.js

@@ -1,6 +1,6 @@
 import { createClient as createWebsocketClient, } from "graphql-ws";
 import { gql } from "../__generated__/gql.js";
-import { print } from "graphql";
+import { getOperationAST, print } from "graphql";
 import { createGraphqlClient } from "../graphql-client.js";
 import { useFragment } from "../__generated__/fragment-masking.js";
 import { TRANSFER_FIELDS } from "./graphql.js";
@@ -51,6 +51,8 @@ async function mpGetExperienceTransfer(graphqlClient, id) {
     return mpTransfer;
 }
 export function startWebsocketProcessor({ casinoId, graphqlUrl, signal, controllerId, apiKey, pool, }) {
+    if (signal.aborted)
+        return;
     logger.info(`Starting websocket processor for ${graphqlUrl}`);
     const client = createWebsocketClient({
         url: httpToWs(new URL(graphqlUrl)).toString(),
@@ -63,9 +65,7 @@ export function startWebsocketProcessor({ casinoId, graphqlUrl, signal, controll
         },
         keepAlive: 1_000,
         retryWait: async (retries) => {
-            const baseDelay = Math.min(1000 * Math.pow(2, retries), 30_000);
-            const jitter = Math.floor(Math.random() * 10_000);
-            await abortableTimeout(baseDelay + jitter, signal);
+            await abortableTimeout(backoffDelayMs(retries), signal);
         },
     });
     client.on("connected", () => {
@@ -78,60 +78,83 @@ export function startWebsocketProcessor({ casinoId, graphqlUrl, signal, controll
         logger.error(error, `[websocketProcessor] Error on websocket ${graphqlUrl}`);
     });
     const graphqlClient = createGraphqlClient({ graphqlUrl, apiKey });
-    const dispose1 = createSubscription(client, MP_NEW_EXPERIENCE_TRANSFER, async (result) => {
-        const transferId = result.data?.newExperienceTransfer?.id;
-        if (!transferId) {
-            logger.warn(result, `Weird, no transfer id in websocket result`);
-            return;
-        }
-        const mpTransfer = await mpGetExperienceTransfer(graphqlClient, transferId);
-        if (!mpTransfer) {
-            logger.warn(`No experience transfer found in casino api for mp transfer id ${transferId}`);
-            return;
-        }
-        assert(mpTransfer.__typename === "ExperienceTransfer", `Expected experience transfer but got ${mpTransfer.__typename}`);
-        await processTransfer({
-            casinoId,
-            controllerId,
-            transfer: mpTransfer,
-            graphqlClient,
-            pool,
-        });
-    });
-    const dispose2 = createSubscription(client, MP_NEW_TAKE_REQUEST, async (result) => {
-        const mpTakeRequestId = result.data?.newTakeRequest?.id;
-        if (!mpTakeRequestId) {
-            logger.warn(result, `Weird, no take request id in websocket result`);
-            return;
-        }
-        const mpTakeRequest = await mpGetTakeRequest(graphqlClient, mpTakeRequestId);
-        if (!mpTakeRequest) {
-            logger.warn(`No take request found in casino api for mp take request id ${mpTakeRequestId}`);
-            return;
-        }
-        await processSingleTakeRequest({
-            mpTakeRequestId,
-            mpTakeRequest,
-            casinoId,
-            graphqlClient,
-            pool,
-        });
-    });
-    signal.addEventListener("abort", () => {
-        dispose1();
-        dispose2();
-    });
-}
-function createSubscription(client, query, handler) {
-    return client.subscribe({ query: print(query) }, {
-        next: (result) => {
-            handler(result).catch((e) => {
-                logger.error(e, `Error while handling websocket result`);
+    keepSubscribed({
+        client,
+        query: MP_NEW_EXPERIENCE_TRANSFER,
+        signal,
+        idOf: (result) => result.data?.newExperienceTransfer?.id,
+        handleId: async (transferId) => {
+            const mpTransfer = await mpGetExperienceTransfer(graphqlClient, transferId);
+            if (!mpTransfer) {
+                logger.warn(`No experience transfer found in casino api for mp transfer id ${transferId}`);
+                return;
+            }
+            assert(mpTransfer.__typename === "ExperienceTransfer", `Expected experience transfer but got ${mpTransfer.__typename}`);
+            await processTransfer({
+                casinoId,
+                controllerId,
+                transfer: mpTransfer,
+                graphqlClient,
+                pool,
             });
         },
-        error: (err) => {
-            logger.error(err, `Error during WebSocket subscription`);
+    });
+    keepSubscribed({
+        client,
+        query: MP_NEW_TAKE_REQUEST,
+        signal,
+        idOf: (result) => result.data?.newTakeRequest?.id,
+        handleId: async (mpTakeRequestId) => {
+            const mpTakeRequest = await mpGetTakeRequest(graphqlClient, mpTakeRequestId);
+            if (!mpTakeRequest) {
+                logger.warn(`No take request found in casino api for mp take request id ${mpTakeRequestId}`);
+                return;
+            }
+            await processSingleTakeRequest({
+                mpTakeRequestId,
+                mpTakeRequest,
+                casinoId,
+                graphqlClient,
+                pool,
+            });
         },
-        complete: () => logger.info("Subscription closed"),
     });
 }
+function backoffDelayMs(attempt) {
+    const baseDelay = Math.min(1000 * Math.pow(2, attempt), 30_000);
+    const jitter = Math.floor(Math.random() * 10_000);
+    return baseDelay + jitter;
+}
+function keepSubscribed({ client, query, signal, idOf, handleId, }) {
+    const name = getOperationAST(query)?.name?.value ?? "subscription";
+    let attempt = 0;
+    let dispose = () => { };
+    const ended = (how, err) => {
+        if (signal.aborted)
+            return;
+        const delay = backoffDelayMs(attempt++);
+        logger.warn({ err }, `[websocketProcessor] Casino ended ${name} with ${how}, resubscribing in ${delay}ms`);
+        abortableTimeout(delay, signal).then(subscribe);
+    };
+    const subscribe = () => {
+        if (signal.aborted)
+            return;
+        dispose = client.subscribe({ query: print(query) }, {
+            next: (result) => {
+                const id = idOf(result);
+                if (!id) {
+                    logger.warn(result, `Weird, no id in ${name} websocket result`);
+                    return;
+                }
+                attempt = 0;
+                handleId(id).catch((e) => {
+                    logger.error(e, `Error while handling websocket result`);
+                });
+            },
+            error: (err) => ended("error", err),
+            complete: () => ended("complete"),
+        });
+    };
+    signal.addEventListener("abort", () => dispose(), { once: true });
+    subscribe();
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@moneypot/hub",
-  "version": "1.22.3",
+  "version": "1.22.4",
   "author": "moneypot.com",
   "homepage": "https://moneypot.com/hub",
   "keywords": [