fix: ReplicationStore schemaGC

dineug · Dec 30, 2023 · f426e3f · f426e3f
1 parent f00c23e
commit f426e3f
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/packages/erd-editor-app/src/services/indexeddb/modules/schema/service.ts b/packages/erd-editor-app/src/services/indexeddb/modules/schema/service.ts
@@ -41,6 +41,10 @@ export class SchemaService {
     const prev = this.cache.get(id);
     if (prev && result) {
       this.cache.set(id, { ...prev, ...entityValue });
+
+      if (entityValue.value) {
+        prev.store.setInitialValue(entityValue.value);
+      }
     }
 
     return result;

diff --git a/packages/erd-editor/src/engine/replication-store.ts b/packages/erd-editor/src/engine/replication-store.ts
@@ -11,6 +11,8 @@ import { actionsFilter } from '@/engine/rx-operators';
 import { createStore } from '@/engine/store';
 import { createHooks } from '@/engine/store-hooks';
 import { Unsubscribe, ValuesType } from '@/internal-types';
+import { procGC } from '@/services/schema-gc/procGC';
+import { SchemaGCService } from '@/services/schema-gc/schemaGCService';
 import { toSafeString } from '@/utils/validation';
 
 type ReducerRecord = {
@@ -45,6 +47,7 @@ export function createReplicationStore(
     store.subscribe(actions => subscriber.next(actions))
   ).pipe(actionsFilter(ChangeActionTypes), debounceTime(200));
   const observers = new Set<Partial<ReducerRecord>>();
+  const schemaGCService = new SchemaGCService();
 
   const on = (reducers: Partial<ReducerRecord>): Unsubscribe => {
     observers.has(reducers) || observers.add(reducers);
@@ -69,6 +72,20 @@ export function createReplicationStore(
     store.dispatchSync(
       initialLoadJsonAction$(isEmpty(safeValue) ? '{}' : safeValue)
     );
+    schemaGCService.run(toJson(store.state)).then(gcIds => {
+      const isChange =
+        gcIds.tableIds.length ||
+        gcIds.tableColumnIds.length ||
+        gcIds.relationshipIds.length ||
+        gcIds.indexIds.length ||
+        gcIds.indexColumnIds.length ||
+        gcIds.memoIds.length;
+
+      if (isChange) {
+        procGC(store.state, gcIds);
+        emit(InternalActionType.change, undefined);
+      }
+    });
   };
 
   const dispatchSync = (actions: Array<AnyAction> | AnyAction) => {