如何限制虚拟线程数量以避免大量数据库连接？

核心思路是：虚拟线程的优势在于轻量、可创建大量实例，但数据库连接是稀缺资源，因此我们需要控制并发执行数据库操作的数量，而非限制虚拟线程的总数。以下是几种实用方案：

方法一：用Semaphore限制并发数据库操作 #

通过信号量（Semaphore）控制同时执行数据库操作的虚拟线程数，确保最多只有指定数量的线程占用数据库连接，其余虚拟线程会在等待许可时被挂起，不占用系统资源。

示例代码：

// 定义信号量，允许10个并发数据库操作
Semaphore dbSemaphore = new Semaphore(10);

// 包装原有任务，加入信号量控制
List<Callable<Integer>> wrappedTasks = tasks.stream()
    .map(task -> (Callable<Integer>) () -> {
        dbSemaphore.acquire(); // 获取许可，无可用则等待
        try {
            return task.call(); // 执行包含数据库连接的任务逻辑
        } finally {
            dbSemaphore.release(); // 无论任务成败，都释放许可
        }
    })
    .collect(Collectors.toList());

try(ExecutorService executorService = Executors.newVirtualThreadPerTaskExecutor()) {
    List<Future<Integer>> resultFutures = executorService.invokeAll(wrappedTasks);
    for (Future<Integer> rf : resultFutures) {
        // 处理任务结果
    }
} catch (InterruptedException e) {
    Thread.currentThread().interrupt();
}

方法二：配置数据库连接池的最大连接数 #

直接利用数据库连接池的原生限流能力，将连接池的最大连接数设置为你能承受的上限（比如10）。当所有连接被占用时，后续请求会进入连接池的等待队列，虚拟线程会在等待连接时自动挂起，不会浪费系统资源。

以HikariCP为例，配置示例：

HikariConfig config = new HikariConfig();
config.setJdbcUrl("jdbc:mysql://localhost:3306/your_db");
config.setUsername("db_user");
config.setPassword("db_pass");
config.setMaximumPoolSize(10); // 限制最大数据库连接数为10
HikariDataSource dataSource = new HikariDataSource(config);

后续任务从该连接池获取连接即可，连接池会自动控制并发连接数，无需额外处理虚拟线程。

方法三：组合虚拟线程池与有限任务队列（特殊场景用） #

如果需要限制虚拟线程的创建速率，可以用ThreadPoolExecutor结合虚拟线程工厂，搭配有界任务队列。但这种方式会削弱虚拟线程的高并发优势，仅适合有特殊限制的场景。

示例代码：

// 核心/最大线程数设为10，队列容量100，使用虚拟线程工厂
ExecutorService executorService = new ThreadPoolExecutor(
    10,
    10,
    0L, TimeUnit.MILLISECONDS,
    new ArrayBlockingQueue<>(100),
    Thread.ofVirtual().factory()
);

try {
    List<Future<Integer>> resultFutures = executorService.invokeAll(tasks);
    for (Future<Integer> rf : resultFutures) {
        // 处理任务结果
    }
} catch (InterruptedException e) {
    Thread.currentThread().interrupt();
} finally {
    executorService.shutdown();
}

优先选择方法一或方法二：方法一精准控制数据库操作并发数，方法二利用连接池原生能力，两者都能有效避免数据库连接耗尽，同时保留虚拟线程的高并发优势。方法三仅适合需要限制虚拟线程总数的特殊场景。

内容的提问来源于stack exchange，提问作者Mandroid