一种高并发流控程序的简单轻量实现

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

public class FlowConcurrentController {
	// 每秒并发访问控制数量
	final static int MAX_QPS = 10;
	// 并发控制信号量
	final static Semaphore semaphore = new Semaphore(MAX_QPS);
	// 监控每秒并发访问次数(理论上accessCount.get() <= 10)
	final static AtomicInteger accessCount = new AtomicInteger(0);

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

	private static void releaseWork() { // 每秒release一次
		// release semaphore thread
		Executors.newScheduledThreadPool(1).scheduleAtFixedRate(new Runnable() {
			@Override
			public void run() {
				semaphore.release(accessCount.get());
				accessCount.set(0);
			}
		}, 1000, 1000, TimeUnit.MILLISECONDS);
	}

	// 模拟并发访问控制
	private static void simulateAccess(final int m, final int n)
			throws Exception { // m : 线程数;n : 调用数
		ExecutorService pool = Executors.newFixedThreadPool(100);
		for (int i = m; i > 0; i--) {
			final int x = i;
			pool.submit(new Runnable() {
				@Override
				public void run() {
					for (int j = n; j > 0; j--) {
						try {
							Thread.sleep(5);
						} catch (InterruptedException e) {
							e.printStackTrace();
						}

						semaphore.acquireUninterruptibly(1);
						accessCount.incrementAndGet();
						remoteCall(x, j);
					}
				}
			});
		}

		pool.shutdown();
		pool.awaitTermination(1, TimeUnit.HOURS);
	}

	public static void main(String[] args) throws Exception {
		// 开启releaseWork
		releaseWork();
		
		// 开始模拟lots of concurrent calls: 100 * 1000
		simulateAccess(100, 1000);
	}

}

 http://blog.youkuaiyun.com/mn11201117/article/details/8809511

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值