Support the usage of SocketOption when nio is used and the java versi… (#8085)
* Support the usage of SocketOption when nio is used and the java version >= 7. Motivation: The JDK uses SocketOption since java7 to support configuration options on the underyling Channel. We should allow to create a ChannelOption from a given SocketOption if nio is used. This also allows us to expose the same featureset in terms of configuration as the java nio implementation does without any extra effort. Modifications: - Add NioChannelOption which allows to wrap an existing SocketOption which then can be applied to the nio transport. - Add test-cases Result: Support the same configuration options as the JDK. Also fixes https://github.com/netty/netty/issues/8072.
This commit is contained in:
parent
77ec839792
commit
952eeb8e1e
2
pom.xml
2
pom.xml
@ -702,6 +702,8 @@
|
||||
<ignore>java.nio.channels.SocketChannel</ignore>
|
||||
<ignore>java.net.StandardProtocolFamily</ignore>
|
||||
<ignore>java.nio.channels.spi.SelectorProvider</ignore>
|
||||
<ignore>java.net.SocketOption</ignore>
|
||||
<ignore>java.nio.channels.NetworkChannel</ignore>
|
||||
|
||||
<!-- Self-signed certificate generation -->
|
||||
<ignore>sun.security.x509.AlgorithmId</ignore>
|
||||
|
@ -0,0 +1,82 @@
|
||||
/*
|
||||
* Copyright 2018 The Netty Project
|
||||
*
|
||||
* The Netty Project licenses this file to you under the Apache License,
|
||||
* version 2.0 (the "License"); you may not use this file except in compliance
|
||||
* with the License. You may obtain a copy of the License at:
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||
* License for the specific language governing permissions and limitations
|
||||
* under the License.
|
||||
*/
|
||||
package io.netty.channel.socket.nio;
|
||||
|
||||
import io.netty.channel.ChannelException;
|
||||
import io.netty.channel.ChannelOption;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.net.SocketOption;
|
||||
import java.nio.channels.NetworkChannel;
|
||||
import java.util.Set;
|
||||
|
||||
/**
|
||||
* Provides {@link ChannelOption} over a given {@link SocketOption} which is then passed through the underlying
|
||||
* {@link NetworkChannel}.
|
||||
*/
|
||||
public final class NioChannelOption<T> extends ChannelOption<T> {
|
||||
|
||||
private final SocketOption<T> option;
|
||||
|
||||
@SuppressWarnings("deprecation")
|
||||
private NioChannelOption(SocketOption<T> option) {
|
||||
super(option.name());
|
||||
this.option = option;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns a {@link ChannelOption} for the given {@link SocketOption}.
|
||||
*/
|
||||
public static <T> ChannelOption<T> of(SocketOption<T> option) {
|
||||
return new NioChannelOption<T>(option);
|
||||
}
|
||||
|
||||
// Internal helper methods to remove code duplication between Nio*Channel implementations.
|
||||
static <T> boolean setOption(NetworkChannel channel, NioChannelOption<T> option, T value) {
|
||||
if (!channel.supportedOptions().contains(option.option)) {
|
||||
return false;
|
||||
}
|
||||
try {
|
||||
channel.setOption(option.option, value);
|
||||
return true;
|
||||
} catch (IOException e) {
|
||||
throw new ChannelException(e);
|
||||
}
|
||||
}
|
||||
|
||||
static <T> T getOption(NetworkChannel channel, NioChannelOption<T> option) {
|
||||
if (!channel.supportedOptions().contains(option.option)) {
|
||||
return null;
|
||||
}
|
||||
try {
|
||||
return channel.getOption(option.option);
|
||||
} catch (IOException e) {
|
||||
throw new ChannelException(e);
|
||||
}
|
||||
}
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
static ChannelOption[] getOptions(NetworkChannel channel) {
|
||||
Set<SocketOption<?>> supportedOpts = channel.supportedOptions();
|
||||
ChannelOption<?>[] extraOpts = new ChannelOption[supportedOpts.size()];
|
||||
|
||||
int i = 0;
|
||||
for (SocketOption<?> opt : supportedOpts) {
|
||||
extraOpts[i++] = new NioChannelOption(opt);
|
||||
}
|
||||
return extraOpts;
|
||||
}
|
||||
}
|
@ -16,6 +16,7 @@
|
||||
package io.netty.channel.socket.nio;
|
||||
|
||||
import io.netty.channel.ChannelException;
|
||||
import io.netty.channel.ChannelOption;
|
||||
import io.netty.channel.socket.DatagramChannelConfig;
|
||||
import io.netty.channel.socket.DefaultDatagramChannelConfig;
|
||||
import io.netty.util.internal.PlatformDependent;
|
||||
@ -27,6 +28,7 @@ import java.net.NetworkInterface;
|
||||
import java.net.SocketException;
|
||||
import java.nio.channels.DatagramChannel;
|
||||
import java.util.Enumeration;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* The default {@link NioDatagramChannelConfig} implementation.
|
||||
@ -205,4 +207,29 @@ class NioDatagramChannelConfig extends DefaultDatagramChannelConfig {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public <T> boolean setOption(ChannelOption<T> option, T value) {
|
||||
if (PlatformDependent.javaVersion() >= 7 && option instanceof NioChannelOption) {
|
||||
return NioChannelOption.setOption(javaChannel, (NioChannelOption<T>) option, value);
|
||||
}
|
||||
return super.setOption(option, value);
|
||||
}
|
||||
|
||||
@Override
|
||||
public <T> T getOption(ChannelOption<T> option) {
|
||||
if (PlatformDependent.javaVersion() >= 7 && option instanceof NioChannelOption) {
|
||||
return NioChannelOption.getOption(javaChannel, (NioChannelOption<T>) option);
|
||||
}
|
||||
return super.getOption(option);
|
||||
}
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
@Override
|
||||
public Map<ChannelOption<?>, Object> getOptions() {
|
||||
if (PlatformDependent.javaVersion() >= 7) {
|
||||
return getOptions(super.getOptions(), NioChannelOption.getOptions(javaChannel));
|
||||
}
|
||||
return super.getOptions();
|
||||
}
|
||||
}
|
||||
|
@ -17,6 +17,7 @@ package io.netty.channel.socket.nio;
|
||||
|
||||
import io.netty.channel.ChannelException;
|
||||
import io.netty.channel.ChannelMetadata;
|
||||
import io.netty.channel.ChannelOption;
|
||||
import io.netty.channel.ChannelOutboundBuffer;
|
||||
import io.netty.util.internal.SocketUtils;
|
||||
import io.netty.channel.nio.AbstractNioMessageChannel;
|
||||
@ -35,6 +36,7 @@ import java.nio.channels.ServerSocketChannel;
|
||||
import java.nio.channels.SocketChannel;
|
||||
import java.nio.channels.spi.SelectorProvider;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* A {@link io.netty.channel.socket.ServerSocketChannel} implementation which uses
|
||||
@ -199,6 +201,35 @@ public class NioServerSocketChannel extends AbstractNioMessageChannel
|
||||
protected void autoReadCleared() {
|
||||
clearReadPending();
|
||||
}
|
||||
|
||||
@Override
|
||||
public <T> boolean setOption(ChannelOption<T> option, T value) {
|
||||
if (PlatformDependent.javaVersion() >= 7 && option instanceof NioChannelOption) {
|
||||
return NioChannelOption.setOption(jdkChannel(), (NioChannelOption<T>) option, value);
|
||||
}
|
||||
return super.setOption(option, value);
|
||||
}
|
||||
|
||||
@Override
|
||||
public <T> T getOption(ChannelOption<T> option) {
|
||||
if (PlatformDependent.javaVersion() >= 7 && option instanceof NioChannelOption) {
|
||||
return NioChannelOption.getOption(jdkChannel(), (NioChannelOption<T>) option);
|
||||
}
|
||||
return super.getOption(option);
|
||||
}
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
@Override
|
||||
public Map<ChannelOption<?>, Object> getOptions() {
|
||||
if (PlatformDependent.javaVersion() >= 7) {
|
||||
return getOptions(super.getOptions(), NioChannelOption.getOptions(jdkChannel()));
|
||||
}
|
||||
return super.getOptions();
|
||||
}
|
||||
|
||||
private ServerSocketChannel jdkChannel() {
|
||||
return ((NioServerSocketChannel) channel).javaChannel();
|
||||
}
|
||||
}
|
||||
|
||||
// Override just to to be able to call directly via unit tests.
|
||||
|
@ -20,6 +20,7 @@ import io.netty.channel.Channel;
|
||||
import io.netty.channel.ChannelException;
|
||||
import io.netty.channel.ChannelFuture;
|
||||
import io.netty.channel.ChannelFutureListener;
|
||||
import io.netty.channel.ChannelOption;
|
||||
import io.netty.channel.ChannelOutboundBuffer;
|
||||
import io.netty.channel.ChannelPromise;
|
||||
import io.netty.channel.EventLoop;
|
||||
@ -44,6 +45,7 @@ import java.nio.ByteBuffer;
|
||||
import java.nio.channels.SelectionKey;
|
||||
import java.nio.channels.SocketChannel;
|
||||
import java.nio.channels.spi.SelectorProvider;
|
||||
import java.util.Map;
|
||||
import java.util.concurrent.Executor;
|
||||
|
||||
import static io.netty.channel.internal.ChannelUtils.MAX_BYTES_PER_GATHERING_WRITE_ATTEMPTED_LOW_THRESHOLD;
|
||||
@ -461,7 +463,6 @@ public class NioSocketChannel extends AbstractNioByteChannel implements io.netty
|
||||
|
||||
private final class NioSocketChannelConfig extends DefaultSocketChannelConfig {
|
||||
private volatile int maxBytesPerGatheringWrite = Integer.MAX_VALUE;
|
||||
|
||||
private NioSocketChannelConfig(NioSocketChannel channel, Socket javaSocket) {
|
||||
super(channel, javaSocket);
|
||||
calculateMaxBytesPerGatheringWrite();
|
||||
@ -479,6 +480,31 @@ public class NioSocketChannel extends AbstractNioByteChannel implements io.netty
|
||||
return this;
|
||||
}
|
||||
|
||||
@Override
|
||||
public <T> boolean setOption(ChannelOption<T> option, T value) {
|
||||
if (PlatformDependent.javaVersion() >= 7 && option instanceof NioChannelOption) {
|
||||
return NioChannelOption.setOption(jdkChannel(), (NioChannelOption<T>) option, value);
|
||||
}
|
||||
return super.setOption(option, value);
|
||||
}
|
||||
|
||||
@Override
|
||||
public <T> T getOption(ChannelOption<T> option) {
|
||||
if (PlatformDependent.javaVersion() >= 7 && option instanceof NioChannelOption) {
|
||||
return NioChannelOption.getOption(jdkChannel(), (NioChannelOption<T>) option);
|
||||
}
|
||||
return super.getOption(option);
|
||||
}
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
@Override
|
||||
public Map<ChannelOption<?>, Object> getOptions() {
|
||||
if (PlatformDependent.javaVersion() >= 7) {
|
||||
return getOptions(super.getOptions(), NioChannelOption.getOptions(jdkChannel()));
|
||||
}
|
||||
return super.getOptions();
|
||||
}
|
||||
|
||||
void setMaxBytesPerGatheringWrite(int maxBytesPerGatheringWrite) {
|
||||
this.maxBytesPerGatheringWrite = maxBytesPerGatheringWrite;
|
||||
}
|
||||
@ -494,5 +520,9 @@ public class NioSocketChannel extends AbstractNioByteChannel implements io.netty
|
||||
setMaxBytesPerGatheringWrite(getSendBufferSize() << 1);
|
||||
}
|
||||
}
|
||||
|
||||
private SocketChannel jdkChannel() {
|
||||
return ((NioSocketChannel) channel).javaChannel();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
@ -0,0 +1,69 @@
|
||||
/*
|
||||
* Copyright 2018 The Netty Project
|
||||
*
|
||||
* The Netty Project licenses this file to you under the Apache License,
|
||||
* version 2.0 (the "License"); you may not use this file except in compliance
|
||||
* with the License. You may obtain a copy of the License at:
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||
* License for the specific language governing permissions and limitations
|
||||
* under the License.
|
||||
*/
|
||||
package io.netty.channel.socket.nio;
|
||||
|
||||
import io.netty.channel.ChannelOption;
|
||||
import io.netty.channel.nio.AbstractNioChannel;
|
||||
import org.junit.Test;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.net.SocketOption;
|
||||
import java.net.StandardSocketOptions;
|
||||
import java.nio.channels.NetworkChannel;
|
||||
|
||||
import static org.junit.Assert.*;
|
||||
|
||||
public abstract class AbstractNioChannelTest<T extends AbstractNioChannel> {
|
||||
|
||||
protected abstract T newNioChannel();
|
||||
|
||||
protected abstract NetworkChannel jdkChannel(T channel);
|
||||
|
||||
protected abstract SocketOption<?> newInvalidOption();
|
||||
|
||||
@Test
|
||||
public void testNioChannelOption() throws IOException {
|
||||
T channel = newNioChannel();
|
||||
try {
|
||||
NetworkChannel jdkChannel = jdkChannel(channel);
|
||||
ChannelOption<Boolean> option = NioChannelOption.of(StandardSocketOptions.SO_REUSEADDR);
|
||||
boolean value1 = jdkChannel.getOption(StandardSocketOptions.SO_REUSEADDR);
|
||||
boolean value2 = channel.config().getOption(option);
|
||||
|
||||
assertEquals(value1, value2);
|
||||
|
||||
channel.config().setOption(option, !value2);
|
||||
boolean value3 = jdkChannel.getOption(StandardSocketOptions.SO_REUSEADDR);
|
||||
boolean value4 = channel.config().getOption(option);
|
||||
assertEquals(value3, value4);
|
||||
assertNotEquals(value1, value4);
|
||||
} finally {
|
||||
channel.unsafe().closeForcibly();
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testInvalidNioChannelOption() {
|
||||
T channel = newNioChannel();
|
||||
try {
|
||||
ChannelOption<?> option = NioChannelOption.of(newInvalidOption());
|
||||
assertFalse(channel.config().setOption(option, null));
|
||||
assertNull(channel.config().getOption(option));
|
||||
} finally {
|
||||
channel.unsafe().closeForcibly();
|
||||
}
|
||||
}
|
||||
}
|
@ -13,24 +13,27 @@
|
||||
* License for the specific language governing permissions and limitations
|
||||
* under the License.
|
||||
*/
|
||||
package io.netty.channel.nio;
|
||||
package io.netty.channel.socket.nio;
|
||||
|
||||
import io.netty.bootstrap.Bootstrap;
|
||||
import io.netty.channel.ChannelHandlerContext;
|
||||
import io.netty.channel.ChannelInboundHandlerAdapter;
|
||||
import io.netty.channel.ChannelOption;
|
||||
import io.netty.channel.group.DefaultChannelGroup;
|
||||
import io.netty.channel.nio.NioEventLoopGroup;
|
||||
import io.netty.channel.socket.DatagramChannel;
|
||||
import io.netty.channel.socket.nio.NioDatagramChannel;
|
||||
import io.netty.util.ReferenceCountUtil;
|
||||
import io.netty.util.concurrent.GlobalEventExecutor;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Test;
|
||||
|
||||
import java.net.InetSocketAddress;
|
||||
import java.net.SocketOption;
|
||||
import java.net.StandardSocketOptions;
|
||||
import java.nio.channels.NetworkChannel;
|
||||
|
||||
|
||||
public class NioDatagramChannelTest {
|
||||
public class NioDatagramChannelTest extends AbstractNioChannelTest<NioDatagramChannel> {
|
||||
|
||||
/**
|
||||
* Test try to reproduce issue #1335
|
||||
@ -61,4 +64,19 @@ public class NioDatagramChannelTest {
|
||||
group.shutdownGracefully().sync();
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
protected NioDatagramChannel newNioChannel() {
|
||||
return new NioDatagramChannel();
|
||||
}
|
||||
|
||||
@Override
|
||||
protected NetworkChannel jdkChannel(NioDatagramChannel channel) {
|
||||
return channel.javaChannel();
|
||||
}
|
||||
|
||||
@Override
|
||||
protected SocketOption<?> newInvalidOption() {
|
||||
return StandardSocketOptions.TCP_NODELAY;
|
||||
}
|
||||
}
|
@ -22,9 +22,12 @@ import org.junit.Test;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.net.InetSocketAddress;
|
||||
import java.net.SocketOption;
|
||||
import java.net.StandardSocketOptions;
|
||||
import java.nio.channels.NetworkChannel;
|
||||
import java.nio.channels.ServerSocketChannel;
|
||||
|
||||
public class NioServerSocketChannelTest {
|
||||
public class NioServerSocketChannelTest extends AbstractNioChannelTest<NioServerSocketChannel> {
|
||||
|
||||
@Test
|
||||
public void testCloseOnError() throws Exception {
|
||||
@ -41,4 +44,19 @@ public class NioServerSocketChannelTest {
|
||||
group.shutdownGracefully();
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
protected NioServerSocketChannel newNioChannel() {
|
||||
return new NioServerSocketChannel();
|
||||
}
|
||||
|
||||
@Override
|
||||
protected NetworkChannel jdkChannel(NioServerSocketChannel channel) {
|
||||
return channel.javaChannel();
|
||||
}
|
||||
|
||||
@Override
|
||||
protected SocketOption<?> newInvalidOption() {
|
||||
return StandardSocketOptions.IP_MULTICAST_IF;
|
||||
}
|
||||
}
|
||||
|
@ -13,7 +13,7 @@
|
||||
* License for the specific language governing permissions and limitations
|
||||
* under the License.
|
||||
*/
|
||||
package io.netty.channel.nio;
|
||||
package io.netty.channel.socket.nio;
|
||||
|
||||
import io.netty.bootstrap.Bootstrap;
|
||||
import io.netty.bootstrap.ServerBootstrap;
|
||||
@ -30,9 +30,8 @@ import io.netty.channel.ChannelPipeline;
|
||||
import io.netty.channel.EventLoop;
|
||||
import io.netty.channel.EventLoopGroup;
|
||||
import io.netty.channel.SimpleChannelInboundHandler;
|
||||
import io.netty.channel.nio.NioEventLoopGroup;
|
||||
import io.netty.channel.socket.SocketChannel;
|
||||
import io.netty.channel.socket.nio.NioServerSocketChannel;
|
||||
import io.netty.channel.socket.nio.NioSocketChannel;
|
||||
import io.netty.util.CharsetUtil;
|
||||
import io.netty.util.NetUtil;
|
||||
import io.netty.util.internal.PlatformDependent;
|
||||
@ -46,7 +45,10 @@ import java.net.InetSocketAddress;
|
||||
import java.net.ServerSocket;
|
||||
import java.net.Socket;
|
||||
import java.net.SocketAddress;
|
||||
import java.net.SocketOption;
|
||||
import java.net.StandardSocketOptions;
|
||||
import java.nio.channels.ClosedChannelException;
|
||||
import java.nio.channels.NetworkChannel;
|
||||
import java.util.Queue;
|
||||
import java.util.concurrent.CountDownLatch;
|
||||
import java.util.concurrent.LinkedBlockingQueue;
|
||||
@ -55,7 +57,7 @@ import static org.hamcrest.CoreMatchers.*;
|
||||
import static org.junit.Assert.*;
|
||||
|
||||
|
||||
public class NioSocketChannelTest {
|
||||
public class NioSocketChannelTest extends AbstractNioChannelTest<NioSocketChannel> {
|
||||
|
||||
/**
|
||||
* Reproduces the issue #1600
|
||||
@ -278,4 +280,19 @@ public class NioSocketChannelTest {
|
||||
group.shutdownGracefully();
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
protected NioSocketChannel newNioChannel() {
|
||||
return new NioSocketChannel();
|
||||
}
|
||||
|
||||
@Override
|
||||
protected NetworkChannel jdkChannel(NioSocketChannel channel) {
|
||||
return channel.javaChannel();
|
||||
}
|
||||
|
||||
@Override
|
||||
protected SocketOption<?> newInvalidOption() {
|
||||
return StandardSocketOptions.IP_MULTICAST_IF;
|
||||
}
|
||||
}
|
Loading…
x
Reference in New Issue
Block a user