diff --git a/src/hooks/useEmployees.js b/src/hooks/useEmployees.js index b911d2dc..40db5105 100644 --- a/src/hooks/useEmployees.js +++ b/src/hooks/useEmployees.js @@ -241,9 +241,9 @@ export const useSuspendEmployee = ({ (store) => store.localVariables.projectId ); return useMutation({ - mutationFn: (id) => { + mutationFn: async (employeeId) => { setemployeeLodaing(true); - return EmployeeRepository.deleteEmployee(id); + return await EmployeeRepository.deleteEmployee(employeeId); }, onSuccess: () => { @@ -254,14 +254,21 @@ export const useSuspendEmployee = ({ }); showToast("Employee deleted successfully.", "success"); setIsDeleteModalOpen(false); + + // Invalidate only the required employee-related queries + queryClient.invalidateQueries({ queryKey: ["employee", employeeId] }); + queryClient.invalidateQueries({ queryKey: ["allEmployees"] }); + + if (selectedProjectId) { + queryClient.invalidateQueries({ queryKey: ["projectEmployees", selectedProjectId] }); + } }, onError: (error) => { - const message = - error.response?.data?.message || - error.message || - "An unexpected error occurred"; - showToast(message, "error"); + showToast( + error.response?.data?.message || error.message || "An unexpected error occurred", + "error" + ); setIsDeleteModalOpen(false); },