Invalidate related linode when removed from PG

linode · Feb 16, 2024 · cde4ef9 · cde4ef9
1 parent c01bec1
commit cde4ef9
Showing 1 changed file with 19 additions and 2 deletions.
diff --git a/packages/manager/src/queries/placementGroups.ts b/packages/manager/src/queries/placementGroups.ts
@@ -17,6 +17,7 @@ import { useMutation, useQuery, useQueryClient } from 'react-query';
 
 import { getAll } from 'src/utilities/getAll';
 
+import { queryKey as LINODES_QUERY_KEY } from './linodes/linodes';
 import { queryKey as PROFILE_QUERY_KEY } from './profile';
 
 import type {
@@ -117,7 +118,7 @@ export const useAssignLinodesToPlacementGroup = (placementGroupId: number) => {
     AssignLinodesToPlacementGroupPayload
   >({
     mutationFn: (data) => assignLinodesToPlacementGroup(placementGroupId, data),
-    onSuccess: (updatedPlacementGroup) => {
+    onSuccess: (updatedPlacementGroup, data) => {
       // Invalidate paginated placement groups
       queryClient.invalidateQueries([queryKey, 'paginated']);
 
@@ -129,6 +130,14 @@ export const useAssignLinodesToPlacementGroup = (placementGroupId: number) => {
         'linode_ids',
       ]);
 
+      // Invalidate linode placement group data
+      queryClient.invalidateQueries([
+        LINODES_QUERY_KEY,
+        'linode',
+        data.linodes[0],
+        'placement_groups',
+      ]);
+
       // Set the updated placement group
       queryClient.setQueryData(
         [queryKey, 'placement-group', placementGroupId],
@@ -149,7 +158,7 @@ export const useUnassignLinodesFromPlacementGroup = (
   >({
     mutationFn: (data) =>
       unassignLinodesFromPlacementGroup(placementGroupId, data),
-    onSuccess: (updatedPlacementGroup) => {
+    onSuccess: (updatedPlacementGroup, data) => {
       // Invalidate paginated placement groups
       queryClient.invalidateQueries([queryKey, 'paginated']);
 
@@ -160,6 +169,14 @@ export const useUnassignLinodesFromPlacementGroup = (
         placementGroupId,
       ]);
 
+      // Invalidate linode placement group data
+      queryClient.invalidateQueries([
+        LINODES_QUERY_KEY,
+        'linode',
+        data.linodes[0],
+        'placement_groups',
+      ]);
+
       // Set the updated placement group
       queryClient.setQueryData(
         [queryKey, 'placement-group', placementGroupId],