From 5a63cc4e1a94855ff9534d3e7cd3825e8dd0a64e Mon Sep 17 00:00:00 2001 From: Trustin Lee Date: Tue, 3 Apr 2012 22:25:01 +0900 Subject: [PATCH] Allow specifying a non-default SelectorProvider --- .../channel/socket/nio/SelectorEventLoop.java | 23 +++++++++++++++---- 1 file changed, 18 insertions(+), 5 deletions(-) diff --git a/transport/src/main/java/io/netty/channel/socket/nio/SelectorEventLoop.java b/transport/src/main/java/io/netty/channel/socket/nio/SelectorEventLoop.java index 6772c4d9ee..4eb97cab09 100644 --- a/transport/src/main/java/io/netty/channel/socket/nio/SelectorEventLoop.java +++ b/transport/src/main/java/io/netty/channel/socket/nio/SelectorEventLoop.java @@ -37,9 +37,11 @@ import java.nio.channels.SelectionKey; import java.nio.channels.Selector; import java.nio.channels.SocketChannel; import java.nio.channels.WritableByteChannel; +import java.nio.channels.spi.SelectorProvider; import java.util.Iterator; import java.util.Queue; import java.util.Set; +import java.util.concurrent.Executors; import java.util.concurrent.ThreadFactory; import java.util.concurrent.atomic.AtomicBoolean; import java.util.concurrent.locks.ReadWriteLock; @@ -87,17 +89,28 @@ abstract class SelectorEventLoop extends SingleThreadEventLoop { protected final SendBufferPool sendBufferPool = new SendBufferPool(); protected SelectorEventLoop() { - selector = openSelector(); + this(Executors.defaultThreadFactory()); } protected SelectorEventLoop(ThreadFactory threadFactory) { - super(threadFactory); - selector = openSelector(); + this(threadFactory, SelectorProvider.provider()); } - private static Selector openSelector() { + protected SelectorEventLoop(SelectorProvider selectorProvider) { + this(Executors.defaultThreadFactory(), selectorProvider); + } + + protected SelectorEventLoop(ThreadFactory threadFactory, SelectorProvider selectorProvider) { + super(threadFactory); + if (selectorProvider == null) { + throw new NullPointerException("selectorProvider"); + } + selector = openSelector(selectorProvider); + } + + private static Selector openSelector(SelectorProvider provider) { try { - return Selector.open(); + return provider.openSelector(); } catch (IOException e) { throw new ChannelException("failed to open a new selector", e); }