Add queryInvalidation on creating workspace

yorkie-team · Jan 24, 2024 · e2c980e · e2c980e
1 parent d208591
commit e2c980e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/frontend/src/components/modals/CreateModal.tsx b/frontend/src/components/modals/CreateModal.tsx
@@ -88,7 +88,7 @@ function CreateModal(props: CreateModalProps) {
 								<TextFieldElement
 									variant="standard"
 									name="title"
-									label="Title of new note"
+									label={`Title of New ${title}`}
 									required
 									fullWidth
 									inputProps={{

diff --git a/frontend/src/hooks/api/workspace.ts b/frontend/src/hooks/api/workspace.ts
@@ -1,4 +1,4 @@
-import { useInfiniteQuery, useMutation, useQuery } from "@tanstack/react-query";
+import { useInfiniteQuery, useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
 import axios from "axios";
 import {
 	CreateWorkspaceInviteTokenRequest,
@@ -55,12 +55,19 @@ export const useGetWorkspaceListQuery = () => {
 };
 
 export const useCreateWorkspaceMutation = () => {
+	const queryClient = useQueryClient();
+
 	return useMutation({
 		mutationFn: async (data: CreateWorkspaceRequest) => {
 			const res = await axios.post<CreateWorkspaceResponse>("/workspaces", data);
 
 			return res.data;
 		},
+		onSuccess: () => {
+			queryClient.invalidateQueries({
+				queryKey: generateGetWorkspaceListQueryKey(),
+			});
+		},
 	});
 };