susbcribe tests pass on local gateways

fireproof-storage · Sep 20, 2024 · fa8b28f · fa8b28f
1 parent 5742164
commit fa8b28f
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 5 deletions.
diff --git a/src/blockstore/fragment-gateway.ts b/src/blockstore/fragment-gateway.ts
@@ -171,7 +171,27 @@ export class FragmentGateway implements bs.Gateway {
     if (this.innerGW.subscribe) {
       return this.innerGW.subscribe(url, callback);
     } else {
-      return Result.Err(this.logger.Error().Msg("Subscribe not supported").AsError());
+      let lastData: Uint8Array | undefined = undefined;
+      let interval = 100;
+      const fetchData = async () => {
+        const result = await this.innerGW.get(url);
+        if (result.isOk()) {
+          const data = result.Ok();
+          if (!lastData || !data.every((value, index) => lastData && value === lastData[index])) {
+            lastData = data;
+            callback(data);
+            interval = 100; // Reset interval when data changes
+          } else {
+            interval *= 2; // Double the interval when data is unchanged
+          }
+        }
+        timeoutId = setTimeout(fetchData, interval);
+      };
+      let timeoutId = setTimeout(fetchData, interval);
+
+      return Result.Ok(() => {
+        clearTimeout(timeoutId);
+      });
     }
   }
 

diff --git a/tests/fireproof/all-gateway.test.ts b/tests/fireproof/all-gateway.test.ts
@@ -330,19 +330,17 @@ describe("noop Gateway subscribe", function () {
       const metaSubscribeResult = await metaGateway?.subscribe?.(metaUrl?.Ok(), async (data: Uint8Array) => {
         const decodedData = new TextDecoder().decode(data);
         expect(decodedData).toContain("parents");
-        didCall = true
+        didCall = true;
         resolve();
       });
       expect(metaSubscribeResult?.Ok()).toBeTruthy();
       const ok = await db.put({ _id: "key1", hello: "world1" });
       expect(ok).toBeTruthy();
       expect(ok.id).toBe("key1");
-      await p
+      await p;
       expect(didCall).toBeTruthy();
     }
   });
-
-  });
 });
 
 describe("Gateway", function () {