/*
* Copyright (c) 2014 AsyncHttpClient Project. All rights reserved.
*
* This program is licensed to you under the Apache License Version 2.0,
* and you may not use this file except in compliance with the Apache License Version 2.0.
* You may obtain a copy of the Apache License Version 2.0 at
* http://www.apache.org/licenses/LICENSE-2.0.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the Apache License Version 2.0 is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the Apache License Version 2.0 for the specific language governing permissions and limitations there under.
*/
package org.asynchttpclient.providers.netty;
import io.netty.util.HashedWheelTimer;
import io.netty.util.Timer;
import java.util.concurrent.atomic.AtomicBoolean;
import org.asynchttpclient.AsyncHandler;
import org.asynchttpclient.AsyncHttpClientConfig;
import org.asynchttpclient.AsyncHttpProvider;
import org.asynchttpclient.ListenableFuture;
import org.asynchttpclient.Request;
import org.asynchttpclient.providers.netty.channel.ChannelManager;
import org.asynchttpclient.providers.netty.channel.pool.ChannelPoolPartitionSelector;
import org.asynchttpclient.providers.netty.request.NettyRequestSender;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class NettyAsyncHttpProvider implements AsyncHttpProvider {
private static final Logger LOGGER = LoggerFactory.getLogger(NettyAsyncHttpProvider.class);
private final NettyAsyncHttpProviderConfig nettyConfig;
private final AtomicBoolean closed = new AtomicBoolean(false);
private final ChannelManager channelManager;
private final NettyRequestSender requestSender;
private final boolean allowStopNettyTimer;
private final Timer nettyTimer;
public NettyAsyncHttpProvider(AsyncHttpClientConfig config) {
nettyConfig = config.getAsyncHttpProviderConfig() instanceof NettyAsyncHttpProviderConfig ? //
(NettyAsyncHttpProviderConfig) config.getAsyncHttpProviderConfig()
: new NettyAsyncHttpProviderConfig();
allowStopNettyTimer = nettyConfig.getNettyTimer() == null;
nettyTimer = allowStopNettyTimer ? newNettyTimer() : nettyConfig.getNettyTimer();
channelManager = new ChannelManager(config, nettyConfig, nettyTimer);
requestSender = new NettyRequestSender(config, nettyConfig, channelManager, nettyTimer, closed);
channelManager.configureBootstraps(requestSender, closed);
}
private Timer newNettyTimer() {
HashedWheelTimer timer = new HashedWheelTimer();
timer.start();
return timer;
}
@Override
public void close() {
if (closed.compareAndSet(false, true)) {
try {
channelManager.close();
if (allowStopNettyTimer)
nettyTimer.stop();
} catch (Throwable t) {
LOGGER.warn("Unexpected error on close", t);
}
}
}
@Override
public <T> ListenableFuture<T> execute(Request request, final AsyncHandler<T> asyncHandler) {
try {
return requestSender.sendRequest(request, asyncHandler, null, false);
} catch (Exception e) {
asyncHandler.onThrowable(e);
return new ListenableFuture.CompletedFailure<>(e);
}
}
public void flushChannelPoolPartition(String partitionId) {
channelManager.flushPartition(partitionId);
}
public void flushChannelPoolPartitions(ChannelPoolPartitionSelector selector) {
channelManager.flushPartitions(selector);
}
}