tinybase 5.4.0-beta.1 → 5.4.0-beta.3

package/umd/synchronizers/synchronizer-ws-server/index.js

@@ -11,6 +11,8 @@
 
   const EMPTY_STRING = '';
   const UTF8 = 'utf8';
+  const MESSAGE = 'message';
+  const ERROR = 'error';
   const UNDEFINED = '\uFFFC';
   const strMatch = (str, regex) => str?.match(regex);
   const strSplit = (str, separator = EMPTY_STRING, limit) =>
@@ -18,6 +20,9 @@
 
   const promise = Promise;
   const GLOBAL = globalThis;
+  const THOUSAND = 1e3;
+  const startTimeout = (callback, sec = 0) =>
+    setTimeout(callback, sec * THOUSAND);
   const math = Math;
   const mathFloor = math.floor;
   const isUndefined = (thing) => thing == void 0;
@@ -140,14 +145,14 @@
     );
   const createPayload = (toClientId, ...args) =>
     createRawPayload(toClientId ?? EMPTY_STRING, jsonStringWithUndefined(args));
-  const createRawPayload = (toClientId, remainder) =>
-    toClientId + MESSAGE_SEPARATOR + remainder;
+  const createRawPayload = (clientId, remainder) =>
+    clientId + MESSAGE_SEPARATOR + remainder;
 
-  const newStamp = (value, time) => (time ? [value, time] : [value]);
+  const stampNew = (value, time) => (time ? [value, time] : [value]);
   const getLatestTime = (time1, time2) =>
     /* istanbul ignore next */
     ((time1 ?? '') > (time2 ?? '') ? time1 : time2) ?? '';
-  const stampNewObj = (time = EMPTY_STRING) => newStamp(objNew(), time);
+  const stampNewObj = (time = EMPTY_STRING) => stampNew(objNew(), time);
 
   const setNew = (entryOrEntries) =>
     new Set(
@@ -247,12 +252,16 @@
 
   const scheduleRunning = mapNew();
   const scheduleActions = mapNew();
-  const getStoreFunctions = (persist = 1 /* StoreOnly */, store) =>
+  const getStoreFunctions = (
+    persist = 1 /* StoreOnly */,
+    store,
+    isSynchronizer,
+  ) =>
     persist != 1 /* StoreOnly */ && store.isMergeable()
       ? [
           1,
           store.getMergeableContent,
-          store.getTransactionMergeableChanges,
+          () => store.getTransactionMergeableChanges(!isSynchronizer),
           ([[changedTables], [changedValues]]) =>
             !objIsEmpty(changedTables) || !objIsEmpty(changedValues),
           store.setDefaultContent,
@@ -276,6 +285,7 @@
     onIgnoredError,
     persist,
     extra = {},
+    isSynchronizer = 0,
     scheduleId = [],
   ) => {
     let status = 0; /* Idle */
@@ -293,7 +303,7 @@
       getChanges,
       hasChanges,
       setDefaultContent,
-    ] = getStoreFunctions(persist, store);
+    ] = getStoreFunctions(persist, store, isSynchronizer);
     const [addListener, callListeners, delListenerImpl] = getListenerFunctions(
       () => persister,
     );
@@ -314,7 +324,10 @@
         ) {
           try {
             await action();
-          } catch (error) {}
+          } catch (error) {
+            /* istanbul ignore next */
+            onIgnoredError?.(error);
+          }
         }
         mapSet(scheduleRunning, scheduleId, 0);
       }
