181 lines
5.1 KiB
Java
181 lines
5.1 KiB
Java
/*
|
|
* Copyright 2012 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.channel.local;
|
|
|
|
import io.netty.channel.AbstractServerChannel;
|
|
import io.netty.channel.ChannelConfig;
|
|
import io.netty.channel.ChannelPipeline;
|
|
import io.netty.channel.DefaultChannelConfig;
|
|
import io.netty.channel.EventLoop;
|
|
import io.netty.channel.EventLoopGroup;
|
|
import io.netty.channel.PreferHeapByteBufAllocator;
|
|
import io.netty.channel.RecvByteBufAllocator;
|
|
import io.netty.channel.ServerChannel;
|
|
import io.netty.util.concurrent.Promise;
|
|
|
|
import java.net.SocketAddress;
|
|
import java.util.ArrayDeque;
|
|
import java.util.Queue;
|
|
|
|
/**
|
|
* A {@link ServerChannel} for the local transport which allows in VM communication.
|
|
*/
|
|
public class LocalServerChannel extends AbstractServerChannel {
|
|
|
|
private final ChannelConfig config = new DefaultChannelConfig(this);
|
|
private final Queue<Object> inboundBuffer = new ArrayDeque<>();
|
|
|
|
private volatile int state; // 0 - open, 1 - active, 2 - closed
|
|
private volatile LocalAddress localAddress;
|
|
private volatile boolean acceptInProgress;
|
|
|
|
public LocalServerChannel(EventLoop eventLoop, EventLoopGroup childEventLoopGroup) {
|
|
super(eventLoop, childEventLoopGroup);
|
|
config().setAllocator(new PreferHeapByteBufAllocator(config.getAllocator()));
|
|
}
|
|
|
|
@Override
|
|
public ChannelConfig config() {
|
|
return config;
|
|
}
|
|
|
|
@Override
|
|
public LocalAddress localAddress() {
|
|
return (LocalAddress) super.localAddress();
|
|
}
|
|
|
|
@Override
|
|
public LocalAddress remoteAddress() {
|
|
return (LocalAddress) super.remoteAddress();
|
|
}
|
|
|
|
@Override
|
|
public boolean isOpen() {
|
|
return state < 2;
|
|
}
|
|
|
|
@Override
|
|
public boolean isActive() {
|
|
return state == 1;
|
|
}
|
|
|
|
@Override
|
|
protected SocketAddress localAddress0() {
|
|
return localAddress;
|
|
}
|
|
|
|
@Override
|
|
protected void doBind(SocketAddress localAddress) throws Exception {
|
|
this.localAddress = LocalChannelRegistry.register(this, this.localAddress, localAddress);
|
|
state = 1;
|
|
}
|
|
|
|
@Override
|
|
protected void doClose() throws Exception {
|
|
if (state <= 1) {
|
|
// Update all internal state before the closeFuture is notified.
|
|
if (localAddress != null) {
|
|
LocalChannelRegistry.unregister(localAddress);
|
|
localAddress = null;
|
|
}
|
|
state = 2;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
protected void doBeginRead() throws Exception {
|
|
if (acceptInProgress) {
|
|
return;
|
|
}
|
|
|
|
Queue<Object> inboundBuffer = this.inboundBuffer;
|
|
if (inboundBuffer.isEmpty()) {
|
|
acceptInProgress = true;
|
|
return;
|
|
}
|
|
|
|
readInbound();
|
|
}
|
|
|
|
LocalChannel serve(final LocalChannel peer) {
|
|
final LocalChannel child = newLocalChannel(peer);
|
|
if (executor().inEventLoop()) {
|
|
serve0(child);
|
|
} else {
|
|
executor().execute(() -> serve0(child));
|
|
}
|
|
return child;
|
|
}
|
|
|
|
private void readInbound() {
|
|
RecvByteBufAllocator.Handle handle = unsafe().recvBufAllocHandle();
|
|
handle.reset(config());
|
|
ChannelPipeline pipeline = pipeline();
|
|
do {
|
|
Object m = inboundBuffer.poll();
|
|
if (m == null) {
|
|
break;
|
|
}
|
|
pipeline.fireChannelRead(m);
|
|
} while (handle.continueReading());
|
|
|
|
pipeline.fireChannelReadComplete();
|
|
readIfIsAutoRead();
|
|
}
|
|
|
|
/**
|
|
* A factory method for {@link LocalChannel}s. Users may override it
|
|
* to create custom instances of {@link LocalChannel}s.
|
|
*/
|
|
protected LocalChannel newLocalChannel(LocalChannel peer) {
|
|
return new LocalChannel(this, childEventLoopGroup().next(), peer);
|
|
}
|
|
|
|
private void serve0(final LocalChannel child) {
|
|
inboundBuffer.add(child);
|
|
if (acceptInProgress) {
|
|
acceptInProgress = false;
|
|
|
|
readInbound();
|
|
}
|
|
}
|
|
|
|
@Override
|
|
protected AbstractUnsafe newUnsafe() {
|
|
return new DefaultServerUnsafe();
|
|
}
|
|
|
|
private final class DefaultServerUnsafe extends AbstractUnsafe implements LocalChannelUnsafe {
|
|
@Override
|
|
public void connect(SocketAddress remoteAddress, SocketAddress localAddress, Promise<Void> promise) {
|
|
safeSetFailure(promise, new UnsupportedOperationException());
|
|
}
|
|
|
|
@Override
|
|
public void register0() {
|
|
}
|
|
|
|
@Override
|
|
public void deregister0() {
|
|
}
|
|
|
|
@Override
|
|
public Promise<Void> newPromise() {
|
|
return LocalServerChannel.this.newPromise();
|
|
}
|
|
}
|
|
}
|