do It (#305)

pyth-network · Dec 15, 2023 · ff8e6be · ff8e6be
1 parent 3b16426
commit ff8e6be
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/frontend/hooks/useJoinDaoLlcMutation.ts b/frontend/hooks/useJoinDaoLlcMutation.ts
@@ -1,8 +1,11 @@
 import { StakeAccount, StakeConnection } from '@pythnetwork/staking'
 import toast from 'react-hot-toast'
-import { useMutation } from 'react-query'
+import { useMutation, useQueryClient } from 'react-query'
+import { StakeConnectionQueryKey } from './useStakeConnection'
 
 export function useJoinDaoLlcMutation() {
+  const queryClient = useQueryClient()
+
   return useMutation(
     ['sign-llc-mutation'],
     async ({
@@ -16,6 +19,13 @@ export function useJoinDaoLlcMutation() {
       toast.success(`Successfully signed LLC agreement!`)
     },
     {
+      onSuccess() {
+        // invalidate all except stake connection
+        queryClient.invalidateQueries({
+          predicate: (query) => query.queryKey[0] !== StakeConnectionQueryKey,
+        })
+      },
+
       onError(error: Error) {
         toast.error(error.message)
       },

diff --git a/staking/app/StakeConnection.ts b/staking/app/StakeConnection.ts
@@ -598,7 +598,7 @@ export class StakeConnection {
   }
 
   /**
-   * Join the DAO LLC for the give stake account.
+   * Join the DAO LLC for the given stake account.
    */
   public async joinDaoLlc(stakeAccount: StakeAccount) {
     const transaction: Transaction = new Transaction();