48 lines
1.3 KiB
TypeScript
48 lines
1.3 KiB
TypeScript
|
'use client';
|
||
|
|
|
||
|
|
import { useEffect } from 'react';
|
||
|
|
import { useQueryClient, type QueryKey } from '@tanstack/react-query';
|
||
|
|
|
||
|
|
import { useSocket } from '@/providers/socket-provider';
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Subscribes to socket events and invalidates React Query caches.
|
||
|
|
*
|
||
|
|
* @param eventMap - Maps socket event names to arrays of query keys to invalidate.
|
||
|
|
*
|
||
|
|
* @example
|
||
|
|
* useRealtimeInvalidation({
|
||
|
|
* 'client:created': [['clients']],
|
||
|
|
* 'client:updated': [['clients'], ['clients', clientId]],
|
||
|
|
* 'client:archived': [['clients']],
|
||
|
|
* });
|
||
|
|
*/
|
||
|
|
export function useRealtimeInvalidation(
|
||
|
|
eventMap: Record<string, QueryKey[]>,
|
||
|
|
) {
|
||
|
|
const socket = useSocket();
|
||
|
|
const queryClient = useQueryClient();
|
||
|
|
|
||
|
|
useEffect(() => {
|
||
|
|
if (!socket) return;
|
||
|
|
|
||
|
|
const handlers: Array<{ event: string; handler: (...args: unknown[]) => void }> = [];
|
||
|
|
|
||
|
|
for (const [event, queryKeys] of Object.entries(eventMap)) {
|
||
|
|
const handler = () => {
|
||
|
|
for (const key of queryKeys) {
|
||
|
|
queryClient.invalidateQueries({ queryKey: key });
|
||
|
|
}
|
||
|
|
};
|
||
|
|
socket.on(event as any, handler);
|
||
|
|
handlers.push({ event, handler });
|
||
|
|
}
|
||
|
|
|
||
|
|
return () => {
|
||
|
|
for (const { event, handler } of handlers) {
|
||
|
|
socket.off(event as any, handler);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
}, [socket, queryClient, eventMap]);
|
||
|
|
}
|