Java21创建VirtualThread虚拟线程的几种方式以及结构化并发(预览功能)

import org.junit.jupiter.api.Test;

import java.time.Duration;
import java.util.concurrent.Executors;
import java.util.concurrent.StructuredTaskScope;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.TimeUnit;
import java.util.stream.IntStream;

/**
 * JDK21虚拟线程测试类
 * 创建日期:2024-05-09
 */
public class VirtualThreadTests {

    /**
     * 方式一:创建一个执行器,为每个任务启动一个新的虚拟线程
     */
    @Test
    void testVirtualThread1() {
        try (var executor = Executors.newVirtualThreadPerTaskExecutor()) {
            IntStream.range(0, 10_000).forEach(i -> executor.submit(() -> {
                Thread.sleep(Duration.ofSeconds(1));
                System.out.println(STR."task: \{i}");
                return i;
            }));
        }
    }

    /**
     * 方式二:
     * 1、创建虚拟线程并自动执行
     * 2、创建不自动启动的线程,手动执行
     */
    @Test
    void testVirtualThread2() {
        // 创建虚拟线程并自动执行
        Thread.Builder.OfVirtual virtual = Thread.ofVirtual().name("pack1");
        virtual.start(() -> System.out.printf("%s - VirtualThread1自动执行任务完成\n", Thread.currentThread().getName()));
        // 创建不自动启动的线程
        Thread thread = virtual.unstarted(() -> System.out.printf("%s - VirtualThread1手动执行任务完成\n", Thread.currentThread().getName()));
        // 手动启动虚拟线程
        thread.start();
        // 打印线程对象:VirtualThread[#21,pack]/runnable
        System.out.println(STR."VirtualThread1: \{thread}");
    }

    /**
     * 方式三:通过 ThreadFactory 工厂创建
     */
    @Test
    void testVirtualThread3() {
        ThreadFactory threadFactory = Thread.ofVirtual().name("threadFactory").factory();
        threadFactory.newThread(() -> System.out.printf("%s - 通过工厂创建执行任务完成", Thread.currentThread().getName())).start();
    }

    /**
     * 方式四:直接通过Thread静态方法创建
     * 该方式无法指定线程名称,线程名称默认为:null
     */
    @Test
    void testVirtualThread4() {
        Thread.startVirtualThread(() -> System.out.printf("线程ID:%s - 任务执行完成", Thread.currentThread().threadId())) ;
    }

    /**
     * 结构化并发(预览功能)
     * 当一个子任务发生错误时,其它的子任务会在未完成的情况下取消
     */
    @Test
    void testVirtualThread5() throws InterruptedException {
        try (var scope = new StructuredTaskScope.ShutdownOnFailure()) {
            StructuredTaskScope.Subtask<String> fork1 = scope.fork(VirtualThreadTests::task1);
            StructuredTaskScope.Subtask<String> fork2 = scope.fork(VirtualThreadTests::task2);
            StructuredTaskScope.Subtask<String> fork3 = scope.fork(VirtualThreadTests::task3);
            //正常执行完的顺序:task3 ->task2 ->task1,异常执行完的顺序:task3(task2、task1停止)
            scope.join();
            System.out.println(fork1);
            System.out.println(fork2);
            System.out.println(fork3);
        }
    }

    private static String task1() throws InterruptedException {
        TimeUnit.SECONDS.sleep(3) ;
        return "task1 ok";
    }

    private static String task2() throws InterruptedException {
        TimeUnit.SECONDS.sleep(2) ;
        //抛出异常并让其它未完成的子任务停止。
        //System.out.println(1 / 0) ;
        return "task2 ok";
    }

    private static String task3() throws InterruptedException {
        TimeUnit.SECONDS.sleep(1) ;
        return "task3 ok";
    }

}
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值