怎样在java主方法中实现限流

在Java主方法中实现限流，可以使用信号量（Semaphore）或者令牌桶（Token Bucket）算法。这里我将给出一个使用信号量实现的简单示例：

import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Semaphore;

public class RateLimiter {
    public static void main(String[] args) {
        // 定义限流速率，例如每秒10个请求
        int permitsPerSecond = 10;
        // 创建一个信号量，初始许可数为限流速率
        Semaphore semaphore = new Semaphore(permitsPerSecond);

        // 创建一个固定大小的线程池
        ExecutorService executorService = Executors.newFixedThreadPool(10);

        for (int i = 0; i < 100; i++) {
            final int requestId = i;
            executorService.submit(() -> {
                try {
                    // 获取许可
                    semaphore.acquire();
                    System.out.println("请求" + requestId + "被处理");
                    // 模拟处理请求的时间
                    Thread.sleep(100);
                } catch (InterruptedException e) {
                    System.out.println("请求" + requestId + "被中断");
                } finally {
                    // 释放许可
                    semaphore.release();
                }
            });
        }

        // 关闭线程池
        executorService.shutdown();
    }
}

在这个示例中，我们创建了一个信号量semaphore，并设置初始许可数为限流速率（每秒10个请求）。然后，我们创建了一个固定大小的线程池executorService，用于执行请求处理任务。

对于每个请求，我们首先尝试获取信号量许可。如果成功获取到许可，说明当前请求可以被处理，我们将请求ID打印出来，并模拟处理请求的时间（这里使用Thread.sleep()方法）。处理完成后，我们在finally块中释放信号量许可，以便其他请求可以使用。

最后，我们关闭线程池以停止接收新的请求。