58 lines
1.4 KiB
Markdown
58 lines
1.4 KiB
Markdown
|
# 每日限流中间件使用指南
|
|||
|
|
|
|||
|
|
## 概述
|
|||
|
|
|
|||
|
|
每日限流中间件实现了多层限流策略：
|
|||
|
|
- 接口一天最大请求200次
|
|||
|
|
- 一个IP一天最多10次
|
|||
|
|
- 支持并发限制和安全防护
|
|||
|
|
|
|||
|
|
## 主要特性
|
|||
|
|
|
|||
|
|
1. **多层限流策略**
|
|||
|
|
- 接口总请求限制：200次/天
|
|||
|
|
- IP请求限制：10次/天/IP
|
|||
|
|
- 并发请求限制：5个/IP
|
|||
|
|
|
|||
|
|
2. **安全防护**
|
|||
|
|
- IP白名单/黑名单
|
|||
|
|
- User-Agent检查
|
|||
|
|
- Referer验证
|
|||
|
|
- 代理检测
|
|||
|
|
|
|||
|
|
## 使用方法
|
|||
|
|
|
|||
|
|
```go
|
|||
|
|
// 配置限流参数
|
|||
|
|
limitConfig := middleware.DailyRateLimitConfig{
|
|||
|
|
MaxRequestsPerDay: 200, // 接口一天最大请求200次
|
|||
|
|
MaxRequestsPerIP: 10, // 一个IP一天最多10次
|
|||
|
|
KeyPrefix: "api_limit", // Redis键前缀
|
|||
|
|
TTL: 24 * time.Hour, // 24小时过期
|
|||
|
|
MaxConcurrent: 5, // 最大并发5个
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 创建中间件实例
|
|||
|
|
rateLimitMiddleware := middleware.NewDailyRateLimitMiddleware(
|
|||
|
|
config, redisClient, response, logger, limitConfig)
|
|||
|
|
|
|||
|
|
// 应用到路由
|
|||
|
|
router.Use(rateLimitMiddleware.Handle())
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
## 限流逻辑
|
|||
|
|
|
|||
|
|
1. 检查IP访问权限
|
|||
|
|
2. 验证User-Agent和Referer
|
|||
|
|
3. 检查并发限制
|
|||
|
|
4. 检查接口总请求次数（200次/天）
|
|||
|
|
5. 检查IP请求次数（10次/天）
|
|||
|
|
6. 更新计数器
|
|||
|
|
|
|||
|
|
## 监控信息
|
|||
|
|
|
|||
|
|
响应头包含隐藏的监控信息：
|
|||
|
|
- `X-Total-Count`: 当前总请求次数
|
|||
|
|
- `X-IP-Count`: 当前IP请求次数
|
|||
|
|
- `X-Reset-Time`: 重置时间
|