org.apache.camel.component.netty.ClientModeTCPNettyServerBootstrapFactory Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of camel-netty Show documentation
Show all versions of camel-netty Show documentation
Camel Netty NIO based socket communication component
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF 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 org.apache.camel.component.netty;
import java.net.ConnectException;
import java.net.InetSocketAddress;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.TimeUnit;
import io.netty.bootstrap.Bootstrap;
import io.netty.channel.Channel;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelFutureListener;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelOption;
import io.netty.channel.EventLoop;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.epoll.EpollSocketChannel;
import io.netty.channel.socket.nio.NioSocketChannel;
import org.apache.camel.CamelContext;
import org.apache.camel.support.service.ServiceSupport;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* A {@link NettyServerBootstrapFactory} which is used by a single consumer (not shared).
*/
public class ClientModeTCPNettyServerBootstrapFactory extends ServiceSupport implements NettyServerBootstrapFactory {
protected static final Logger LOG = LoggerFactory.getLogger(ClientModeTCPNettyServerBootstrapFactory.class);
private CamelContext camelContext;
private ThreadFactory threadFactory;
private NettyServerBootstrapConfiguration configuration;
private ChannelInitializer pipelineFactory;
private Bootstrap clientBootstrap;
private Channel channel;
private EventLoopGroup workerGroup;
public ClientModeTCPNettyServerBootstrapFactory() {
}
@Override
public void init(
CamelContext camelContext, NettyServerBootstrapConfiguration configuration,
ChannelInitializer pipelineFactory) {
this.camelContext = camelContext;
this.configuration = configuration;
this.pipelineFactory = pipelineFactory;
}
@Override
public void init(
ThreadFactory threadFactory, NettyServerBootstrapConfiguration configuration,
ChannelInitializer pipelineFactory) {
this.threadFactory = threadFactory;
this.configuration = configuration;
this.pipelineFactory = pipelineFactory;
}
@Override
public void addChannel(Channel channel) {
// we don't need to track the channel in client mode
}
@Override
public void removeChannel(Channel channel) {
// we don't need to track the channel in client mode
}
@Override
public void addConsumer(NettyConsumer consumer) {
// does not allow sharing
}
@Override
public void removeConsumer(NettyConsumer consumer) {
// does not allow sharing
}
@Override
protected void doStart() throws Exception {
if (camelContext == null && threadFactory == null) {
throw new IllegalArgumentException("Either CamelContext or ThreadFactory must be set on " + this);
}
startServerBootstrap();
}
@Override
protected void doStop() throws Exception {
stopServerBootstrap();
}
protected void startServerBootstrap() throws Exception {
// prefer using explicit configured thread pools
EventLoopGroup wg = configuration.getWorkerGroup();
if (wg == null) {
// create new pool which we should shutdown when stopping as its not shared
workerGroup = new NettyWorkerPoolBuilder()
.withNativeTransport(configuration.isNativeTransport())
.withWorkerCount(configuration.getWorkerCount())
.withName("NettyServerTCPWorker")
.build();
wg = workerGroup;
}
clientBootstrap = new Bootstrap();
if (configuration.isNativeTransport()) {
clientBootstrap.channel(EpollSocketChannel.class);
} else {
clientBootstrap.channel(NioSocketChannel.class);
}
clientBootstrap.group(wg);
clientBootstrap.option(ChannelOption.SO_KEEPALIVE, configuration.isKeepAlive());
clientBootstrap.option(ChannelOption.TCP_NODELAY, configuration.isTcpNoDelay());
clientBootstrap.option(ChannelOption.SO_REUSEADDR, configuration.isReuseAddress());
clientBootstrap.option(ChannelOption.CONNECT_TIMEOUT_MILLIS, configuration.getConnectTimeout());
LOG.debug("Created ClientBootstrap {}", clientBootstrap);
clientBootstrap.handler(pipelineFactory);
ChannelFuture channelFuture
= clientBootstrap.connect(new InetSocketAddress(configuration.getHost(), configuration.getPort()));
LOG.debug("Created new TCP client bootstrap connecting to {}:{} with options: {}",
configuration.getHost(), configuration.getPort(), clientBootstrap);
LOG.info("ClientModeServerBootstrap binding to {}:{}", configuration.getHost(), configuration.getPort());
channel = openChannel(channelFuture);
}
protected void stopServerBootstrap() {
// close all channels
LOG.info("ClientModeServerBootstrap unbinding from {}:{}", configuration.getHost(), configuration.getPort());
if (workerGroup != null) {
workerGroup.shutdownGracefully();
workerGroup = null;
}
}
protected void doReconnectIfNeeded() throws Exception {
if (channel == null || !channel.isActive()) {
LOG.debug("ClientModeServerBootstrap re-connect to {}:{}", configuration.getHost(), configuration.getPort());
ChannelFuture connectFuture
= clientBootstrap.connect(new InetSocketAddress(configuration.getHost(), configuration.getPort()));
channel = openChannel(connectFuture);
}
}
protected Channel openChannel(final ChannelFuture channelFuture) throws Exception {
// blocking for channel to be done
if (LOG.isTraceEnabled()) {
LOG.trace("Waiting for operation to complete {} for {} millis", channelFuture, configuration.getConnectTimeout());
}
// wait for the channel to be open (see io.netty.channel.ChannelFuture javadoc for example/recommendation)
channelFuture.awaitUninterruptibly();
if (!channelFuture.isDone() || !channelFuture.isSuccess()) {
//check if reconnect is enabled and schedule a reconnect, if from handler then don't schedule a reconnect
if (configuration.isReconnect()) {
scheduleReconnect(channelFuture);
return null;
} else {
ConnectException cause = new ConnectException("Cannot connect to " + configuration.getAddress());
if (channelFuture.cause() != null) {
cause.initCause(channelFuture.cause());
}
throw cause;
}
}
Channel answer = channelFuture.channel();
if (LOG.isDebugEnabled()) {
LOG.debug("Creating connector to address: {}", configuration.getAddress());
}
// schedule a reconnect to happen when the channel closes
if (configuration.isReconnect()) {
answer.closeFuture().addListener(new ChannelFutureListener() {
@Override
public void operationComplete(ChannelFuture future) throws Exception {
scheduleReconnect(channelFuture);
}
});
}
return answer;
}
private void scheduleReconnect(final ChannelFuture channelFuture) {
final EventLoop loop = channelFuture.channel().eventLoop();
loop.schedule(() -> {
try {
LOG.trace("Re-connecting to {} if needed", configuration.getAddress());
doReconnectIfNeeded();
} catch (Exception e) {
LOG.warn("Error during re-connect to {}. Will attempt again in {} millis. This exception is ignored.",
configuration.getAddress(), configuration.getReconnectInterval(),
e);
}
}, configuration.getReconnectInterval(), TimeUnit.MILLISECONDS);
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy