Update react-query-service.ts

bsu3338 · Apr 12, 2024 · be9e53c · be9e53c
1 parent 39de165
commit be9e53c
Showing 1 changed file with 34 additions and 0 deletions.
diff --git a/packages/data-provider/src/react-query/react-query-service.ts b/packages/data-provider/src/react-query/react-query-service.ts
@@ -394,6 +394,40 @@ export const useUpdateUserPluginsMutation = (): UseMutationResult<
   });
 };
 
+export const usePreferenceQuery = (
+  name: string,
+  config?: UseQueryOptions<t.TPreference>,
+): QueryObserverResult<
+  t.TPreference
+> => {
+  return useQuery<t.TPreference>(
+    [QueryKeys.name, name],
+    () => {
+      return dataService.getUserPreference(name);
+    },
+    {
+      refetchOnWindowFocus: false,
+      refetchOnReconnect: false,
+      refetchOnMount: false,
+      retry: false,
+    },
+  );
+};
+
+export const useUpdatePreferenceMutation = (): UseMutationResult<
+  t.TUser,
+  unknown,
+  t.TPreference,
+  unknown
+> => {
+  const queryClient = useQueryClient();
+  return useMutation((payload: t.TPreference) => dataService.updateUserPreference(payload), {
+    onSuccess: (data, variables) => {
+      queryClient.invalidateQueries([QueryKeys.name, variables.name]);
+    },
+  });
+};
+
 export const useGetStartupConfig = (): QueryObserverResult<t.TStartupConfig> => {
   return useQuery<t.TStartupConfig>(
     [QueryKeys.startupConfig],