qinglong/back/services/grpc.ts

import { Server, ServerCredentials } from '@grpc/grpc-js';
import { CronService } from '../protos/cron';
import { HealthService } from '../protos/health';
import { ApiService } from '../protos/api';
import { addCron } from '../schedule/addCron';
import { delCron } from '../schedule/delCron';
import { check } from '../schedule/health';
import * as Api from '../schedule/api';
import Logger from '../loaders/logger';
import { promisify } from 'util';
import config from '../config';
import { metricsService } from './metrics';
import { Service } from 'typedi';

@Service()
export class GrpcServerService {
  private server: Server = new Server({ 'grpc.enable_http_proxy': 0 });

  async initialize() {
    try {
      this.server.addService(HealthService, { check });
      this.server.addService(CronService, { addCron, delCron });
      this.server.addService(ApiService, Api);

      const grpcPort = config.grpcPort;
      const bindAsync = promisify(this.server.bindAsync).bind(this.server);
      await bindAsync(
        `[::]:${grpcPort}`,
        ServerCredentials.createInsecure(),
      );
      Logger.debug(`✌️ gRPC service started successfully`);

      metricsService.record('grpc_service_start', 1, {
        port: grpcPort.toString(),
      });

      return grpcPort;
    } catch (err) {
      Logger.error('Failed to start gRPC service:', err);
      throw err;
    }
  }

  async shutdown() {
    try {
      if (this.server) {
        await new Promise((resolve) => {
          this.server.tryShutdown(() => {
            Logger.debug('gRPC service stopped');
            metricsService.record('grpc_service_stop', 1);
            resolve(null);
          });
        });
      }
    } catch (err) {
      Logger.error('Error while shutting down gRPC service:', err);
      throw err;
    }
  }

  getServer() {
    return this.server;
  }
}