fix(prometheus): bootstrap for egg (#1045)

midwayjs · May 17, 2021 · aeb0888 · aeb0888
1 parent 7d3eede
commit aeb0888
Show file tree

Hide file tree

Showing 3 changed files with 51 additions and 4 deletions.
diff --git a/packages/prometheus/package.json b/packages/prometheus/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@midwayjs/prometheus",
-  "version": "2.10.15",
+  "version": "2.10.15-beta.3",
   "description": "midway component for prometheus",
   "main": "dist/index.js",
   "typings": "dist/index.d.ts",
@@ -31,5 +31,8 @@
   "dependencies": {
     "prom-client": "^13.1.0",
     "request": "^2.88.2"
+  },
+  "engines": {
+    "node": ">= 12.0.0"
   }
 }
diff --git a/packages/prometheus/src/configuration.ts b/packages/prometheus/src/configuration.ts
@@ -7,7 +7,7 @@ import {
 } from '@midwayjs/decorator';
 import { join } from 'path';
 import * as PromClient from 'prom-client';
-import { isMaster } from './utils/utils';
+import { isMaster, closeLock } from './utils/utils';
 import * as os from 'os';
 import * as path from 'path';
 import * as fs from 'fs';
@@ -31,8 +31,12 @@ export class AutoConfiguration {
     PromClient.collectDefaultMetrics(this.prometheusConfig);
     const modules = listModule('prometheus:master');
     const handlers = {};
-    const sockFile = path.join(os.tmpdir(), 'midway-master.sock');
-    if (modules.length > 0) {
+    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);
@@ -87,6 +91,7 @@ export class AutoConfiguration {
 
   async onStop() {
     if (isMaster()) {
+      closeLock();
       this.http_server.close();
     }
   }

diff --git a/packages/prometheus/src/utils/utils.ts b/packages/prometheus/src/utils/utils.ts
@@ -1,10 +1,41 @@
 import * as cluster from 'cluster';
+import * as path from 'path';
+import * as os from 'os';
+import * as fs from 'fs';
 
+let gfp = null;
+let bInit = false;
+const ppid = process.ppid;
+const now = new Date();
+const lockFile = path.join(
+  os.tmpdir(),
+  `midway-master-${now.getFullYear()}-${
+    now.getMonth() + 1
+  }-${now.getDate()}-${ppid}.lock`
+);
 export function isMaster() {
   if (cluster.isMaster) {
     return true;
   }
 
+  if (process.argv[1].indexOf('egg-cluster') >= 0) {
+    // Is run with egg-scripts
+    if (bInit && gfp) {
+      return true;
+    } else if (bInit) {
+      return false;
+    } else {
+      bInit = true;
+      try {
+        const result = fs.openSync(lockFile, 'wx');
+        gfp = result;
+        return true;
+      } catch (e) {
+        return false;
+      }
+    }
+  }
+
   if (process.env && process.env.pm_id) {
     //Is run with PM2
     if (parseInt(process.env.NODE_APP_INSTANCE) === 0) {
@@ -14,3 +45,11 @@ export function isMaster() {
 
   return false;
 }
+
+export function closeLock() {
+  if (gfp) {
+    fs.closeSync(gfp);
+    fs.unlinkSync(lockFile);
+    gfp = null;
+  }
+}