feat: update

dream-num · Nov 23, 2024 · a489c7c · a489c7c
1 parent 487bc1a
commit a489c7c
Showing 1 changed file with 21 additions and 3 deletions.
diff --git a/packages/sheets-data-validation/src/services/dv-cache.service.ts b/packages/sheets-data-validation/src/services/dv-cache.service.ts
@@ -15,8 +15,9 @@
  */
 
 import type { DataValidationStatus, IRange, ISheetDataValidationRule, Nullable } from '@univerjs/core';
-import { Disposable, ICommandService, Inject, ObjectMatrix, Range } from '@univerjs/core';
-import { type ISetRangeValuesMutationParams, SetRangeValuesMutation } from '@univerjs/sheets';
+import type { IRemoveSheetMutationParams, ISetRangeValuesMutationParams } from '@univerjs/sheets';
+import { Disposable, ICommandService, Inject, IUniverInstanceService, ObjectMatrix, Range, UniverInstanceType } from '@univerjs/core';
+import { RemoveSheetMutation, SetRangeValuesMutation } from '@univerjs/sheets';
 import { Subject } from 'rxjs';
 
 export class DataValidationCacheService extends Disposable {
@@ -26,10 +27,12 @@ export class DataValidationCacheService extends Disposable {
     readonly dirtyRanges$ = this._dirtyRanges$.asObservable();
 
     constructor(
-        @Inject(ICommandService) private readonly _commandService: ICommandService
+        @Inject(ICommandService) private readonly _commandService: ICommandService,
+        @Inject(IUniverInstanceService) private readonly _univerInstanceService: IUniverInstanceService
     ) {
         super();
         this._initDirtyRanges();
+        this._initSheetRemove();
     }
 
     private _initDirtyRanges() {
@@ -45,6 +48,21 @@ export class DataValidationCacheService extends Disposable {
         }));
     }
 
+    private _initSheetRemove() {
+        this.disposeWithMe(this._commandService.onCommandExecuted((commandInfo) => {
+            if (commandInfo.id === RemoveSheetMutation.id) {
+                const { unitId, subUnitId } = commandInfo.params as IRemoveSheetMutationParams;
+                this._cacheMatrix.get(unitId)?.delete(subUnitId);
+            }
+        }));
+
+        this.disposeWithMe(this._univerInstanceService.unitDisposed$.subscribe((univerInstance) => {
+            if (univerInstance.type === UniverInstanceType.UNIVER_SHEET) {
+                this._cacheMatrix.delete(univerInstance.getUnitId());
+            }
+        }));
+    }
+
     private _ensureCache(unitId: string, subUnitId: string) {
         let unitMap = this._cacheMatrix.get(unitId);