Method onReady

packages/prometheus/src/configuration.ts:36–99

()

  http_server: any;

  async onReady() {
    PromClient.collectDefaultMetrics(this.prometheusConfig);
    const modules = DecoratorManager.listModule('prometheus:master');
    const handlers = {};
    let sockFile = path.join(os.tmpdir(), 'midway-master.sock');
    if (process.platform === 'win32') {
      sockFile =
        '\\\\.\\pipe\\' + sockFile.replace(/^\//, '').replace(/\//g, '-');
    }
    if (modules.length > 0 && process.platform !== 'win32') {
      if (isMaster()) {
        if (fs.existsSync(sockFile)) {
          fs.unlinkSync(sockFile);
        }
        this.http_server = http
          .createServer((req, res) => {
            const query = qs.parse(req.url.slice('/?'.length));
            const params = JSON.parse(query.params as string);
            handlers[`${query.path}`](...params).then(result => {
              res.end(result);
            });
          })
          .listen(sockFile);
      }
    }
    for (const module of modules) {
      const rules = MetadataManager.getOwnMetadata(
        'prometheus:master:options',
        module
      );
      for (const rule of rules) {
        if (isMaster()) {
          handlers[rule.name] = async (...args) => {
            const service = await this.app
              .createAnonymousContext()
              .requestContext.getAsync(module);
            return rule.value.call(service, ...args);
          };
        }
      }
    }
    this.app.use(async (ctx, next) => {
      const service = await ctx.requestContext.getAsync(DataService);
      const startAt = process.hrtime();
      try {
        if (ctx.path === '/metrics') {
          const result = await service.getData();
          const Register = PromClient.register;
          ctx.set('Content-Type', Register.contentType);
          ctx.body = result;
        } else {
          await next();
        }
        const diff = process.hrtime(startAt);
        const time = diff[0] * 1e3 + diff[1] * 1e-6;
        service.getUser(ctx.method, '200', ctx.path, time);
      } catch (e) {
        const diff = process.hrtime(startAt);