wirejs-deploy-amplify-basic 0.0.118-realtime → 0.0.119-realtime

package/amplify-hosting-assets/compute/default/package.json

@@ -3,6 +3,6 @@
 	"dependencies": {
 		"jsdom": "^25.0.1",
 		"wirejs-dom": "^1.0.41",
-		"wirejs-resources": "^0.1.86-realtime"
+		"wirejs-resources": "^0.1.87-realtime"
 	}
 }

package/dist/client/realtime.js

@@ -100,37 +100,53 @@ export function subscribe(url, channel, token, authHost, subscriber) {
                 }
             }
         };
-        const notifyClosed = (reason) => {
+        const cleanup = (reason) => {
+            // remove the top-level connection entry *first* so that subsequent
+            // subscribers don't try to access a closed WebSocket. they will instead
+            // see no connection and a new one will be created.
+            connections.delete(urlKey);
+            // next. we need to identify all subscriptions to notify -- but wait to notify
+            // until after we have removed them from the look-up table so that the callbacks
+            // don't interfere with the bookkeeping.
             const subscriptionIds = Array.from(channelSubIdLUT.entries())
                 .filter(([urlChannel, _subId]) => urlChannel.startsWith(`${url}#`))
                 .map(([_urlChannel, subId]) => subscriptionIdString(subId));
+            const subs = [];
+            for (const subscriptionId of subscriptionIds) {
+                subs.concat(subIdSubscribers.get(subscriptionId));
+            }
+            // same story with channel names.
+            const fullChannelNames = Array.from(channelSubIdLUT.keys())
+                .filter(k => k.startsWith(`${url}#`));
+            // now that we've *identified* everything that requires bookkeeping and
+            // notification, we do the bookkeeping ONLY.
             for (const subscriptionId of subscriptionIds) {
-                const subs = subIdSubscribers.get(subscriptionId);
-                if (subs) {
-                    for (const subscriber of subs) {
-                        if (subscriber.onclose) {
-                            try {
-                                subscriber.onclose(reason);
-                            }
-                            catch (error) {
-                                console.error('Error in subscriber onclose:', error);
-                            }
-                        }
-                    }
-                }
                 subIdSubscribers.delete(subscriptionId);
                 subscriptionState.delete(subscriptionId);
             }
-            channelSubIdLUT.delete(fullChannelName);
-            connections.delete(urlKey);
+            for (const fullChannelName of fullChannelNames) {
+                channelSubIdLUT.delete(fullChannelName);
+            }
+            // all bookkeeping is done, now we can notify subscribers.
+            // they can no longer interfere with the bookkeeping.
+            for (const subscriber of subs || []) {
+                try {
+                    subscriber.onclose?.(reason);
+                }
+                catch (error) {
+                    console.error('Error in subscriber onclose:', error);
+                }
+            }
             console.debug('closed', ws);
         };
-        ws.onclose = () => notifyClosed('closed');
-        ws.onerror = () => notifyClosed('error');
+        ws.onclose = () => cleanup('closed');
+        ws.onerror = () => cleanup('error');
     }
     if (!channelSubIdLUT.has(fullChannelName)) {
         const subscriptionId = subscriptionIdString(crypto.randomUUID());
         subscriptionState.set(subscriptionId, 'connecting');
+        channelSubIdLUT.set(fullChannelName, subscriptionId);
+        subIdSubscribers.set(subscriptionId, [subscriber]);
         const ws = connections.get(urlKey);
         const subscribe = () => {
             ws.send(JSON.stringify({
@@ -139,8 +155,6 @@ export function subscribe(url, channel, token, authHost, subscriber) {
                 channel,
                 authorization
             }));
-            channelSubIdLUT.set(fullChannelName, subscriptionId);
-            subIdSubscribers.set(subscriptionId, [subscriber]);
         };
         if (ws.readyState === WebSocket.OPEN) {
             subscribe();
@@ -151,9 +165,21 @@ export function subscribe(url, channel, token, authHost, subscriber) {
     }
     else {
         const subscriptionId = channelSubIdLUT.get(fullChannelName);
-        subIdSubscribers.get(subscriptionId).push(subscriber);
+        if (!subscriptionId) {
+            throw new Error(`No subscription ID found for channel: ${fullChannelName}`);
+        }
+        const subs = subIdSubscribers.get(subscriptionId);
+        if (!subs) {
+            throw new Error(`No subscriber list found for subscription ID: ${subscriptionId}`);
+        }
+        subs?.push(subscriber);
         if (subscriptionState.get(subscriptionId) === 'open') {
-            subscriber.onopen?.();
+            try {
+                subscriber.onopen?.();
+            }
+            catch (error) {
+                console.error('Error in subscriber onopen:', error);
+            }
         }
     }
 }
@@ -165,27 +191,48 @@ export function unsubscribe(url, channel, subscriber) {
     const subs = subId && subIdSubscribers.get(subId);
     const sub = subs ? subs.find(s => s === subscriber) : undefined;
     if (sub && subs) {
-        sub.onclose?.('unsubscribed');
+        try {
+            sub.onclose?.('unsubscribed');
+        }
+        catch (error) {
+            console.error('Error in subscriber onclose:', error);
+        }
         const i = subs.indexOf(sub);
         if (i > -1)
             subs.splice(i, 1);
     }
     if (subs && subs.length === 0) {
         // No subIdSubscribers left for this channel. We can unsubscribe from channel.
+        subscriptionState.delete(subId);
+        subIdSubscribers.delete(subId);
+        channelSubIdLUT.delete(fullChannelName);
         ws?.send(JSON.stringify({
             id: subId,
             type: 'unsubscribe',
         }));
-        subscriptionState.delete(subId);
-        subIdSubscribers.delete(subId);
-        channelSubIdLUT.delete(fullChannelName);
     }
-    const socketSubs = Array.from(channelSubIdLUT.keys())
-        .filter(k => k.startsWith(`${url}#`));
-    if (socketSubs.length === 0) {
-        // No channels left for this URL. We can close the WebSocket connection.
-        ws?.close();
-        connections.delete(urlKey);
-        console.debug('closed', ws);
+    // debounce closing the WebSocket connection for use-cases where a client
+    // might just be changing channels. ws should only be closed if there are
+    // no more subscriptions running over the socket.
+    setTimeout(() => {
+        const socketSubs = Array.from(channelSubIdLUT.keys())
+            .filter(k => k.startsWith(`${url}#`));
+        if (socketSubs.length === 0) {
+            // No channels left for this URL. We can close the WebSocket connection.
+            connections.delete(urlKey);
+            ws?.close();
+            console.debug('closed', ws);
+        }
+    }, 5000);
+    // tell the subscriber that they have been unsubscribed as the last step so
+    // they can clean up any resources they have allocated without interfering
+    // with the preceding bookkeeping.
+    if (sub) {
+        try {
+            sub.onclose?.('unsubscribed');
+        }
+        catch (error) {
+            console.error('Error in subscriber onclose:', error);
+        }
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "wirejs-deploy-amplify-basic",
-	"version": "0.0.118-realtime",
+	"version": "0.0.119-realtime",
 	"type": "module",
 	"main": "./dist/index.js",
 	"types": "./dist/index.d.ts",
@@ -41,7 +41,7 @@
 		"recursive-copy": "^2.0.14",
 		"rimraf": "^6.0.1",
 		"wirejs-dom": "^1.0.41",
-		"wirejs-resources": "^0.1.86-realtime"
+		"wirejs-resources": "^0.1.87-realtime"
 	},
 	"devDependencies": {
 		"@aws-amplify/backend": "^1.14.0",