Class Group

packages/message/server.ts:234–288

}

export class Group {
  private middlewares: MiddlewareFunction[] = [];

  constructor(
    private server: Server,
    private name: string,
    middleware?: MiddlewareFunction
  ) {
    if (!name.endsWith("/") && name.length > 0) {
      this.name += "/";
    }
    if (middleware) {
      this.middlewares.push(middleware);
    }
  }

  group(name: string, middleware?: MiddlewareFunction) {
    const newGroup = new Group(this.server, `${this.name}${name}`, middleware);
    // 继承父级的中间件
    newGroup.middlewares = [...this.middlewares, ...newGroup.middlewares];
    return newGroup;
  }

  use(middleware: MiddlewareFunction): Group {
    const newGroup = new Group(this.server, `${this.name}`, middleware);
    newGroup.middlewares = [...this.middlewares, ...newGroup.middlewares];
    return newGroup;
  }

  on(name: string, func: ApiFunction) {
    const fullName = `${this.name}${name}`;

    if (this.middlewares.length === 0) {
      // 没有中间件，直接注册
      this.server.on(fullName, func);
    } else {
      // 有中间件，需要包装处理函数
      this.server.on(fullName, async (params: any, con: IGetSender) => {
        let index = 0;

        const next = async (): Promise<any> => {
          if (index < this.middlewares.length) {
            const middleware = this.middlewares[index++];
            return await middleware(params, con, next);
          } else {
            // 所有中间件都执行完毕，执行最终的处理函数
            return await func(params, con);
          }
        };

        return await next();
      });
    }
  }
}

// 转发消息
export function forwardMessage(