Skip to content

Commit a8519a7

Browse files
committed
Fix benchmark
1 parent 539fae3 commit a8519a7

1 file changed

Lines changed: 25 additions & 5 deletions

File tree

benchmarks/src/main/java/io/netty/loom/benchmark/PollerBenchmark.java

Lines changed: 25 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -133,11 +133,31 @@ public void init() throws IOException, InterruptedException, BrokenBarrierExcept
133133
blockingReadTasks.add(task);
134134
LockSupport.unpark(carrierParked);
135135
};
136-
readThreadFactory = Thread.ofVirtual().scheduler(Thread.VirtualThreadScheduler.adapt(readScheduler)).factory();
137-
writeThreadFactory = Thread.ofVirtual().scheduler(Thread.VirtualThreadScheduler.adapt(task -> {
138-
blockingWriteTasks.add(task);
139-
LockSupport.unpark(carrierParked);
140-
})).factory();
136+
readThreadFactory = Thread.ofVirtual().scheduler(new Thread.VirtualThreadScheduler() {
137+
@Override
138+
public void onStart(Thread.VirtualThreadTask virtualThreadTask) {
139+
readScheduler.execute(virtualThreadTask);
140+
}
141+
142+
@Override
143+
public void onContinue(Thread.VirtualThreadTask virtualThreadTask) {
144+
readScheduler.execute(virtualThreadTask);
145+
}
146+
}).factory();
147+
writeThreadFactory = Thread.ofVirtual().scheduler(new Thread.VirtualThreadScheduler() {
148+
149+
@Override
150+
public void onStart(Thread.VirtualThreadTask virtualThreadTask) {
151+
blockingWriteTasks.add(virtualThreadTask);
152+
LockSupport.unpark(carrierParked);
153+
}
154+
155+
@Override
156+
public void onContinue(Thread.VirtualThreadTask virtualThreadTask) {
157+
blockingWriteTasks.add(virtualThreadTask);
158+
LockSupport.unpark(carrierParked);
159+
}
160+
}).factory();
141161
clientIn = clientSocket.getInputStream();
142162
serverOut = serverSocket.getOutputStream();
143163
clientSocket.setTcpNoDelay(true);

0 commit comments

Comments
 (0)