NestJS学习笔记之中间件

在NestJS中，中间件是一种组织应用程序逻辑的方式，它可以拦截进入的请求和传出的响应。

1. 创建一个中间件

在NestJS中，你可以通过定义一个类并使用@Injectable()装饰器来创建一个中间件，并实现NestJS提供的NestMiddleware接口。

import { Injectable, NestMiddleware } from '@nestjs/common';
import { Request, Response, NextFunction } from 'express';
 
@Injectable()
export class MyMiddleware implements NestMiddleware {
  use(req: Request, res: Response, next: NextFunction): void {
    console.log('Request...');
    next();
    console.log('Response...');
  }
}

2. 应用中间件

你可以在模块的根级别或者在特定的控制器或者路由级别应用中间件。

import { Module, NestModule, MiddlewareConsumer } from '@nestjs/common';
import { MyMiddleware } from './my.middleware';
 
@Module({})
export class ApplicationModule implements NestModule {
  configure(consumer: MiddlewareConsumer) {
    consumer
      .apply(MyMiddleware)
      .forRoutes('*'); // 应用于所有路由
  }
}

在上述代码中，MyMiddleware将会应用于所有的路由。你也可以通过指定路由的路径或者特定的控制器来限制中间件的应用范围。

3. 中间件链

NestJS支持中间件链，你可以通过调用consumer的.apply()方法多次来构建一个中间件链。

configure(consumer: MiddlewareConsumer) {
  consumer
    .apply(Middleware1, Middleware2)
    .forRoutes('*');
}

4. 使用第三方中间件

NestJS提供了一个方便的方式来使用第三方中间件。你只需要将第三方中间件包装成NestJS的中间件格式即可。

import { NestFactory } from '@nestjs/core';
import { AppModule } from './app.module';
import * as expressSession from 'express-session';
 
async function bootstrap() {
  const app = await NestFactory.create(AppModule);
 
  app.use(expressSession({
    secret: 'my-secret',
    resave: false,
    saveUninitialized: true,
  }));
 
  await app.listen(3000);
}
bootstrap();

在上述代码中，我们使用了express-session库，它是一个非常流行的session中间件，在NestJS应用中，我们通过app.use()方法将其应用到了我们的应用中。

以上就是NestJS中间件的基本使用方法。