fix: add agent to invalidate queries on restore backup (#704)

krud-dev · Jul 18, 2023 · 54d9ebc · 54d9ebc
1 parent 1ca2648
commit 54d9ebc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/app/src/renderer/apis/requests/backup/importBackup.ts b/app/src/renderer/apis/requests/backup/importBackup.ts
@@ -9,6 +9,7 @@ import { crudKeys } from '../crud/crudKeys';
 import { instanceCrudEntity } from '../crud/entity/entities/instance.crudEntity';
 import { applicationCrudEntity } from '../crud/entity/entities/application.crudEntity';
 import { folderCrudEntity } from '../crud/entity/entities/folder.crudEntity';
+import { agentCrudEntity } from 'renderer/apis/requests/crud/entity/entities/agent.crudEntity';
 
 type Variables = {
   jsonData: string;
@@ -34,6 +35,7 @@ export const useImportBackup = (
     invalidateQueriesKeysFn: (data, variables) => [
       crudKeys.entity(instanceCrudEntity),
       crudKeys.entity(applicationCrudEntity),
+      crudKeys.entity(agentCrudEntity),
       crudKeys.entity(folderCrudEntity),
     ],
   });
diff --git a/app/src/renderer/apis/requests/system-backup/restoreSystemBackup.ts b/app/src/renderer/apis/requests/system-backup/restoreSystemBackup.ts
@@ -9,6 +9,7 @@ import { crudKeys } from 'renderer/apis/requests/crud/crudKeys';
 import { instanceCrudEntity } from 'renderer/apis/requests/crud/entity/entities/instance.crudEntity';
 import { applicationCrudEntity } from 'renderer/apis/requests/crud/entity/entities/application.crudEntity';
 import { folderCrudEntity } from 'renderer/apis/requests/crud/entity/entities/folder.crudEntity';
+import { agentCrudEntity } from 'renderer/apis/requests/crud/entity/entities/agent.crudEntity';
 
 type Variables = {
   fileName: string;
@@ -34,6 +35,7 @@ export const useRestoreSystemBackup = (
     invalidateQueriesKeysFn: (data, variables) => [
       crudKeys.entity(instanceCrudEntity),
       crudKeys.entity(applicationCrudEntity),
+      crudKeys.entity(agentCrudEntity),
       crudKeys.entity(folderCrudEntity),
     ],
   });