分享一个话题：如何限制线程池中一秒钟只运行十个线程？

最新推荐文章于 2023-06-04 21:48:08 发布

原创最新推荐文章于 2023-06-04 21:48:08 发布 · 2.1k 阅读

2 ·

CC 4.0 BY-SA版权

并发编程专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种使用Semaphore和ScheduledThreadPool实现每秒限流至十个线程的方法，并提供了完整的Java示例代码。

如何限制线程池中一秒钟只运行十个线程，参考代码如下：

package cn.starty.test;

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

public class TestOne {

	final static int MAX_QPS = 10;

	final static Semaphore semaphore = new Semaphore(MAX_QPS);

	public static void main(String[] args) throws InterruptedException {
		Executors.newScheduledThreadPool(1).scheduleAtFixedRate(new Runnable() {

			@Override
			public void run() {
				semaphore.release(MAX_QPS / 2);
			}

		}, 1000, 500, TimeUnit.MILLISECONDS);

		ExecutorService pool = Executors.newFixedThreadPool(100);
		for (int i = 100; i > 0; i--) {
			final int x = i;
			pool.submit(new Runnable() {

				@Override
				public void run() {
					for (int j = 1000; j > 0; j--) {
						semaphore.acquireUninterruptibly(1);
						remoteCall(x, j);
					}
				}

			});
		}
		pool.shutdown();
		pool.awaitTermination(1, TimeUnit.HOURS);
		System.out.println("DONE");
	}

	private static void remoteCall(int i, int j) {
		System.out.println(String.format("%s - %s: %d %d", new Date(),
				Thread.currentThread(), i, j));
	}

}