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";
}
}
08-22
1059
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)
05-29
6951
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)
11-27
3192
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)
“相关推荐”对你有帮助么?
-
非常没帮助
-
没帮助
-
一般
-
有帮助
-
非常有帮助
提交