@Override
public void completed(final ClientExchange result) {
final ClientResponse response = result.getResponse();
if (log.isDebugEnabled()) {
log.debugf(
"Received response %s for request %s for exchange %s",
response, result.getRequest(), exchange);
}
final HeaderMap inboundResponseHeaders = response.getResponseHeaders();
final HeaderMap outboundResponseHeaders = exchange.getResponseHeaders();
exchange.setStatusCode(response.getResponseCode());
copyHeaders(outboundResponseHeaders, inboundResponseHeaders);
if (exchange.isUpgrade()) {
exchange.upgradeChannel(
new HttpUpgradeListener() {
@Override
public void handleUpgrade(
StreamConnection streamConnection, HttpServerExchange exchange) {
if (log.isDebugEnabled()) {
log.debugf(
"Upgraded request %s to for exchange %s", result.getRequest(), exchange);
}
StreamConnection clientChannel = null;
try {
clientChannel = result.getConnection().performUpgrade();
final ClosingExceptionHandler handler =
new ClosingExceptionHandler(streamConnection, clientChannel);
Transfer.initiateTransfer(
clientChannel.getSourceChannel(),
streamConnection.getSinkChannel(),
ChannelListeners.closingChannelListener(),
ChannelListeners.writeShutdownChannelListener(
ChannelListeners.<StreamSinkChannel>flushingChannelListener(
ChannelListeners.closingChannelListener(),
ChannelListeners.closingChannelExceptionHandler()),
ChannelListeners.closingChannelExceptionHandler()),
handler,
handler,
result.getConnection().getBufferPool());
Transfer.initiateTransfer(
streamConnection.getSourceChannel(),
clientChannel.getSinkChannel(),
ChannelListeners.closingChannelListener(),
ChannelListeners.writeShutdownChannelListener(
ChannelListeners.<StreamSinkChannel>flushingChannelListener(
ChannelListeners.closingChannelListener(),
ChannelListeners.closingChannelExceptionHandler()),
ChannelListeners.closingChannelExceptionHandler()),
handler,
handler,
result.getConnection().getBufferPool());
} catch (IOException e) {
IoUtils.safeClose(streamConnection, clientChannel);
}
}
});
}
final IoExceptionHandler handler = new IoExceptionHandler(exchange, result.getConnection());
Transfer.initiateTransfer(
result.getResponseChannel(),
exchange.getResponseChannel(),
ChannelListeners.closingChannelListener(),
new HTTPTrailerChannelListener(result, exchange),
handler,
handler,
exchange.getConnection().getByteBufferPool());
}
