Function useSocket

src/client/api/socketio.ts:63–110

()

>[1];

export function useSocket() {
  const socket = useSocketStore((state) => state.socket);

  const emit = useEvent(
    async <T extends keyof SocketEventMap>(
      eventName: T,
      eventData: SocketEventData<T>
    ): Promise<SocketEventRet<T>> => {
      if (!socket) {
        throw new Error('Socketio not init');
      }

      return await socket.emitWithAck(eventName, eventData);
    }
  );

  const subscribe = useEvent(
    <T extends keyof SubscribeEventMap>(
      name: T,
      onData: (data: SubscribeEventData<T>) => void
    ) => {
      if (!socket) {
        throw new Error('Socketio not init');
      }

      const p = emit('$subscribe', { name });

      const receiveDataUpdate = (data: any) => {
        onData(data);
      };

      const unsubscribe = () => {
        Promise.resolve(p).then((cursor) => {
          emit('$unsubscribe', { name, cursor });
          socket.off(`${name}#${cursor}` as string, receiveDataUpdate);
        });
      };

      Promise.resolve(p).then((cursor) => {
        socket.on(`${name}#${cursor}` as string, receiveDataUpdate);
      });

      return unsubscribe;
    }
  );

  return { socket, emit, subscribe };
}

export function useSocketSubscribe<K extends keyof SubscribeEventMap>(
  name: K,