Nestjs限制請求次數方式

更新時間：2025年10月31日 09:36:41 作者：coderJason2020

在NestJS中限制請求次數,可以通過創(chuàng)建中間件并使用第三方庫（如express-rate-limit、fastify-rate-limit）來實現,在根模塊中全局應用該中間件,并根據需求調整窗口時間和最大請求次數

創(chuàng)建一個限制請求次數的中間件

創(chuàng)建一個名為 RateLimitMiddleware 的中間件，該中間件用于限制請求次數。

在該中間件中，你可以使用使用第三方庫（如 express-rate-limit、fastify-rate-limit）來實現請求限制邏輯，這些庫提供了方便的方式來設置每個IP地址或其他標識符的請求次數限制。

import { Injectable, NestMiddleware } from '@nestjs/common';
import rateLimit from 'express-rate-limit';

@Injectable()
export class RateLimitMiddleware implements NestMiddleware {
  use(req: any, res: any, next: () => void): any {
    return rateLimit({
      windowMs: 5 * 60 * 1000, // 5分鐘內
      max: 100, // 最多允許100次請求
      message: '請求過于頻繁，請稍后再試。',
    })(req, res, next);
  }
}

在根模塊中使用中間件

在 NestJS 應用程序的根模塊（通常是 app.module.ts）中，在全局范圍內使用 use() 方法來應用該中間件。

import { Module, NestModule, MiddlewareConsumer } from '@nestjs/common';
import { RateLimitMiddleware } from './rate-limit.middleware';
import { AppController } from './app.controller';
import { AppService } from './app.service';

@Module({
  controllers: [AppController],
  providers: [AppService],
})
export class AppModule implements NestModule {
  configure(consumer: MiddlewareConsumer) {
    consumer.apply(RateLimitMiddleware).forRoutes('*');
  }
}

在上述代碼中，RateLimitMiddleware 被應用于所有的路由 ('*')

這樣，請求次數將被限制為每個 IP 地址或其他標識符在指定時間窗口內的最大請求次數。你可以根據需求調整窗口時間和最大請求次數。

請注意，上述示例使用了 express-rate-limit 庫來實現請求限制。如果你使用的是 Fastify 或其他 HTTP 框架，可以選擇相應的庫來實現相似的功能。

此外，你也可以根據業(yè)務需求自定義中間件來實現請求次數的限制邏輯。