Update logic to listen for a restoration completion event
This commit is contained in:
parent
f558bc880a
commit
c8ecbe6e79
9 changed files with 68 additions and 48 deletions
|
@ -1,7 +1,8 @@
|
|||
import { ServerContext } from '@/state/server';
|
||||
import { useEffect, useRef } from 'react';
|
||||
import { SocketEvent } from '@/components/server/events';
|
||||
|
||||
const useWebsocketEvent = (event: string, callback: (data: string) => void) => {
|
||||
const useWebsocketEvent = (event: SocketEvent, callback: (data: string) => void) => {
|
||||
const { connected, instance } = ServerContext.useStoreState(state => state.socket);
|
||||
const savedCallback = useRef<any>(null);
|
||||
|
||||
|
@ -10,7 +11,7 @@ const useWebsocketEvent = (event: string, callback: (data: string) => void) => {
|
|||
}, [ callback ]);
|
||||
|
||||
return useEffect(() => {
|
||||
const eventListener = (event: any) => savedCallback.current(event);
|
||||
const eventListener = (event: SocketEvent) => savedCallback.current(event);
|
||||
if (connected && instance) {
|
||||
instance.addListener(event, eventListener);
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue