Class ZodSocketConnection

packages/core/src/v3/zodSocket.ts:317–400

}

export class ZodSocketConnection<
  TClientMessages extends ZodSocketMessageCatalogSchema,
  TServerMessages extends ZodSocketMessageCatalogSchema,
> {
  #sender: ZodSocketMessageSender<TClientMessages>;
  socket: ZodSocket<TServerMessages, TClientMessages>;

  #handler: ZodSocketMessageHandler<TServerMessages>;
  #logger: StructuredLogger;

  constructor(opts: ZodSocketConnectionOptions<TClientMessages, TServerMessages>) {
    const uri = `${opts.secure ? "wss" : "ws"}://${opts.host}:${
      opts.port ?? (opts.secure ? "443" : "80")
    }/${opts.namespace}`;

    const logger = new SimpleStructuredLogger(opts.namespace, LogLevel.info);
    logger.log("new zod socket", { uri });

    this.socket = io(uri, {
      transports: ["websocket"],
      auth: {
        token: opts.authToken,
      },
      extraHeaders: opts.extraHeaders,
      reconnectionDelay: 500,
      reconnectionDelayMax: 1000,
    });

    this.#logger = logger.child({
      socketId: this.socket.id,
    });

    this.#handler = new ZodSocketMessageHandler({
      schema: opts.serverMessages,
      handlers: opts.handlers,
    });
    this.#handler.registerHandlers(this.socket, this.#logger);

    this.#sender = new ZodSocketMessageSender({
      schema: opts.clientMessages,
      socket: this.socket,
    });

    this.socket.on("connect_error", async (error) => {
      this.#logger.error(`connect_error: ${error}`);

      if (opts.onError) {
        await opts.onError(this.socket, error, this.#logger);
      }
    });

    this.socket.on("connect", async () => {
      this.#logger.info("connect");

      if (opts.onConnection) {
        await opts.onConnection(this.socket, this.#handler, this.#sender, this.#logger);
      }
    });