Java并发编程-信号量

Semaphore 直译是信号量,它的功能比较好理解,就是通过构造函数设定一个数量的许可,然后通过 acquire 方法获得许可,release 方法释放许可。它还有 tryAcquire 和 acquireUninterruptibly 方法,可以根据自己的需要选择。
以下是模拟一个连接池,控制同一时间最多只能有50个线程访问,主线程每毫秒创建一个子线程,连接池50个许可,每个获取连接的子线程持球300ms,连接等待超时500ms。

import java.util.UUID;
import java.util.concurrent.Semaphore;
import java.util.concurrent.TimeUnit;

public class TestSemaphore extends Thread {
    public static void main(String[] args) {
        int i = 0;
        while (i < 500) {
            i++;
            new TestSemaphore().start();
            try {
                Thread.sleep(1);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }

    /**
     * 控制某资源同时被访问的个数的类 控制同一时间最后只能有50个访问
     */
    static Semaphore semaphore = new Semaphore(50);
    static int timeout = 500;

    public void run() {
        try {
            Object connec = getConnection();
            System.out.println("获得一个连接" + connec);
            Thread.sleep(300);
            releaseConnection(connec);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }

    public void releaseConnection(Object connec) {
        /* 释放许可 */
        semaphore.release();
        System.out.println("释放一个连接" + connec);
    }

    public Object getConnection() {
        try {/* 获取许可 */
            boolean getAccquire = semaphore.tryAcquire(timeout, TimeUnit.MILLISECONDS);
            if (getAccquire) {
                return UUID.randomUUID().toString();
            }
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        throw new IllegalArgumentException("timeout");
    }
}
posted @ 2015-11-08 23:51  素轩  阅读(274)  评论(0编辑  收藏  举报