src/java.base/linux/classes/sun/nio/ch/DefaultPollerProvider.java

@@ -29,36 +29,79 @@
  
  /**
   * Default PollerProvider for Linux.
   */
  class DefaultPollerProvider extends PollerProvider {
-     DefaultPollerProvider() { }
+     private static final boolean USE_IOURING;
+     private static final boolean USE_IORING_OP_READ;
+     private static final boolean USE_IORING_OP_WRITE;
+     static {
+         String s = System.getProperty("jdk.io_uring");
+         if ("".equals(s) || Boolean.parseBoolean(s)) {
+             USE_IOURING = true;
+             s = System.getProperty("jdk.io_uring.read");
+             USE_IORING_OP_READ = "".equals(s) || Boolean.parseBoolean(s);
+             s = System.getProperty("jdk.io_uring.write");
+             USE_IORING_OP_WRITE = "".equals(s) || Boolean.parseBoolean(s);
+         } else {
+             USE_IOURING = false;
+             USE_IORING_OP_READ = false;
+             USE_IORING_OP_WRITE = false;
+         }
+     }
+ 
+     DefaultPollerProvider(Poller.Mode mode) {
+         super(mode);
+     }
+ 
+     DefaultPollerProvider() {
+         this(ContinuationSupport.isSupported()
+                 ? Poller.Mode.VTHREAD_POLLERS
+                 : Poller.Mode.SYSTEM_THREADS);
+     }
+ 
+     @Override
+     int defaultReadPollers() {
+         int ncpus = Runtime.getRuntime().availableProcessors();
+         return switch (pollerMode()) {
+             case SYSTEM_THREADS  -> Math.max(Integer.highestOneBit(ncpus / 4), 1);
+             case VTHREAD_POLLERS -> Math.min(Integer.highestOneBit(ncpus), 32);
+             default              -> super.defaultReadPollers();
+         };
+     }
  
      @Override
-     Poller.Mode defaultPollerMode() {
-         if (ContinuationSupport.isSupported()) {
-             return Poller.Mode.VTHREAD_POLLERS;
+     Poller readPoller(boolean subPoller) throws IOException {
+         Poller.Mode mode = pollerMode();
+         if (USE_IOURING) {
+             // read poller is system thread in SYSTEM_THREADS mode
+             boolean supportReadOps = USE_IORING_OP_READ
+                     && ((mode == Poller.Mode.SYSTEM_THREADS) || subPoller);
+             return new IoUringPoller(mode, subPoller, true, supportReadOps);
          } else {
-             return Poller.Mode.SYSTEM_THREADS;
+             return new EPollPoller(mode, subPoller, true);
          }
      }
  
      @Override
-     int defaultReadPollers(Poller.Mode mode) {
-         int ncpus = Runtime.getRuntime().availableProcessors();
-         if (mode == Poller.Mode.VTHREAD_POLLERS) {
-             return Math.min(Integer.highestOneBit(ncpus), 32);
+     Poller writePoller(boolean subPoller) throws IOException {
+         Poller.Mode mode = pollerMode();
+         if (USE_IOURING) {
+             // write poller is system thread in SYSTEM_THREADS and POLLER_PER_CARRIER modes
+             boolean supportWriteOps = USE_IORING_OP_WRITE
+                     && ((mode != Poller.Mode.VTHREAD_POLLERS) || subPoller);
+             return new IoUringPoller(mode, subPoller, false, supportWriteOps);
          } else {
-             return Math.max(Integer.highestOneBit(ncpus / 4), 1);
+             return new EPollPoller(mode, subPoller, false);
          }
      }
  
      @Override
-     Poller readPoller(boolean subPoller) throws IOException {
-         return new EPollPoller(subPoller, true);
+     boolean supportReadOps() {
+         return USE_IORING_OP_READ;
      }
  
      @Override
-     Poller writePoller(boolean subPoller) throws IOException {
-         return new EPollPoller(subPoller, false);
+     boolean supportWriteOps() {
+         return USE_IORING_OP_WRITE;
      }
  }