如何控制java虚拟线程的并发度？

jdk 21中的虚拟线程已经推出好一段时间了，确实很轻量，先来一段示例：

假如有一段提交订单的业务代码：

    public void submitOrder(Integer orderId) {
        sleep(1000);
        System.out.println("order:" + orderId + " is submitted");
    }

这里我们用sleep来模拟，每提交1个订单，大致耗时1秒。

如果有10个订单一起下单，顺序执行的话，至少10秒，很容易想到用多线程：

    @Test
    public void submitOrder1() {
        ThreadFactory threadFactory = new ThreadFactoryBuilder().setNameFormat("submitOrder-pool-%d").build();
        ExecutorService submitOrderPool = new ThreadPoolExecutor(4, 16, 300L, TimeUnit.SECONDS,
                new LinkedBlockingQueue<>(128), threadFactory, new ThreadPoolExecutor.DiscardOldestPolicy());
        CompletableFuture<Void>[] futures = new CompletableFuture[10];
        StopWatch watch = new StopWatch();
        watch.start();
        //下10个订单
        for (int i = 0; i < 10; i++) {
            final int orderId = i;
            CompletableFuture<Void> future = CompletableFuture.runAsync(() -> submitOrder(orderId), submitOrderPool);
            futures[i] = future;
        }
        CompletableFuture.allOf(futures).join();
        watch.stop();
        System.out.println("Time: " + watch.getTime());
    }

这里我们使用了传统的线程池，核心线程数是4，小于10，所以会内部排队，最终执行效果大致如下：

order:0 is submitted
order:3 is submitted
order:1 is submitted
order:2 is submitted
order:4 is submitted
order:6 is submitted
order:7 is submitted
order:5 is submitted
order:8 is submitted
order:9 is submitted
Time: 3035

明显快多了，但是占用了4个传统的java线程。

 

我们改用虚拟线程来改写：

    @Test
    public void submitOrder2() {
        CompletableFuture<Void>[] futures = new CompletableFuture[10];
        StopWatch watch = new StopWatch();
        watch.start();
        //下10个订单(这里使用虚拟线程)
        try (ExecutorService service = Executors.newVirtualThreadPerTaskExecutor()) {
            for (int i = 0; i < 10; i++) {
                final int orderId = i;
                CompletableFuture<Void> future = CompletableFuture.runAsync(() -> submitOrder(orderId), service);
                futures[i] = future;
            }
            CompletableFuture.allOf(futures).join();
        }
        watch.stop();
        System.out.println("Time: " + watch.getTime());
    }

 

参考文档：
https://blog.moyucoding.com/jvm/2023/09/23/ultimate-guide-to-java-virtual-thread

https://docs.oracle.com/en/java/javase/21/core/virtual-threads.html#GUID-DC4306FC-D6C1-4BCC-AECE-48C32C1A8DAA