@@ -341,9 +354,10 @@
             } else if (initialContent) {
               setDefaultContent(initialContent);
             } else {
-              errorNew(`Content is not an array ${content}`);
+              errorNew(`Content is not an array: ${content}`);
             }
           } catch (error) {
+            onIgnoredError?.(error);
             if (initialContent) {
               setDefaultContent(initialContent);
             }
@@ -354,7 +368,8 @@
       return persister;
     };
     const startAutoLoad = async (initialContent) => {
-      await stopAutoLoad().load(initialContent);
+      stopAutoLoad();
+      await load(initialContent);
       try {
         autoLoadHandle = await addPersisterListener(
           async (content, changes) => {
@@ -371,7 +386,10 @@
             }
           },
         );
-      } catch (error) {}
+      } catch (error) {
+        /* istanbul ignore next */
+        onIgnoredError?.(error);
+      }
       return persister;
     };
     const stopAutoLoad = () => {
@@ -390,14 +408,18 @@
         await schedule(async () => {
           try {
             await setPersisted(getContent, changes);
-          } catch (error) {}
+          } catch (error) {
+            /* istanbul ignore next */
+            onIgnoredError?.(error);
+          }
           setStatus(0 /* Idle */);
         });
       }
       return persister;
     };
     const startAutoSave = async () => {
-      await stopAutoSave().save();
+      stopAutoSave();
+      await save();
       autoSaveListenerId = store.addDidFinishTransactionListener(() => {
         const changes = getChanges();
         if (hasChanges(changes)) {
@@ -407,8 +429,10 @@
       return persister;
     };
     const stopAutoSave = () => {
-      ifNotUndefined(autoSaveListenerId, store.delListener);
-      autoSaveListenerId = void 0;
+      if (autoSaveListenerId) {
+        store.delListener(autoSaveListenerId);
+        autoSaveListenerId = void 0;
+      }
       return persister;
     };
     const isAutoSaving = () => !isUndefined(autoSaveListenerId);
@@ -479,30 +503,32 @@
     onIgnoredError,
     extra = {},
   ) => {
+    let syncing = 0;
     let persisterListener;
     let sends = 0;
     let receives = 0;
     const pendingRequests = mapNew();
+    const getTransactionId = () => getUniqueId(11);
     const sendImpl = (toClientId, requestId, message, body) => {
       sends++;
       send(toClientId, requestId, message, body);
     };
-    const request = async (toClientId, message, body = EMPTY_STRING) =>
+    const request = async (toClientId, message, body, transactionId) =>
       promiseNew((resolve, reject) => {
-        const requestId = getUniqueId();
-        const timeout = setTimeout(() => {
+        const requestId = transactionId + '.' + getUniqueId(4);
+        const timeout = startTimeout(() => {
           collDel(pendingRequests, requestId);
           reject(
             `No response from ${toClientId ?? 'anyone'} to ${requestId}, ` +
               message,
           );
-        }, requestTimeoutSeconds * 1e3);
+        }, requestTimeoutSeconds);
         mapSet(pendingRequests, requestId, [
           toClientId,
           (response, fromClientId) => {
             clearTimeout(timeout);
             collDel(pendingRequests, requestId);
-            resolve([response, fromClientId]);
+            resolve([response, fromClientId, transactionId]);
           },
         ]);
         sendImpl(toClientId, requestId, message, body);
@@ -515,7 +541,7 @@
           objForEach(
             cellStamps2,
             ([cell2, cellTime2], cellId) =>
-              (rowStamp[0][cellId] = newStamp(cell2, cellTime2)),
+              (rowStamp[0][cellId] = stampNew(cell2, cellTime2)),
           );
           rowStamp[1] = getLatestTime(rowStamp[1], rowTime2);
         });
@@ -526,80 +552,99 @@
     const getChangesFromOtherStore = async (
       otherClientId = null,
       otherContentHashes,
+      transactionId = getTransactionId(),
     ) => {
-      if (isUndefined(otherContentHashes)) {
-        [otherContentHashes, otherClientId] = await request(
-          otherClientId,
-          1 /* GetContentHashes */,
-        );
-      }
-      const [otherTablesHash, otherValuesHash] = otherContentHashes;
-      const [tablesHash, valuesHash] = store.getMergeableContentHashes();
-      let tablesChanges = stampNewObj();
-      if (tablesHash != otherTablesHash) {
-        const [newTables, differentTableHashes] = (
-          await request(
-            otherClientId,
-            4 /* GetTableDiff */,
-            store.getMergeableTableHashes(),
-          )
-        )[0];
-        tablesChanges = newTables;
-        if (!objIsEmpty(differentTableHashes)) {
-          const [newRows, differentRowHashes] = (
+      try {
+        if (isUndefined(otherContentHashes)) {
+          [otherContentHashes, otherClientId, transactionId] = await request(
+            null,
+            1 /* GetContentHashes */,
+            EMPTY_STRING,
+            transactionId,
+          );
+        }
+        const [otherTablesHash, otherValuesHash] = otherContentHashes;
+        const [tablesHash, valuesHash] = store.getMergeableContentHashes();
+        let tablesChanges = stampNewObj();
+        if (tablesHash != otherTablesHash) {
+          const [newTables, differentTableHashes] = (
             await request(
               otherClientId,
-              5 /* GetRowDiff */,
-              store.getMergeableRowHashes(differentTableHashes),
+              4 /* GetTableDiff */,
+              store.getMergeableTableHashes(),
+              transactionId,
             )
           )[0];
-          mergeTablesStamps(tablesChanges, newRows);
-          if (!objIsEmpty(differentRowHashes)) {
-            const newCells = (
+          tablesChanges = newTables;
+          if (!objIsEmpty(differentTableHashes)) {
+            const [newRows, differentRowHashes] = (
               await request(
                 otherClientId,
-                6 /* GetCellDiff */,
-                store.getMergeableCellHashes(differentRowHashes),
+                5 /* GetRowDiff */,
+                store.getMergeableRowHashes(differentTableHashes),
+                transactionId,
               )
             )[0];
-            mergeTablesStamps(tablesChanges, newCells);
+            mergeTablesStamps(tablesChanges, newRows);
+            if (!objIsEmpty(differentRowHashes)) {
+              const newCells = (
+                await request(
+                  otherClientId,
+                  6 /* GetCellDiff */,
+                  store.getMergeableCellHashes(differentRowHashes),
+                  transactionId,
+                )
+              )[0];
+              mergeTablesStamps(tablesChanges, newCells);
+            }
           }
         }
+        return [
+          tablesChanges,
+          valuesHash == otherValuesHash
+            ? stampNewObj()
+            : (
+                await request(
+                  otherClientId,
+                  7 /* GetValueDiff */,
+                  store.getMergeableValueHashes(),
+                  transactionId,
+                )
+              )[0],
+          1,
+        ];
+      } catch (error) {
+        onIgnoredError?.(error);
       }
-      return [
-        tablesChanges,
-        valuesHash == otherValuesHash
-          ? stampNewObj()
-          : (
-              await request(
-                otherClientId,
-                7 /* GetValueDiff */,
-                store.getMergeableValueHashes(),
-              )
-            )[0],
-        1,
-      ];
     };
     const getPersisted = async () => {
       const changes = await getChangesFromOtherStore();
-      return !objIsEmpty(changes[0][0]) || !objIsEmpty(changes[1][0])
+      return changes &&
+        (!objIsEmpty(changes[0][0]) || !objIsEmpty(changes[1][0]))
         ? changes
         : void 0;
     };
     const setPersisted = async (_getContent, changes) =>
       changes
-        ? sendImpl(null, null, 3 /* ContentDiff */, changes)
+        ? sendImpl(null, getTransactionId(), 3 /* ContentDiff */, changes)
         : sendImpl(
             null,
-            null,
+            getTransactionId(),
             2 /* ContentHashes */,
             store.getMergeableContentHashes(),
           );
     const addPersisterListener = (listener) => (persisterListener = listener);
     const delPersisterListener = () => (persisterListener = void 0);
-    const startSync = async (initialContent) =>
-      await (await persister.startAutoLoad(initialContent)).startAutoSave();
-    const stopSync = () => persister.stopAutoLoad().stopAutoSave();
+    const startSync = async (initialContent) => {
+      syncing = 1;
+      return await (
+        await persister.startAutoLoad(initialContent)
+      ).startAutoSave();
+    };
+    const stopSync = () => {
+      syncing = 0;
+      return persister.stopAutoLoad().stopAutoSave();
+    };
     const destroy = () => {
       return persister.stopSync();
     };
@@ -614,30 +659,36 @@
       2,
       // MergeableStoreOnly
       {startSync, stopSync, destroy, getSynchronizerStats, ...extra},
+      1,
     );
-    registerReceive((fromClientId, requestId, message, body) => {
+    registerReceive((fromClientId, transactionOrRequestId, message, body) => {
+      const isAutoLoading = syncing || persister.isAutoLoading();
       receives++;
       if (message == 0 /* Response */) {
         ifNotUndefined(
-          mapGet(pendingRequests, requestId),
+          mapGet(pendingRequests, transactionOrRequestId),
           ([toClientId, handleResponse]) =>
             isUndefined(toClientId) || toClientId == fromClientId
               ? handleResponse(body, fromClientId)
               : /* istanbul ignore next */
                 0,
         );
-      } else if (
-        message == 2 /* ContentHashes */ &&
-        persister.isAutoLoading()
-      ) {
-        getChangesFromOtherStore(fromClientId, body).then((changes) => {
-          persisterListener?.(void 0, changes);
-        });
-      } else if (message == 3 /* ContentDiff */ && persister.isAutoLoading()) {
+      } else if (message == 2 /* ContentHashes */ && isAutoLoading) {
+        getChangesFromOtherStore(
+          fromClientId,
+          body,
+          transactionOrRequestId ?? void 0,
+        )
+          .then((changes) => {
+            persisterListener?.(void 0, changes);
+          })
+          .catch(onIgnoredError);
+      } else if (message == 3 /* ContentDiff */ && isAutoLoading) {
         persisterListener?.(void 0, body);
       } else {
         ifNotUndefined(
-          message == 1 /* GetContentHashes */ && persister.isAutoSaving()
+          message == 1 /* GetContentHashes */ &&
+            (syncing || persister.isAutoSaving())
             ? store.getMergeableContentHashes()
             : message == 4 /* GetTableDiff */
               ? store.getMergeableTableDiff(body)
@@ -649,7 +700,12 @@
                     ? store.getMergeableValueDiff(body)
                     : void 0,
           (response) => {
-            sendImpl(fromClientId, requestId, 0 /* Response */, response);
+            sendImpl(
+              fromClientId,
+              transactionOrRequestId,
+              0 /* Response */,
+              response,
+            );
           },
         );
       }
@@ -659,7 +715,11 @@
 
   const PATH_REGEX = /\/([^?]*)/;
   const SERVER_CLIENT_ID = 'S';
-  const createWsServer = (webSocketServer, createPersisterForPath) => {
+  const createWsServer = (
+    webSocketServer,
+    createPersisterForPath,
+    onIgnoredError,
+  ) => {
     const pathIdListeners = mapNew();
     const clientIdListeners = mapNew();
     const clientsByPath = mapNew();
@@ -691,6 +751,9 @@
                 receivePayload(payload, receive)),
             () => {},
             1,
+            void 0,
+            void 0,
+            onIgnoredError,
           );
           serverClient[4 /* Buffer */] = [];
           serverClient[5 /* Then */] = isArray(persisterMaybeThen)
@@ -719,10 +782,8 @@
           if (clientId !== SERVER_CLIENT_ID) {
             serverClient[3 /* Send */]?.(forwardedPayload);
           }
-          mapForEach(clients, (otherClientId, otherWebSocket) =>
-            otherClientId !== clientId
-              ? otherWebSocket.send(forwardedPayload)
-              : 0,
+          mapForEach(clients, (otherClientId, otherClient) =>
+            otherClientId !== clientId ? otherClient.send(forwardedPayload) : 0,
           );
         } else if (toClientId === SERVER_CLIENT_ID) {
           serverClient[3 /* Send */]?.(forwardedPayload);
@@ -730,7 +791,7 @@
           mapGet(clients, toClientId)?.send(forwardedPayload);
         }
       });
-    webSocketServer.on('connection', (webSocket, request) =>
+    webSocketServer.on('connection', (client, request) =>
       ifNotUndefined(strMatch(request.url, PATH_REGEX), ([, pathId]) =>
         ifNotUndefined(
           request.headers['sec-websocket-key'],
@@ -748,9 +809,9 @@
               callListeners(pathIdListeners, void 0, pathId, 1);
               await configureServerClient(serverClient, pathId, clients);
             }
-            mapSet(clients, clientId, webSocket);
+            mapSet(clients, clientId, client);
             callListeners(clientIdListeners, [pathId], clientId, 1);
-            webSocket.on('message', (data) => {
+            client.on(MESSAGE, (data) => {
               const payload = data.toString(UTF8);
               if (serverClient[0 /* State */] == 0 /* Ready */) {
                 messageHandler(payload);
@@ -763,7 +824,7 @@
               arrayForEach(serverClient[4 /* Buffer */], messageHandler);
               serverClient[4 /* Buffer */] = [];
             }
-            webSocket.on('close', () => {
+            client.on('close', () => {
               collDel(clients, clientId);
               callListeners(clientIdListeners, [pathId], clientId, -1);
               if (collIsEmpty(clients)) {
@@ -773,13 +834,19 @@
                 callListeners(pathIdListeners, void 0, pathId, -1);
               }
             });
+            if (onIgnoredError) {
+              client.on(ERROR, onIgnoredError);
+            }
           },
         ),
       ),
     );
+    if (onIgnoredError) {
+      webSocketServer.on(ERROR, onIgnoredError);
+    }
     const getWebSocketServer = () => webSocketServer;
     const getPathIds = () => mapKeys(clientsByPath);
-    const getClientIds = (path) => mapKeys(mapGet(clientsByPath, path));
+    const getClientIds = (pathId) => mapKeys(mapGet(clientsByPath, pathId));
     const addPathIdsListener = (listener) =>
       addListener(listener, pathIdListeners);
     const addClientIdsListener = (pathId, listener) =>