grpc
/
grpc-java
mirror of https://github.com/grpc/grpc-java.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Simplify connection callback handling in Netty 

Channel is available immediately after connect(), so register callbacks
immediately instead of delaying.

Setting channel is now delayed until it is actually safe to use.
This commit is contained in:
Eric Anderson 2015-02-04 13:01:17 -08:00
parent e6a7b6c8a9
commit c84ef8332e
1 changed files with 36 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

70
netty/src/main/java/io/grpc/transport/netty/NettyClientTransport.java
View File

@ -178,7 +178,8 @@ class NettyClientTransport extends AbstractClientTransport {
b.handler(negotiation.initializer());
// Start the connection operation to the server.
b.connect(address).addListener(new ChannelFutureListener() {
final ChannelFuture connectFuture = b.connect(address);
connectFuture.addListener(new ChannelFutureListener() {
@Override
public void operationComplete(ChannelFuture future) throws Exception {
if (!future.isSuccess()) {
@ -188,43 +189,44 @@ class NettyClientTransport extends AbstractClientTransport {
}
// Connected successfully, start the protocol negotiation.
channel = future.channel();
negotiation.onConnected(channel);
}
});
Futures.addCallback(negotiation.completeFuture(), new FutureCallback<Void>() {
@Override
public void onSuccess(Void result) {
// The negotiation was successful.
// We should not send on the channel until negotiation completes. This is a hard requirement
// by SslHandler but is appropriate for HTTP/1.1 Upgrade as well.
channel = connectFuture.channel();
notifyStarted();
}
final ListenableFuture<Void> negotiationFuture = negotiation.completeFuture();
Futures.addCallback(negotiationFuture, new FutureCallback<Void>() {
@Override
public void onSuccess(Void result) {
// The negotiation was successful.
notifyStarted();
@Override
public void onFailure(Throwable t) {
// The negotiation failed.
notifyFailed(t);
}
});
// Handle transport shutdown when the channel is closed.
channel.closeFuture().addListener(new ChannelFutureListener() {
@Override
public void operationComplete(ChannelFuture future) throws Exception {
if (!future.isSuccess()) {
// The close failed. Just notify that transport shutdown failed.
notifyFailed(future.cause());
return;
}
Channel channel = connectFuture.channel();
// Handle transport shutdown when the channel is closed.
channel.closeFuture().addListener(new ChannelFutureListener() {
@Override
public void operationComplete(ChannelFuture future) throws Exception {
if (!future.isSuccess()) {
// The close failed. Just notify that transport shutdown failed.
notifyFailed(future.cause());
return;
}
if (handler.connectionError() != null) {
// The handler encountered a connection error.
notifyFailed(handler.connectionError());
} else {
// Normal termination of the connection.
notifyStopped();
}
}
});
}
@Override
public void onFailure(Throwable t) {
// The negotiation failed.
notifyFailed(t);
}
});
if (handler.connectionError() != null) {
// The handler encountered a connection error.
notifyFailed(handler.connectionError());
} else {
// Normal termination of the connection.
notifyStopped();
}
}
});
}