From c5c0a1f39d4331e5864aef7803dc504926011da5 Mon Sep 17 00:00:00 2001
From: Kuizuo <hi@kuizuo.cn>
Date: Thu, 3 Aug 2023 12:54:27 +0800
Subject: [PATCH] feat: logging interceptor

---
 .../src/interceptors/logging.interceptor.ts   | 36 +++++++++++++++++++
 apps/api/src/main.ts                          | 35 +++++++++++++-----
 2 files changed, 62 insertions(+), 9 deletions(-)
 create mode 100644 apps/api/src/interceptors/logging.interceptor.ts

diff --git a/apps/api/src/interceptors/logging.interceptor.ts b/apps/api/src/interceptors/logging.interceptor.ts
new file mode 100644
index 0000000..4efa6f1
--- /dev/null
+++ b/apps/api/src/interceptors/logging.interceptor.ts
@@ -0,0 +1,36 @@
+import {
+  CallHandler,
+  ExecutionContext,
+  Injectable,
+  Logger,
+  NestInterceptor,
+} from '@nestjs/common';
+import { Observable, tap } from 'rxjs';
+
+@Injectable()
+export class LoggingInterceptor implements NestInterceptor {
+  private logger: Logger;
+
+  constructor() {
+    this.logger = new Logger(LoggingInterceptor.name, { timestamp: false });
+  }
+
+  intercept(
+    context: ExecutionContext,
+    next: CallHandler<any>,
+  ): Observable<any> {
+    const call$ = next.handle();
+    const request = context.switchToHttp().getRequest();
+    const content = `${request.method} -> ${request.url}`;
+    this.logger.debug(`+++ 请求：${content}`);
+    const now = Date.now();
+
+    return call$.pipe(
+      tap(() =>
+        this.logger.debug(
+          `--- 响应请求：${content}${` +${Date.now() - now}ms`}`,
+        ),
+      ),
+    );
+  }
+}
diff --git a/apps/api/src/main.ts b/apps/api/src/main.ts
index a17ac7c..5b8fb90 100644
--- a/apps/api/src/main.ts
+++ b/apps/api/src/main.ts
@@ -1,4 +1,6 @@
+import cluster from 'cluster';
 import path from 'path';
+import { performance } from 'perf_hooks';
 
 import {
   ClassSerializerInterceptor,
@@ -22,6 +24,8 @@ import { AppModule } from './app.module';
 
 import { IAppConfig } from './config';
 import { AppFilter } from './filters/app.filter';
+import { isDev, isMainProcess } from './global/env';
+import { LoggingInterceptor } from './interceptors/logging.interceptor';
 import { TimeoutInterceptor } from './interceptors/timeout.interceptor';
 import { TransformInterceptor } from './interceptors/transform.interceptor';
 import { AppLoggerService } from './modules/shared/services/app-logger.service';
@@ -68,11 +72,13 @@ async function bootstrap() {
   app.useGlobalFilters(new AppFilter());
 
   app.useGlobalInterceptors(
-    // 请求超时处理
+    // 请求超时
     new TimeoutInterceptor(30000),
-    // 序列化处理
+    // 序列化
     new ClassSerializerInterceptor(reflector),
-    // 返回数据处理
+    // Logging
+    isDev ? new LoggingInterceptor() : null,
+    // 返回数据转换
     new TransformInterceptor(new Reflector()),
   );
 
@@ -105,14 +111,25 @@ async function bootstrap() {
 
   setupSwagger(app, configService);
 
-  await app.listen(port, '0.0.0.0');
+  await app.listen(port, '0.0.0.0', async () => {
+    const url = await app.getUrl();
+    const { pid } = process;
+    const env = cluster.isPrimary;
+    const prefix = env ? 'P' : 'W';
 
-  // started log
-  const logger = new Logger('NestApplication');
-  logger.log(`Server running on ${await app.getUrl()}`);
+    if (!isMainProcess) {
+      return;
+    }
+
+    const logger = new Logger('NestApplication');
+    logger.log(`[${prefix + pid}] Server running on ${url}`);
+
+    if (isDev) {
+      logger.log(`[${prefix + pid}] OpenApi: ${url}/api-docs`);
+    }
+    logger.log(`Server is up. ${`+${performance.now() | 0}ms`}`);
+  });
 
-  // eslint-disable-next-line @typescript-eslint/ban-ts-comment
-  // @ts-ignore
   if (module.hot) {
     module.hot.accept();
     module.hot.dispose(() => app.close());