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

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

@@ -1,5 +1,7 @@
 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) =>
@@ -7,6 +9,9 @@ const strSplit = (str, separator = EMPTY_STRING, limit) =>
 
 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;
@@ -125,14 +130,14 @@ const receivePayload = (payload, receive) =>
   );
 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(
@@ -230,12 +235,16 @@ const getListenerFunctions = (getThing) => {
 
 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,
@@ -259,6 +268,7 @@ const createCustomPersister = (
   onIgnoredError,
   persist,
   extra = {},
+  isSynchronizer = 0,
   scheduleId = [],
 ) => {
   let status = 0; /* Idle */
@@ -276,7 +286,7 @@ const createCustomPersister = (
     getChanges,
     hasChanges,
     setDefaultContent,
-  ] = getStoreFunctions(persist, store);
+  ] = getStoreFunctions(persist, store, isSynchronizer);
   const [addListener, callListeners, delListenerImpl] = getListenerFunctions(
     () => persister,
   );
@@ -295,7 +305,10 @@ const createCustomPersister = (
       ) {
         try {
           await action();
-        } catch (error) {}
+        } catch (error) {
+          /* istanbul ignore next */
+          onIgnoredError?.(error);
+        }
       }
       mapSet(scheduleRunning, scheduleId, 0);
     }
@@ -322,9 +335,10 @@ const createCustomPersister = (
           } 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);
           }
@@ -335,7 +349,8 @@ const createCustomPersister = (
     return persister;
   };
   const startAutoLoad = async (initialContent) => {
-    await stopAutoLoad().load(initialContent);
+    stopAutoLoad();
+    await load(initialContent);
     try {
       autoLoadHandle = await addPersisterListener(async (content, changes) => {
         if (changes || content) {
@@ -350,7 +365,10 @@ const createCustomPersister = (
           await load();
         }
       });
-    } catch (error) {}
+    } catch (error) {
+      /* istanbul ignore next */
+      onIgnoredError?.(error);
+    }
     return persister;
   };
   const stopAutoLoad = () => {
@@ -369,14 +387,18 @@ const createCustomPersister = (
       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)) {
@@ -386,8 +408,10 @@ const createCustomPersister = (
     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);
@@ -458,30 +482,32 @@ const createCustomSynchronizer = (
   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);
@@ -494,7 +520,7 @@ const createCustomSynchronizer = (
         objForEach(
           cellStamps2,
           ([cell2, cellTime2], cellId) =>
-            (rowStamp[0][cellId] = newStamp(cell2, cellTime2)),
+            (rowStamp[0][cellId] = stampNew(cell2, cellTime2)),
         );
         rowStamp[1] = getLatestTime(rowStamp[1], rowTime2);
       });
@@ -505,80 +531,98 @@ const createCustomSynchronizer = (
   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();
   };
@@ -593,27 +637,36 @@ const createCustomSynchronizer = (
     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)
@@ -625,7 +678,12 @@ const createCustomSynchronizer = (
                   ? store.getMergeableValueDiff(body)
                   : void 0,
         (response) => {
-          sendImpl(fromClientId, requestId, 0 /* Response */, response);
+          sendImpl(
+            fromClientId,
+            transactionOrRequestId,
+            0 /* Response */,
+            response,
+          );
         },
       );
     }
@@ -635,7 +693,11 @@ const createCustomSynchronizer = (
 
 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();
@@ -665,6 +727,9 @@ const createWsServer = (webSocketServer, createPersisterForPath) => {
               receivePayload(payload, receive)),
           () => {},
           1,
+          void 0,
+          void 0,
+          onIgnoredError,
         );
         serverClient[4 /* Buffer */] = [];
         serverClient[5 /* Then */] = isArray(persisterMaybeThen)
@@ -693,10 +758,8 @@ const createWsServer = (webSocketServer, createPersisterForPath) => {
         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);
@@ -704,7 +767,7 @@ const createWsServer = (webSocketServer, createPersisterForPath) => {
         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'], async (clientId) => {
         const clients = mapEnsure(clientsByPath, pathId, mapNew);
@@ -720,9 +783,9 @@ const createWsServer = (webSocketServer, createPersisterForPath) => {
           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);
@@ -735,7 +798,7 @@ const createWsServer = (webSocketServer, createPersisterForPath) => {
           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)) {
@@ -745,12 +808,18 @@ const createWsServer = (webSocketServer, createPersisterForPath) => {
             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) =>