java 控制qps Java 控制并发数量

转载

墨色天香 2023-08-16 17:37:28

文章标签 java 控制qps java 开发语言后端信号量 文章分类 Java 后端开发

Semaphore （信号量）是用来控制同时访问特定资源的线程数量，它通过协调各个线程，以

保证合理的使用公共资源。

计数信号灯。从概念上讲，信号量维护一组许可。每个{@link#acquire}都会在必要时阻塞，直到有许可证可用，然后再获取它。每个{@link#release}都会添加一个许可证，可能会释放一个阻塞收单机构。但是，没有使用实际的许可证对象；{@code信号量}只保留可用数量的计数，并相应地进行操作。

应用场景：

Semaphore 可以用于做流量控制，特别是公用资源有限的应用场景，比如数据库连接。假

如有一个需求，要读取几万个文件的数据，因为都是 IO 密集型任务，我们可以启动几十个线程

并发地读取，但是如果读到内存后，还需要存储到数据库中，而数据库的连接数只有 10 个，这

时我们必须控制只有 10 个线程同时获取数据库连接保存数据，否则会报错无法获取数据库连

接。这个时候，就可以使用 Semaphore 来做流量控制

package com.netty.obj.demo5.d4;

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

/**
 * @PackageName:com.netty.obj.demo5.d4 Description
 * @author:
 * @date:2022/1/12
 */
public class SemaphoreTest {
    private static final int THREAD_COUNT = 30;
    private static ExecutorService threadPool =Executors.newFixedThreadPool(THREAD_COUNT);
    private static Semaphore s = new Semaphore(10);

    public static void main(String[] args) {
        for (int i = 0; i < THREAD_COUNT; i++) {
            threadPool.execute(new Runnable() {
                @Override
                public void run() {
                    try {
                        s.acquire();
                        System.out.println("save data");
                        s.release();
                    } catch (InterruptedException e) {
                    }
                }
            });
        }
        threadPool.shutdown();
    }
}

在代码中，虽然有 30 个线程在执行，但是只允许 10 个并发执行。 Semaphore 的构造方法

Semaphore （ int permits ）接受一个整型的数字，表示可用的许可证数量。 Semaphore （ 10 ）表示允

许 10 个线程获取许可证，也就是最大并发数是 10 。 Semaphore 的用法也很简单，首先线程使用

Semaphore 的 acquire() 方法获取一个许可证，使用完之后调用 release() 方法归还许可证。还可以

用 tryAcquire() 方法尝试获取许可证。

Semaphore 还提供一些其他方法，具体如下。

·intavailablePermits() ：返回此信号量中当前可用的许可证数。

·intgetQueueLength() ：返回正在等待获取许可证的线程数。

·booleanhasQueuedThreads() ：是否有线程正在等待获取许可证。

·void reducePermits （ int reduction ）：减少 reduction 个许可证，是个 protected 方法。

·Collection getQueuedThreads() ：返回所有等待获取许可证的线程集合，是个 protected 方

法。