feat(nbstore): share worker between workspaces (#9947)

packages/frontend/core/src/modules/userspace/entities/current-user-db.ts

@@ -1,8 +1,9 @@
 import { Entity, LiveData } from '@toeverything/infra';
-import { finalize, of, switchMap } from 'rxjs';
+import { Observable, of, switchMap } from 'rxjs';
 
 import type { AuthService } from '../../cloud';
 import type { UserspaceService } from '../services/userspace';
+import type { UserDBWithTables } from './user-db';
 
 export class CurrentUserDB extends Entity {
   constructor(
@@ -19,11 +20,12 @@ export class CurrentUserDB extends Entity {
         switchMap(userId => {
           if (userId) {
             const ref = this.userDBService.openDB(userId);
-            return of(ref.obj).pipe(
-              finalize(() => {
+            return new Observable<UserDBWithTables>(subscriber => {
+              subscriber.next(ref.obj);
+              return () => {
                 ref.release();
-              })
-            );
+              };
+            });
           } else {
             return of(null);
           }

packages/frontend/core/src/modules/userspace/entities/user-db-engine.ts

@@ -1,6 +1,6 @@
 import { IndexedDBDocStorage } from '@affine/nbstore/idb';
 import { SqliteDocStorage } from '@affine/nbstore/sqlite';
-import type { WorkerClient } from '@affine/nbstore/worker/client';
+import type { StoreClient } from '@affine/nbstore/worker/client';
 import { Entity } from '@toeverything/infra';
 
 import type { ServerService } from '../../cloud';
@@ -10,7 +10,7 @@ export class UserDBEngine extends Entity<{
   userId: string;
 }> {
   private readonly userId = this.props.userId;
-  readonly client: WorkerClient;
+  readonly client: StoreClient;
 
   DocStorageType =
     BUILD_CONFIG.isElectron || BUILD_CONFIG.isIOS

packages/frontend/core/src/modules/userspace/entities/user-db.ts

@@ -39,6 +39,10 @@ export class UserDB extends Entity<{
       });
     });
   }
+
+  override dispose(): void {
+    this.engine.dispose();
+  }
 }
 
 export type UserDBWithTables = UserDB & {