Load CPU and memory usage

Author: ivanvs

.prettierignore

@@ -0,0 +1 @@
+src/public/**

examples/test-status-monitor/src/main.ts

@@ -3,6 +3,6 @@ import { AppModule } from './app.module';
 
 async function bootstrap() {
   const app = await NestFactory.create(AppModule);
-  await app.listen(3000);
+  await app.listen(3001);
 }
 bootstrap();

package-lock.json

@@ -27,21 +27,22 @@
     "typescript": "^3.0.3"
   },
   "dependencies": {
+    "@nestjs/websockets": "^5.4.0",
     "debug": "^2.6.8",
     "handlebars": "~4.0.12",
     "on-headers": "^1.0.1",
     "pidusage": "^1.1.6",
+    "reflect-metadata": "^0.1.12",
     "request-promise-native": "^1.0.5",
-    "socket.io": "^2.0.3",
-    "reflect-metadata": "^0.1.12"
+    "socket.io": "^2.0.3"
   },
   "scripts": {
     "test": "jest",
     "coverage": "jest --coverage",
-    "coveralls": "yarn run coverage --coverageReporters=text-lcov | coveralls",
+    "coveralls": "npm run coverage --coverageReporters=text-lcov | coveralls",
     "test:watch": "jest --watch",
     "build": "rm -rf ./dist && tsc",
-    "format": "prettier **/**/*.ts --ignore-path ./.prettierignore --write && git status",
+    "format": "prettier src/**/*.ts --ignore-path ./.prettierignore --write && git status",
     "prepublish": "npm run format && npm run build"
   },
   "jest": {

package.json

@@ -0,0 +1,34 @@
+const configuration = {
+  title: 'Express Status',
+  theme: 'default.css',
+  path: '/status',
+  spans: [
+    {
+      interval: 1,
+      retention: 60,
+    },
+    {
+      interval: 5,
+      retention: 60,
+    },
+    {
+      interval: 15,
+      retention: 60,
+    },
+  ],
+  port: null,
+  websocket: null,
+  iframe: false,
+  chartVisibility: {
+    cpu: true,
+    mem: true,
+    load: true,
+    responseTime: true,
+    rps: true,
+    statusCodes: true,
+  },
+  ignoreStartsWith: '/admin',
+  healthChecks: [],
+};
+
+export default configuration;

src/default.config.ts

@@ -6,13 +6,7 @@
     <script src="https://cdnjs.cloudflare.com/ajax/libs/Chart.js/2.2.2/Chart.bundle.min.js"></script>
     <script src="https://cdnjs.cloudflare.com/ajax/libs/socket.io/1.4.5/socket.io.min.js"></script>
     <style>
-        {
-                {
-                    {
-                    style
-                }
-            }
-        }
+        {{{style}}}
     </style>
 </head>
 
@@ -98,7 +92,7 @@ <h1>{{status}}</h1>
     </div>
     <script>
         var port = '{{port}}';
-        { { { script } } }
+        {{{script}}}
     </script>
 </body>
 

src/index.ts

@@ -7,23 +7,32 @@ const controllerPath = 'monitor';
 
 @Controller(controllerPath)
 export class StatusMonitorController {
-  @Get()
-  @HttpCode(200)
-  root() {
-    const data = {
+  data;
+  render;
+
+  constructor() {
+    this.data = {
+      title: 'Nest.js Status',
+      port: 3001,
+      bodyClasses: '',
       script: fs.readFileSync(
-        path.join(__dirname, '/public/javascripts/app.js'),
+        path.join(__dirname, '../src/public/javascripts/app.js'),
       ),
       style: fs.readFileSync(
-        path.join(__dirname, '/public/stylesheets/style.css'),
+        path.join(__dirname, '../src/public/stylesheets/style.css'),
       ),
     };
 
     const htmlTmpl = fs
-      .readFileSync(path.join(__dirname, '/public/index.html'))
+      .readFileSync(path.join(__dirname, '../src/public/index.html'))
       .toString();
 
-    const render = Handlebars.compile(htmlTmpl);
-    return render(data);
+    this.render = Handlebars.compile(htmlTmpl, { strict: true });
+  }
+
+  @Get()
+  @HttpCode(200)
+  root() {
+    return this.render(this.data);
   }
 }

src/public/index.html

@@ -0,0 +1,42 @@
+import {
+  SubscribeMessage,
+  WebSocketGateway,
+  WebSocketServer,
+  WsResponse,
+  OnGatewayConnection,
+} from '@nestjs/websockets';
+import { StatusMonitoringService } from './status.monitoring.service';
+import { Inject, forwardRef } from '@nestjs/common';
+
+@WebSocketGateway()
+export class StatusMonitorGateway implements OnGatewayConnection {
+  @WebSocketServer()
+  server;
+
+  constructor(
+    @Inject(forwardRef(() => StatusMonitoringService))
+    private readonly statusMonitoringService: StatusMonitoringService,
+  ) {}
+
+  @SubscribeMessage('esm_change')
+  onEvent(client, data: any) {
+    const event = 'esm_start';
+    const spans = this.statusMonitoringService.getData();
+    return { event, spans };
+  }
+
+  handleConnection(client) {
+    const spans = this.statusMonitoringService.getData();
+    client.emit('esm_start', spans);
+  }
+
+  sendMetrics(metrics) {
+    const data = {
+      os: metrics.os[metrics.os.length - 2],
+      responses: metrics.responses[metrics.responses.length - 2],
+      interval: metrics.interval,
+      retention: metrics.retention,
+    };
+    this.server.emit('esm_stats', data);
+  }
+}

src/status.monitor.controller.ts

@@ -1,7 +1,10 @@
 import { Module } from '@nestjs/common';
 import { StatusMonitorController } from './status.monitor.controller';
+import { StatusMonitorGateway } from './status.monitor.gateway';
+import { StatusMonitoringService } from './status.monitoring.service';
 
 @Module({
   controllers: [StatusMonitorController],
+  providers: [StatusMonitorGateway, StatusMonitoringService],
 })
 export class StatusMonitorModule {}