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

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

@@ -1,11 +1,17 @@
 const EMPTY_STRING = '';
 const UTF8 = 'utf8';
+const OPEN = 'open';
+const MESSAGE = 'message';
+const ERROR = 'error';
 const UNDEFINED = '\uFFFC';
 const strSplit = (str, separator = EMPTY_STRING, limit) =>
   str.split(separator, 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;
@@ -76,8 +82,8 @@ 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 collSize = (coll) => coll?.size ?? 0;
 const collHas = (coll, keyOrValue) => coll?.has(keyOrValue) ?? false;
@@ -119,11 +125,11 @@ const visitTree = (node, path, ensureLeaf, pruneLeaf, p = 0) =>
     },
   );
 
-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(
@@ -221,12 +227,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,
@@ -250,6 +260,7 @@ const createCustomPersister = (
   onIgnoredError,
   persist,
   extra = {},
+  isSynchronizer = 0,
   scheduleId = [],
 ) => {
   let status = 0; /* Idle */
@@ -267,7 +278,7 @@ const createCustomPersister = (
     getChanges,
     hasChanges,
     setDefaultContent,
-  ] = getStoreFunctions(persist, store);
+  ] = getStoreFunctions(persist, store, isSynchronizer);
   const [addListener, callListeners, delListenerImpl] = getListenerFunctions(
     () => persister,
   );
@@ -316,7 +327,7 @@ 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);
@@ -330,7 +341,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) {
@@ -377,7 +389,8 @@ const createCustomPersister = (
     return persister;
   };
   const startAutoSave = async () => {
-    await stopAutoSave().save();
+    stopAutoSave();
+    await save();
     autoSaveListenerId = store.addDidFinishTransactionListener(() => {
       const changes = getChanges();
       if (hasChanges(changes)) {
@@ -387,8 +400,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);
@@ -459,31 +474,33 @@ 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++;
     onSend?.(toClientId, requestId, message, body);
     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);
@@ -496,7 +513,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);
       });
@@ -507,80 +524,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 = () => {
     destroyImpl();
     return persister.stopSync();
@@ -596,28 +631,37 @@ 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++;
-    onReceive?.(fromClientId, requestId, message, body);
+    onReceive?.(fromClientId, transactionOrRequestId, message, body);
     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)
@@ -629,7 +673,12 @@ const createCustomSynchronizer = (
                   ? store.getMergeableValueDiff(body)
                   : void 0,
         (response) => {
-          sendImpl(fromClientId, requestId, 0 /* Response */, response);
+          sendImpl(
+            fromClientId,
+            transactionOrRequestId,
+            0 /* Response */,
+            response,
+          );
         },
       );
     }
@@ -645,10 +694,12 @@ const createWsSynchronizer = async (
   onReceive,
   onIgnoredError,
 ) => {
-  const addEventListener = (event, handler) =>
+  const addEventListener = (event, handler) => {
     webSocket.addEventListener(event, handler);
+    return () => webSocket.removeEventListener(event, handler);
+  };
   const registerReceive = (receive) =>
-    addEventListener('message', ({data}) =>
+    addEventListener(MESSAGE, ({data}) =>
       receivePayload(data.toString(UTF8), receive),
     );
   const send = (toClientId, ...args) =>
@@ -667,10 +718,18 @@ const createWsSynchronizer = async (
     onIgnoredError,
     {getWebSocket: () => webSocket},
   );
-  return promiseNew((resolve, reject) => {
+  return promiseNew((resolve) => {
     if (webSocket.readyState != webSocket.OPEN) {
-      addEventListener('error', reject);
-      addEventListener('open', () => resolve(synchronizer));
+      const onAttempt = (error) => {
+        if (error) {
+          onIgnoredError?.(error);
+        }
+        removeOpenListener();
+        removeErrorListener();
+        resolve(synchronizer);
+      };
+      const removeOpenListener = addEventListener(OPEN, () => onAttempt());
+      const removeErrorListener = addEventListener(ERROR, onAttempt);
     } else {
       resolve(synchronizer);
     }