@absolutejs/sync 2.0.0 → 2.1.0

package/dist/index.js

@@ -496,40 +496,59 @@ var syncSocket = ({
   };
   return new Elysia2({ name: "@absolutejs/sync/socket" }).ws(path, {
     async open(ws) {
-      const ctx = resolveContext ? await resolveContext(ws.data) : {};
       const bunWs = ws;
       const tracked = {
-        connection: createSyncConnection({
-          engine,
-          ctx,
-          presence,
-          serializer,
-          send: (frame) => {
-            const payload = serializer.encodeServer(frame);
-            const ret = bunWs.send(typeof payload === "string" ? payload : payload);
-            const buffered = bunWs.getBufferedAmount?.() ?? 0;
-            const overBuffer = buffered > threshold;
-            const backpressure = ret === -1;
-            if ((overBuffer || backpressure) && !tracked.slowSignaled) {
-              tracked.slowSignaled = true;
-              fireSlow({
-                bufferedAmount: buffered,
-                reason: backpressure ? "send-backpressure" : "buffer-threshold",
-                stats: tracked.connection.stats(),
-                wsId: bunWs.id
-              });
-              if (closeOnSlow)
-                bunWs.close?.();
-            }
-            return ret;
-          }
-        }),
+        connection: null,
+        pending: [],
         slowSignaled: false
       };
       connections.set(bunWs.id, tracked);
+      const ctx = resolveContext ? await resolveContext(ws.data) : {};
+      const connection = createSyncConnection({
+        engine,
+        ctx,
+        presence,
+        serializer,
+        send: (frame) => {
+          const payload = serializer.encodeServer(frame);
+          const ret = bunWs.send(typeof payload === "string" ? payload : payload);
+          const buffered2 = bunWs.getBufferedAmount?.() ?? 0;
+          const overBuffer = buffered2 > threshold;
+          const backpressure = ret === -1;
+          if ((overBuffer || backpressure) && !tracked.slowSignaled) {
+            tracked.slowSignaled = true;
+            fireSlow({
+              bufferedAmount: buffered2,
+              reason: backpressure ? "send-backpressure" : "buffer-threshold",
+              stats: connection.stats(),
+              wsId: bunWs.id
+            });
+            if (closeOnSlow)
+              bunWs.close?.();
+          }
+          return ret;
+        }
+      });
+      if (!connections.has(bunWs.id)) {
+        connection.close();
+        return;
+      }
+      tracked.connection = connection;
+      const buffered = tracked.pending;
+      tracked.pending = [];
+      for (const frame of buffered) {
+        await connection.handle(frame);
+      }
     },
     async message(ws, message) {
-      await connections.get(ws.id)?.connection.handle(message);
+      const tracked = connections.get(ws.id);
+      if (!tracked)
+        return;
+      if (tracked.connection) {
+        await tracked.connection.handle(message);
+      } else {
+        tracked.pending.push(message);
+      }
     },
     drain(ws) {
       const tracked = connections.get(ws.id);
@@ -539,7 +558,7 @@ var syncSocket = ({
     close(ws) {
       const tracked = connections.get(ws.id);
       if (tracked) {
-        tracked.connection.close();
+        tracked.connection?.close();
         connections.delete(ws.id);
       }
     }
@@ -1356,6 +1375,9 @@ var createSyncEngine = (options = {}) => {
   let mutationsFailed = 0;
   let mutationsRetried = 0;
   let mutationsInFlight = 0;
+  let connectedSources = 0;
+  let sourceChangesReceived = 0;
+  let sourceLastChangeAt = null;
   const mutationWaiters = [];
   let mutationsQueued = 0;
   const activeFences = new Set;
@@ -2409,8 +2431,18 @@ var createSyncEngine = (options = {}) => {
     },
     applyChange: (table, change) => applyChange(table, change),
     connectSource: async (source) => {
-      await source.start((table, change) => applyChange(table, change));
+      await source.start((table, change) => {
+        sourceChangesReceived += 1;
+        sourceLastChangeAt = Date.now();
+        return applyChange(table, change);
+      });
+      connectedSources += 1;
+      let stopped = false;
       return async () => {
+        if (!stopped) {
+          stopped = true;
+          connectedSources -= 1;
+        }
         await source.stop();
       };
     },
@@ -2971,6 +3003,12 @@ var createSyncEngine = (options = {}) => {
         schedules: {
           registered: schedules.size
         },
+        source: {
+          changesReceived: sourceChangesReceived,
+          connected: connectedSources,
+          lastChangeAgeMs: sourceLastChangeAt === null ? null : now - sourceLastChangeAt,
+          lastChangeAt: sourceLastChangeAt
+        },
         subscriptions: {
           byCollection,
           byTenant: Object.fromEntries(subscriptionsByTenant),
@@ -3467,5 +3505,5 @@ export {
   createPresenceHub
 };
 
-//# debugId=62313B55CB7C7F6064756E2164756E21
+//# debugId=B94BDA3C56450BBB64756E2164756E21
 //# sourceMappingURL=index.js.map