add spent time when accessing milvus

server/src/app.module.ts

@@ -12,6 +12,7 @@ import { AuthModule } from './auth/auth.module';
 import { join } from 'path';
 import { PartitionsModule } from './partitions/partitions.module';
 import { SchemaModule } from './schema/schema.module';
+import { EventsModule } from './events/events.module';
 import { LoggerMiddleware } from './middlewares/logger';
 
 @Module({
@@ -26,6 +27,7 @@ import { LoggerMiddleware } from './middlewares/logger';
     AuthModule,
     PartitionsModule,
     SchemaModule,
+    EventsModule,
   ],
   controllers: [AppController],
   providers: [

server/src/collections/collections.controller.ts

@@ -24,10 +24,12 @@ import {
   VectorSearch,
 } from './dto';
 import { cacheKeys } from '../cache/config';
+import { LoggingInterceptor } from '../interceptors/index';
 
-//Including 2 kind of cache check getCollections and getStatistics for detail
+//Including 2 kind of cache contorl, check getCollections and getStatistics for detail
 @ApiTags('collections')
 @Controller('collections')
+@UseInterceptors(LoggingInterceptor)
 export class CollectionsController {
   constructor(private collectionsService: CollectionsService, @Inject(CACHE_MANAGER) private cacheManager: Cache) { }
 

server/src/interceptors/index.ts

@@ -7,7 +7,7 @@ import {
   HttpStatus,
   BadRequestException,
 } from '@nestjs/common';
-import { map, catchError } from 'rxjs/operators';
+import { map, catchError, tap } from 'rxjs/operators';
 import { Observable, throwError } from 'rxjs';
 
 export interface Response<T> {
@@ -50,3 +50,18 @@ export class ErrorInterceptor implements NestInterceptor {
     );
   }
 }
+
+/**
+ * add spent time looger when accessing milvus.
+ */
+@Injectable()
+export class LoggingInterceptor implements NestInterceptor {
+  intercept(context: ExecutionContext, next: CallHandler): Observable<any> {
+    const now = Date.now();
+    return next
+      .handle()
+      .pipe(
+        tap(() => Logger.log(`request to ${context.getArgs()[0]['url']} takes ${Date.now() - now}ms`)),
+      );
+  }
+}

server/src/partitions/partitions.controller.ts

@@ -7,6 +7,7 @@ import {
   Query,
   UsePipes,
   ValidationPipe,
+  UseInterceptors,
 } from '@nestjs/common';
 import { ApiTags } from '@nestjs/swagger';
 import {
@@ -16,11 +17,13 @@ import {
   ManageType,
 } from './dto';
 import { PartitionsService } from './partitions.service';
+import { LoggingInterceptor } from '../interceptors/index';
 
 @ApiTags('partitions')
 @Controller('partitions')
+@UseInterceptors(LoggingInterceptor)
 export class PartitionsController {
-  constructor(private partitionsService: PartitionsService) {}
+  constructor(private partitionsService: PartitionsService) { }
 
   @Get()
   @UsePipes(new ValidationPipe())

server/src/schema/schema.controller.ts

@@ -6,6 +6,7 @@ import {
   Query,
   UsePipes,
   ValidationPipe,
+  UseInterceptors,
 } from '@nestjs/common';
 import { ApiTags } from '@nestjs/swagger';
 import {
@@ -16,11 +17,12 @@ import {
   GetIndexState,
 } from './dto';
 import { SchemaService } from './schema.service';
-
+import { LoggingInterceptor } from '../interceptors/index';
 @ApiTags('schema')
 @Controller('schema')
+@UseInterceptors(LoggingInterceptor)
 export class SchemaController {
-  constructor(private schemaService: SchemaService) {}
+  constructor(private schemaService: SchemaService) { }
 
   @Post('index')
   @UsePipes(new ValidationPipe())
@@ -28,10 +30,10 @@ export class SchemaController {
     const { type, collection_name, extra_params, field_name } = body;
     return type === ManageType.CREATE
       ? await this.schemaService.createIndex({
-          collection_name,
-          extra_params,
-          field_name,
-        })
+        collection_name,
+        extra_params,
+        field_name,
+      })
       : await this.schemaService.dropIndex({ collection_name, field_name });
   }