增加容器健康检查

back/schedule/client.ts

@@ -2,14 +2,14 @@ import { credentials } from '@grpc/grpc-js';
 import {
   AddCronRequest,
   AddCronResponse,
-  CronServiceClient,
+  CronClient,
   DeleteCronRequest,
   DeleteCronResponse,
 } from '../protos/cron';
 import config from '../config';
 
 class Client {
-  private client = new CronServiceClient(
+  private client = new CronClient(
     `localhost:${config.cronPort}`,
     credentials.createInsecure(),
   );

back/schedule/health.ts

@@ -0,0 +1,30 @@
+import { ServerUnaryCall, sendUnaryData } from '@grpc/grpc-js';
+import { HealthCheckRequest, HealthCheckResponse } from '../protos/health';
+import { exec } from 'child_process';
+import config from '../config';
+import { promiseExec } from '../config/util';
+
+const check = async (
+  call: ServerUnaryCall<HealthCheckRequest, HealthCheckResponse>,
+  callback: sendUnaryData<HealthCheckResponse>,
+) => {
+  switch (call.request.service) {
+    case 'cron':
+      const res = await promiseExec(
+        `curl -sf http://localhost:${config.port}/api/system`,
+      );
+      console.log(res);
+      if (res.includes('200')) {
+        return callback(null, { status: 1 });
+      }
+      const errLog = await promiseExec(
+        `tail -n 300 ~/.pm2/logs/panel-error.log`,
+      );
+      return callback(new Error(errLog));
+
+    default:
+      return callback(null, { status: 1 });
+  }
+};
+
+export { check };

back/schedule/index.ts

@@ -1,12 +1,15 @@
 import { Server, ServerCredentials } from '@grpc/grpc-js';
-import { CronServiceService } from '../protos/cron';
+import { CronService } from '../protos/cron';
 import { addCron } from './addCron';
 import { delCron } from './delCron';
+import { HealthService } from '../protos/health';
+import { check } from './health';
 import config from '../config';
 import Logger from '../loaders/logger';
 
 const server = new Server();
-server.addService(CronServiceService, { addCron, delCron });
+server.addService(HealthService, { check });
+server.addService(CronService, { addCron, delCron });
 server.bindAsync(
   `localhost:${config.cronPort}`,
   ServerCredentials.createInsecure(),