2023-08-29 08:26:53 來源 : 博客園
用過express與koa的同學，對中間件這個概念應該非常熟悉了，中間件可以拿到Request、Response對象和next函數(shù).
一般來講中間件有以下作用：
(資料圖片)
request-response周期通過next()調(diào)用下一個中間件如果當前中間件沒有結束當前request-response周期，必須調(diào)用next()函數(shù)，否則請求會處于掛起狀態(tài)，阻塞整個應用中間件一般有兩種：類中間件、函數(shù)中間件
類中間件創(chuàng)建類中間件使用@Injectable()裝飾器，并且需要實現(xiàn)NestMiddleware接口（use方法）
// Logger.middleware.tsimport { Injectable, NestMiddleware } from "@nestjs/common";import { Request, Response } from "express";@Injectable()export class LoggerMiddleware implements NestMiddleware { use(req: Request, res: Response, next: () => void) { console.log("logger middleware", `url: ${req.url}`); next(); }}類中間創(chuàng)建完之后，需要在模塊中進行掛載，但@Module裝飾器并沒有中間件的相關配置，我們需要讓module類實現(xiàn)NestModule接口，實現(xiàn)里面configure方法來進行掛載
// user.module.tsimport { Module, NestModule } from "@nestjs/common";import { UserService } from "./user.service";import { UserController } from "./user.controller";import { LoggerMiddleware } from "../middleware/Logger.middleware";@Module({ controllers: [UserController], providers: [UserService]})export class UserModule implements NestModule { configure(consumer) { consumer .apply(LoggerMiddleware) .forRoutes(UserController); }}apply方法表示掛載的是哪個中間件forRoutes方法表示對哪個請求路徑起作用，這種方式與app.use(path, middleware)作用是一樣，只針對部分路徑起作用當給forRoutes方法傳遞的是一個controller控制器時，那么該中間件則對整個控制器下的路徑生效比如這里傳遞的是UserController控制器，那么針對該控制器下的路徑都會生效
forRootes方法還能做更詳細的配置，比如可以針對特定的請求方法、請求路徑可以使用正則匹配（需要注意的是使用fastify驅動不能使用）export class UserModule implements NestModule { configure(consumer) { consumer .apply(LoggerMiddleware) .forRoutes({ path: "user", method: RequestMethod.GET}); }}apply可以同時掛載多個中間件export class UserModule implements NestModule { configure(consumer) { consumer .apply(LoggerMiddleware, aaaMiddleware, ...) .forRoutes({ path: "user", method: RequestMethod.GET}); }}forRoutes可以使用單個string路徑，多個string路徑，RouteInfo對象，單個Controller，多個Controllerexport class AppModule implements NestModule { configure(consumer) { consumer .apply(LoggerMiddleware, NjMiddleware, ...) .forRoutes(UserController, NjController, ...); }}exclude可以用來排除不使用中間件的路徑export class UserModule implements NestModule { configure(consumer) { consumer .apply(LoggerMiddleware) .exclude({ path: "/user/a", method: RequestMethod.GET}) .forRoutes(UserController); }}需要注意的是forRoutes需要最后調(diào)用
這種方式較為簡單，使用起來與類中間件一致
創(chuàng)建函數(shù)中間件export function LoggerMiddleware(req: Request, res: Response, next: () => void) { console.log("logger middleware", `url: ${req.url}`); next();}export class UserModule implements NestModule { configure(consumer) { consumer .apply(LoggerMiddleware) .exclude({ path: "/user/a", method: RequestMethod.GET}) .forRoutes(UserController); }}可以直接在入口文件main.ts中使用app.use來掛載中間件，這樣掛載的中間件將全局生效
app.use(LoggerMiddleware) // 日志中間件中間件其實可以用來實現(xiàn)很多功能，比如：日志系統(tǒng)、cors跨域處理、圖片防盜等...
對圖片防盜感興趣的可以看我這篇文章：你不知道的 HTTP Referer
