控制并发线程数的Semaphore

随声附和 提交于 2019-12-02 08:50:36

Semaphore被用于控制特定资源在同一个时间被访问的线程数量,它通过协调各个线程,以保证资源可以被合理的使用。

做个比喻,把Semaphore比作是控制流量的红绿灯,比如xx马路要现在流量,只允许同时有一百辆车在马路上行驶,其他的都必须在路口等待,所以前一百辆会看到绿灯,可以开进马路,后面的车会看到红灯,不能开进马路,但是如果前面一百辆车中有5辆已经离开了马路,那后面就允许有5辆车驶入马路,这里例子里说的车就是线程,驶入马路就代表线程正在执行,离开马路就表示线程执行完成,看到红灯就代表线程被阻塞,不能执行。

应用场景

Semaph可以用来做流量限制,特别是公共资源有限的应用场景,比如说数据库连接。

假如有一个需求,要读取几万个文件的数据,因为都是IO密集型人物,我们可以启动几十个线程并发的读取,但是如果读取到内存后,还需要储存到数据库,而数据库的连接数只有10个,这时候我们就必须控制只有10个线程同时获取到数据库连接,否则会抛出异常提示无法连接数据库。针对这种情况,我们就可以使用Semaphore来做流量控制。代码如下:

package com.dreyer.javadoc.thread;



import java.util.concurrent.*;



/**

 * @description

 * @author: Dreyer

 * @date: 16/5/15 上午11:59

 */

public class SemaphoreDemo {

    /**

     * 线程数量

     */

    private static final int THREAD_COUNT = 30;



    /**

     * 线程池

     */

    private static ExecutorService executor = Executors.newFixedThreadPool(THREAD_COUNT);



    private static Semaphore semaphore = new Semaphore(10);





    public static void main(String[] args) {

        for (int i = 0; i < THREAD_COUNT; i++) {

            executor.execute(new Runnable() {

                public void run() {

                    try {

                        // 获取一个"许可证"

                        semaphore.acquire();



                        // 模拟数据保存

                        TimeUnit.SECONDS.sleep(2);

                        System.out.println("save date...");



                        // 执行完后,归还"许可证"

                        semaphore.release();

                    } catch (InterruptedException e) {

                        e.printStackTrace();

                    }

                }

            });

        }

        executor.shutdown();

    }

}

在代码中,虽然有30个线程在执行,但是只运行10个并发的执行。所以我们可以看到在执行的过程中

save data...是每10输出的。

Semaphore的构造方法Semaphore(int permits)接受一个整形的数字,表示可用的许可证数量。

Semaphore(10)表示运行10个线程获取许可证,也就是最大的并发数是10。

Semaphore的用法也很简单,首先使用Semaphore.acquire()方法获取一个许可证,使用完之后调用

release()方法归还许可证

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!