feat(nbstore): new doc sync engine (#8918)

2026-07-25 14:28:51 +08:00 · 2024-12-07 08:05:02 +00:00
parent fafacdb265
commit f54f6e88cb
23 changed files with 1252 additions and 43 deletions
@@ -0,0 +1,85 @@
+import 'fake-indexeddb/auto';
+
+import { expect, test } from 'vitest';
+import { Doc as YDoc, encodeStateAsUpdate } from 'yjs';
+
+import { IndexedDBDocStorage, IndexedDBSyncStorage } from '../impls/idb';
+import { SpaceStorage } from '../storage';
+import { SyncEngine } from '../sync';
+
+test('sync', async () => {
+  const doc = new YDoc();
+  doc.getMap('test').set('hello', 'world');
+  const update = encodeStateAsUpdate(doc);
+
+  const peerADoc = new IndexedDBDocStorage({
+    id: 'ws1',
+    peer: 'a',
+    type: 'workspace',
+  });
+
+  const peerASync = new IndexedDBSyncStorage({
+    id: 'ws1',
+    peer: 'a',
+    type: 'workspace',
+  });
+
+  const peerBDoc = new IndexedDBDocStorage({
+    id: 'ws1',
+    peer: 'b',
+    type: 'workspace',
+  });
+  const peerCDoc = new IndexedDBDocStorage({
+    id: 'ws1',
+    peer: 'c',
+    type: 'workspace',
+  });
+
+  const peerA = new SpaceStorage([peerADoc, peerASync]);
+  const peerB = new SpaceStorage([peerBDoc]);
+  const peerC = new SpaceStorage([peerCDoc]);
+
+  await peerA.connect();
+  await peerB.connect();
+  await peerC.connect();
+
+  await peerA.get('doc').pushDocUpdate({
+    docId: 'doc1',
+    bin: update,
+  });
+
+  const sync = new SyncEngine(peerA, [peerB, peerC]);
+  const abort = new AbortController();
+  sync.run(abort.signal);
+
+  await new Promise(resolve => setTimeout(resolve, 1000));
+
+  {
+    const b = await peerB.get('doc').getDoc('doc1');
+    expect(b).not.toBeNull();
+    expect(b?.bin).toEqual(update);
+
+    const c = await peerC.get('doc').getDoc('doc1');
+    expect(c).not.toBeNull();
+    expect(c?.bin).toEqual(update);
+  }
+
+  doc.getMap('test').set('foo', 'bar');
+  const update2 = encodeStateAsUpdate(doc);
+  await peerC.get('doc').pushDocUpdate({
+    docId: 'doc1',
+    bin: update2,
+  });
+
+  await new Promise(resolve => setTimeout(resolve, 1000));
+
+  {
+    const a = await peerA.get('doc').getDoc('doc1');
+    expect(a).not.toBeNull();
+    expect(a?.bin).toEqual(update2);
+
+    const c = await peerC.get('doc').getDoc('doc1');
+    expect(c).not.toBeNull();
+    expect(c?.bin).toEqual(update2);
+  }
+});