<dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <dependency>
            <groupId>org.redisson</groupId>
            <artifactId>redisson-spring-boot-starter</artifactId>
            <version>3.48.0</version>
        </dependency>

        <dependency>
            <groupId>com.google.guava</groupId>
            <artifactId>guava</artifactId>
            <version>33.4.8-jre</version>
        </dependency>
    </dependencies>

spring:
  data:
    redis:
      host: redis
      port: 6379

@SpringBootApplication
public class RateLimitApplication {

    public static void main(String[] args) {
        SpringApplication.run(RateLimitApplication.class, args);
    }
}

@Slf4j
@RestController
@RequestMapping(value = "redisson/rate-limit")
public class RedissonRateLimitController {

    @Autowired
    private RedissonClient redissonClient;

    @PostMapping("create-order")
    public Object createOrder(String info){
        RRateLimiter rateLimiter = getLimiter();
        boolean b = rateLimiter.tryAcquire();
        if(b) {
            log.info("允許訪問(wèn)");
            return "SUCCESS";
        }
        log.warn("限流了");
        throw new RuntimeException("請(qǐng)求太火爆");
    }

    /**
     * 獲取限流器
     * @return
     */
    private RRateLimiter getLimiter() {
        RRateLimiter rateLimiter = redissonClient.getRateLimiter("create-order");
        rateLimiter.trySetRate(RateType.PER_CLIENT, 20, Duration.ofSeconds(5L));
        return rateLimiter;
    }

}

@Slf4j
@RestController
@RequestMapping(value = "guava/rate-limit")
public class GuavaRateLimitController {

    private static final RateLimiter rateLimiter = RateLimiter.create(5);

    @PostMapping("create-order")
    public Object createOrder(String info){
        boolean b = rateLimiter.tryAcquire();
        if(b) {
            log.info("允許訪問(wèn)");
            return "SUCCESS";
        }
        log.warn("限流了");
        throw new RuntimeException("請(qǐng)求太火爆");
    }


}

http {
    # 定義限流共享內(nèi)存區(qū)（名稱=one，大小=10MB，速率=10請(qǐng)求/秒）
    limit_req_zone $binary_remote_addr zone=one:10m rate=10r/s;

    # 其他全局配置...
    include       mime.types;
    default_type  application/octet-stream;
}

server {
    listen       80;
    server_name  example.com;

    location /api/ {
        # 應(yīng)用限流規(guī)則（使用上面定義的 zone=one）
        limit_req zone=one burst=20 nodelay;  # burst=突發(fā)容量，nodelay=不等待直接拒絕
        
        # 被限流時(shí)返回 429 狀態(tài)碼，并跳轉(zhuǎn)到自定義錯(cuò)誤頁(yè)
        limit_req_status_code 429;
        error_page 429 /custom_limit_msg;

        # 正常請(qǐng)求的反向代理（示例）
        proxy_pass http://backend_server;
    }

    # 自定義限流消息體（返回 JSON）
    location = /custom_limit_msg {
        internal;  # 僅內(nèi)部訪問(wèn)，禁止外部直接請(qǐng)求
        add_header Content-Type application/json;
        return 429 '{"code": 429, "message": "請(qǐng)求過(guò)于頻繁，請(qǐng)稍后再試"}';
    }
}

http {
    # 限流鍵為 X-User-ID 頭的值（需確保頭存在）
    limit_req_zone $http_x_user_id zone=user_limit:10m rate=5r/s;

    server {
        location /api/user/ {
            # 校驗(yàn) X-User-ID 頭是否存在，不存在則拒絕
            if ($http_x_user_id = "") {
                return 400 "Missing X-User-ID header";
            }
            
            # 應(yīng)用限流（鍵為用戶 ID）
            limit_req zone=user_limit burst=10 nodelay;
            limit_req_status_code 429;
            error_page 429 /custom_limit_msg;
            
            proxy_pass http://backend_server;
        }
    }
}