Revert "feat: enhance credit hooks to trigger updates on store changes"

This reverts commit 7851a715a3.
This commit is contained in:
javayhu 2025-08-25 10:01:54 +08:00
parent 0ae3f27c78
commit 1c0c46fa34
2 changed files with 2 additions and 55 deletions

View File

@ -2,10 +2,8 @@ import { consumeCreditsAction } from '@/actions/consume-credits';
import { getCreditBalanceAction } from '@/actions/get-credit-balance';
import { getCreditStatsAction } from '@/actions/get-credit-stats';
import { getCreditTransactionsAction } from '@/actions/get-credit-transactions';
import { useCreditsStore } from '@/stores/credits-store';
import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
import type { SortingState } from '@tanstack/react-table';
import { useEffect } from 'react';
// Query keys
export const creditsKeys = {
@ -23,9 +21,7 @@ export const creditsKeys = {
// Hook to fetch credit balance
export function useCreditBalance() {
const updateTrigger = useCreditsStore((state) => state.updateTrigger);
const query = useQuery({
return useQuery({
queryKey: creditsKeys.balance(),
queryFn: async () => {
console.log('Fetching credit balance...');
@ -39,23 +35,11 @@ export function useCreditBalance() {
return result.data.credits || 0;
},
});
// Refetch when updateTrigger changes
useEffect(() => {
if (updateTrigger > 0) {
console.log('Credits update triggered, refetching balance...');
query.refetch();
}
}, [updateTrigger, query]);
return query;
}
// Hook to fetch credit statistics
export function useCreditStats() {
const updateTrigger = useCreditsStore((state) => state.updateTrigger);
const query = useQuery({
return useQuery({
queryKey: creditsKeys.stats(),
queryFn: async () => {
console.log('Fetching credit stats...');
@ -67,22 +51,11 @@ export function useCreditStats() {
return result.data.data;
},
});
// Refetch when updateTrigger changes
useEffect(() => {
if (updateTrigger > 0) {
console.log('Credits update triggered, refetching stats...');
query.refetch();
}
}, [updateTrigger, query]);
return query;
}
// Hook to consume credits
export function useConsumeCredits() {
const queryClient = useQueryClient();
const triggerUpdate = useCreditsStore((state) => state.triggerUpdate);
return useMutation({
mutationFn: async ({
@ -102,9 +75,6 @@ export function useConsumeCredits() {
return result.data;
},
onSuccess: () => {
// Trigger credits update in store to notify all components
triggerUpdate();
// Invalidate credit balance and stats after consuming credits
queryClient.invalidateQueries({
queryKey: creditsKeys.balance(),

View File

@ -1,23 +0,0 @@
import { create } from 'zustand';
interface CreditsState {
// trigger for credit updates, incremented each time credits change
updateTrigger: number;
// method to trigger credit updates
triggerUpdate: () => void;
}
/**
* Credits store for managing credit balance updates.
*
* This store provides a simple trigger mechanism to notify components
* when credits have been consumed or updated, ensuring UI components can
* refetch the latest credit balance.
*/
export const useCreditsStore = create<CreditsState>((set) => ({
updateTrigger: 0,
triggerUpdate: () =>
set((state) => ({
updateTrigger: state.updateTrigger + 1,
})),
}));