[Nestjs] 限制请求次数

在NestJS中限制请求次数，可以使用中间件来实现。以下是一种常见的方式：

1.创建一个限制请求次数的中间件： 创建一个名为 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);
  }
}

2.在根模块中使用中间件： 在 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 框架，可以选择相应的库来实现相似的功能。

此外，你也可以根据业务需求自定义中间件来实现请求次数的限制逻辑。