快捷導(dǎo)航

Nestjs限制請(qǐng)求次數(shù)方式

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

在NestJS中限制請(qǐng)求次數(shù),可以通過(guò)創(chuàng)建中間件并使用第三方庫(kù)（如express-rate-limit、fastify-rate-limit）來(lái)實(shí)現(xiàn),在根模塊中全局應(yīng)用該中間件,并根據(jù)需求調(diào)整窗口時(shí)間和最大請(qǐng)求次數(shù)

創(chuàng)建一個(gè)限制請(qǐng)求次數(shù)的中間件

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

在該中間件中，你可以使用使用第三方庫(kù)（如 express-rate-limit、fastify-rate-limit）來(lái)實(shí)現(xiàn)請(qǐng)求限制邏輯，這些庫(kù)提供了方便的方式來(lái)設(shè)置每個(gè)IP地址或其他標(biāo)識(shí)符的請(qǐng)求次數(shù)限制。

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分鐘內(nèi)
      max: 100, // 最多允許100次請(qǐng)求
      message: '請(qǐng)求過(guò)于頻繁，請(qǐng)稍后再試。',
    })(req, res, next);
  }
}

在根模塊中使用中間件

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

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 被應(yīng)用于所有的路由 ('*')

這樣，請(qǐng)求次數(shù)將被限制為每個(gè) IP 地址或其他標(biāo)識(shí)符在指定時(shí)間窗口內(nèi)的最大請(qǐng)求次數(shù)。你可以根據(jù)需求調(diào)整窗口時(shí)間和最大請(qǐng)求次數(shù)。

請(qǐng)注意，上述示例使用了 express-rate-limit 庫(kù)來(lái)實(shí)現(xiàn)請(qǐng)求限制。如果你使用的是 Fastify 或其他 HTTP 框架，可以選擇相應(yīng)的庫(kù)來(lái)實(shí)現(xiàn)相似的功能。

此外，你也可以根據(jù)業(yè)務(wù)需求自定義中間件來(lái)實(shí)現(xiàn)請(qǐng)求次數(shù)的限制邏輯。