feat: cache workspace instances

packages/data-center/src/datacenter.ts

@@ -1,6 +1,6 @@
 import { Workspaces } from './workspaces';
 import type { WorkspacesChangeEvent } from './workspaces';
-import { BlobStorage, Workspace } from '@blocksuite/store';
+import { Workspace } from '@blocksuite/store';
 import { BaseProvider } from './provider/base';
 import { LocalProvider } from './provider/local/local';
 import { AffineProvider } from './provider';
@@ -17,6 +17,7 @@ import { applyUpdate, encodeStateAsUpdate } from 'yjs';
 export class DataCenter {
   private readonly _workspaces = new Workspaces();
   private readonly _logger = getLogger('dc');
+  private _workspaceInstances: Map<string, Workspace> = new Map();
   /**
    * A mainProvider must exist as the only data trustworthy source.
    */
@@ -109,9 +110,14 @@ export class DataCenter {
    * @param {string} workspaceId workspace id
    */
   private _getWorkspace(workspaceId: string) {
-    return new Workspace({
-      room: workspaceId,
-    }).register(BlockSchema);
+    const workspaceInfo = this._workspaces.find(workspaceId);
+    assert(workspaceInfo, 'Workspace not found');
+    return (
+      this._workspaceInstances.get(workspaceId) ||
+      new Workspace({
+        room: workspaceId,
+      }).register(BlockSchema)
+    );
   }
 
   /**
@@ -149,8 +155,9 @@ export class DataCenter {
     const provider = this.providerMap.get(workspaceInfo.provider);
     assert(provider, `provide '${workspaceInfo.provider}' is not registered`);
     this._logger(`Loading ${workspaceInfo.provider} workspace: `, workspaceId);
-
-    return await provider.warpWorkspace(this._getWorkspace(workspaceId));
+    const workspace = this._getWorkspace(workspaceId);
+    this._workspaceInstances.set(workspaceId, workspace);
+    return await provider.warpWorkspace(workspace);
   }
 
   /**
@@ -332,21 +339,23 @@ export class DataCenter {
     return;
   }
 
-  // /**
-  //  * get blob url by workspaces id
-  //  * @param id
-  //  * @returns {Promise<string | null>} blob url
-  //  */
-  // async getBlob(id: string): Promise<string | null> {
-  //   return await this._blobStorage.get(id);
-  // }
+  /**
+   * get blob url by workspaces id
+   * @param id
+   * @returns {Promise<string | null>} blob url
+   */
+  async getBlob(workspace: Workspace, id: string): Promise<string | null> {
+    const blob = await workspace.blobs;
+    return (await blob?.get(id)) || '';
+  }
 
-  // /**
-  //  * up load blob and get a blob url
-  //  * @param id
-  //  * @returns {Promise<string | null>} blob url
-  //  */
-  // async setBlob(blob: Blob): Promise<string> {
-  //   return await this._blobStorage.set(blob);
-  // }
+  /**
+   * up load blob and get a blob url
+   * @param id
+   * @returns {Promise<string | null>} blob url
+   */
+  async setBlob(workspace: Workspace, blob: Blob): Promise<string> {
+    const blobStorage = await workspace.blobs;
+    return (await blobStorage?.set(blob)) || '';
+  }
 }

packages/data-center/src/provider/affine/affine.ts

@@ -69,7 +69,7 @@ export class AffineProvider extends BaseProvider {
   override async warpWorkspace(workspace: Workspace) {
     const { doc, room } = workspace;
     assert(room);
-    this._initWorkspaceDb(workspace);
+    this.linkLocal(workspace);
     const updates = await downloadWorkspace(room);
     if (updates) {
       await new Promise(resolve => {
@@ -77,8 +77,12 @@ export class AffineProvider extends BaseProvider {
         applyUpdate(doc, new Uint8Array(updates));
       });
     }
-    const ws = new WebsocketProvider('/', room, doc);
-    this._wsMap.set(room, ws);
+    let ws = this._wsMap.get(room);
+    if (!ws) {
+      ws = new WebsocketProvider('/', room, doc);
+      this._wsMap.set(room, ws);
+    }
+    ws.connect();
     await new Promise<void>((resolve, reject) => {
       // TODO: synced will also be triggered on reconnection after losing sync
       // There needs to be an event mechanism to emit the synchronization state to the upper layer
@@ -231,7 +235,7 @@ export class AffineProvider extends BaseProvider {
     return await removeMember({ permissionId });
   }
 
-  private async _initWorkspaceDb(workspace: Workspace) {
+  public override async linkLocal(workspace: Workspace) {
     assert(workspace.room);
     let idb = this._idbMap.get(workspace.room);
     idb?.destroy();
@@ -239,7 +243,7 @@ export class AffineProvider extends BaseProvider {
     this._idbMap.set(workspace.room, idb);
     await idb.whenSynced;
     this._logger('Local data loaded');
-    return idb;
+    return workspace;
   }
 
   public override async createWorkspace(
@@ -252,7 +256,7 @@ export class AffineProvider extends BaseProvider {
       room: id,
     }).register(BlockSchema);
     nw.meta.setName(meta.name);
-    this._initWorkspaceDb(nw);
+    this.linkLocal(nw);
 
     const workspaceInfo: WS = {
       name: meta.name,

packages/data-center/src/provider/base.ts

@@ -169,4 +169,13 @@ export class BaseProvider {
     email;
     return null;
   }
+
+  /**
+   * link workspace to local caches
+   * @param workspace
+   * @returns
+   */
+  public async linkLocal(workspace: Workspace): Promise<Workspace> {
+    return workspace;
+  }
 }

packages/data-center/src/provider/local/local.ts

@@ -22,21 +22,21 @@ export class LocalProvider extends BaseProvider {
     storage.setItem(WORKSPACE_KEY, JSON.stringify(workspaces));
   }
 
-  private async _initWorkspaceDb(workspace: Workspace) {
+  public override async linkLocal(workspace: Workspace) {
     assert(workspace.room);
     let idb = this._idbMap.get(workspace.room);
     idb?.destroy();
     idb = new IndexedDBProvider(workspace.room, workspace.doc);
     this._idbMap.set(workspace.room, idb);
     this._logger('Local data loaded');
-    return idb;
+    return workspace;
   }
 
   public override async warpWorkspace(
     workspace: Workspace
   ): Promise<Workspace> {
     assert(workspace.room);
-    await this._initWorkspaceDb(workspace);
+    await this.linkLocal(workspace);
     return workspace;
   }
 
@@ -93,7 +93,7 @@ export class LocalProvider extends BaseProvider {
     };
 
     const workspace = new Workspace({ room: workspaceInfo.id });
-    this._initWorkspaceDb(workspace);
+    this.linkLocal(workspace);
     workspace.meta.setName(meta.name);
     if (!meta.avatar) {
       // set default avatar