io.netty.incubator.codec.quic.QuicheQuicStreamChannelConfig Maven / Gradle / Ivy
/*
* Copyright 2020 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:
*
* https://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.incubator.codec.quic;
import io.netty.buffer.ByteBufAllocator;
import io.netty.channel.ChannelOption;
import io.netty.channel.DefaultChannelConfig;
import io.netty.channel.MessageSizeEstimator;
import io.netty.channel.RecvByteBufAllocator;
import io.netty.channel.WriteBufferWaterMark;
import java.util.Map;
final class QuicheQuicStreamChannelConfig extends DefaultChannelConfig implements QuicStreamChannelConfig {
// We should use half-closure sementatics by default as this is what QUIC does by default.
// If you receive a FIN you should still keep the stream open until you write a FIN as well.
private volatile boolean allowHalfClosure = true;
private volatile boolean readFrames;
volatile DirectIoByteBufAllocator allocator;
QuicheQuicStreamChannelConfig(QuicStreamChannel channel) {
super(channel);
allocator = new DirectIoByteBufAllocator(super.getAllocator());
}
@Override
public Map, Object> getOptions() {
if (isHalfClosureSupported()) {
return getOptions(super.getOptions(), ChannelOption.ALLOW_HALF_CLOSURE, QuicChannelOption.READ_FRAMES);
}
return super.getOptions();
}
@SuppressWarnings("unchecked")
@Override
public T getOption(ChannelOption option) {
if (option == ChannelOption.ALLOW_HALF_CLOSURE) {
return (T) Boolean.valueOf(isAllowHalfClosure());
}
if (option == QuicChannelOption.READ_FRAMES) {
return (T) Boolean.valueOf(isReadFrames());
}
return super.getOption(option);
}
@Override
public boolean setOption(ChannelOption option, T value) {
validate(option, value);
if (option == ChannelOption.ALLOW_HALF_CLOSURE) {
if (isHalfClosureSupported()) {
setAllowHalfClosure((Boolean) value);
return true;
}
return false;
}
if (option == QuicChannelOption.READ_FRAMES) {
setReadFrames((Boolean) value);
}
return super.setOption(option, value);
}
@Override
public QuicStreamChannelConfig setReadFrames(boolean readFrames) {
this.readFrames = readFrames;
return this;
}
@Override
public boolean isReadFrames() {
return readFrames;
}
@Override
public QuicStreamChannelConfig setConnectTimeoutMillis(int connectTimeoutMillis) {
super.setConnectTimeoutMillis(connectTimeoutMillis);
return this;
}
@Override
public QuicStreamChannelConfig setMaxMessagesPerRead(int maxMessagesPerRead) {
super.setMaxMessagesPerRead(maxMessagesPerRead);
return this;
}
@Override
public QuicStreamChannelConfig setWriteSpinCount(int writeSpinCount) {
super.setWriteSpinCount(writeSpinCount);
return this;
}
@Override
public QuicStreamChannelConfig setAllocator(ByteBufAllocator allocator) {
this.allocator = new DirectIoByteBufAllocator(allocator);
return this;
}
@Override
public QuicStreamChannelConfig setRecvByteBufAllocator(RecvByteBufAllocator allocator) {
super.setRecvByteBufAllocator(allocator);
return this;
}
@Override
public QuicStreamChannelConfig setAutoRead(boolean autoRead) {
super.setAutoRead(autoRead);
return this;
}
@Override
public QuicStreamChannelConfig setAutoClose(boolean autoClose) {
super.setAutoClose(autoClose);
return this;
}
@Override
public QuicStreamChannelConfig setWriteBufferHighWaterMark(int writeBufferHighWaterMark) {
super.setWriteBufferHighWaterMark(writeBufferHighWaterMark);
return this;
}
@Override
public QuicStreamChannelConfig setWriteBufferLowWaterMark(int writeBufferLowWaterMark) {
super.setWriteBufferLowWaterMark(writeBufferLowWaterMark);
return this;
}
@Override
public QuicStreamChannelConfig setWriteBufferWaterMark(WriteBufferWaterMark writeBufferWaterMark) {
super.setWriteBufferWaterMark(writeBufferWaterMark);
return this;
}
@Override
public QuicStreamChannelConfig setMessageSizeEstimator(MessageSizeEstimator estimator) {
super.setMessageSizeEstimator(estimator);
return this;
}
@Override
public QuicStreamChannelConfig setAllowHalfClosure(boolean allowHalfClosure) {
if (!isHalfClosureSupported()) {
throw new UnsupportedOperationException("Undirectional streams don't support half-closure");
}
this.allowHalfClosure = allowHalfClosure;
return this;
}
@Override
public ByteBufAllocator getAllocator() {
return allocator.wrapped();
}
@Override
public boolean isAllowHalfClosure() {
return allowHalfClosure;
}
private boolean isHalfClosureSupported() {
return ((QuicStreamChannel) channel).type() == QuicStreamType.BIDIRECTIONAL;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy