Class Server

packages/message/server.ts:129–232

};

export class Server {
  private apiFunctionMap: Map<string, ApiFunction> = new Map();

  private logger = LoggerCore.getInstance().logger({ service: "messageServer" });

  constructor(
    prefix: string,
    msgReceiver: Message | Message[],
    private enableConnect: boolean = true
  ) {
    const msgReceiverList = Array.isArray(msgReceiver) ? msgReceiver : [msgReceiver];
    if (this.enableConnect) {
      msgReceiverList.forEach((msg) => {
        msg.onConnect((msg: TMessage, con: MessageConnect) => {
          if (typeof msg.action !== "string") return;
          this.logger.trace("server onConnect", { msg });
          if (msg.action?.startsWith(prefix)) {
            return this.connectHandle(msg.action.slice(prefix.length + 1), msg.data, con);
          }
          return false;
        });
      });
    }

    msgReceiverList.forEach((msg) => {
      msg.onMessage((msg: TMessage, sendResponse, sender) => {
        if (typeof msg.action !== "string") return;
        this.logger.trace("server onMessage", { msg: msg as any });
        if (msg.action?.startsWith(prefix)) {
          return this.messageHandle(msg.action.slice(prefix.length + 1), msg.data, sendResponse, sender);
        }
      });
      return false;
    });
  }

  group(name: string, middleware?: MiddlewareFunction) {
    return new Group(this, name, middleware);
  }

  on(name: string, func: ApiFunction) {
    this.apiFunctionMap.set(name, func);
  }

  private connectHandle(msg: string, params: any, con: MessageConnect) {
    const func = this.apiFunctionMap.get(msg);
    if (func) {
      const ret = func(params, new SenderConnect(con));
      if (ret) {
        if (ret instanceof Promise) {
          ret
            .then((data) => {
              data && con.sendMessage({ code: 0, data });
            })
            .catch((e: Error) => {
              con.sendMessage({ code: -1, message: formatErrorToClient(e) });
              this.logger.error("connectHandle error", Logger.E(e));
            });