Add handshake complete callback back

Author: Kyrylo Stokoz

src/main/java/com/ning/http/client/AsyncHandlerExtensions.java

@@ -64,4 +64,9 @@ public interface AsyncHandlerExtensions {
      * Notify the callback after DNS resolution has completed.
      */
     void onDnsResolved();
+
+    /**
+     * Notify the callback when the SSL handshake performed to establish an HTTPS connection has been completed.
+     */
+    void onSslHandshakeCompleted();
 }

src/main/java/com/ning/http/client/providers/netty/request/NettyConnectListener.java

@@ -15,6 +15,7 @@
 
 import static com.ning.http.util.AsyncHttpProviderUtils.getBaseUrl;
 
+import com.ning.http.client.AsyncHandler;
 import org.jboss.netty.channel.Channel;
 import org.jboss.netty.channel.ChannelFuture;
 import org.jboss.netty.channel.ChannelFutureListener;
@@ -82,21 +83,26 @@ private void writeRequest(Channel channel) {
     }
 
     private void onFutureSuccess(final Channel channel) throws ConnectException {
-        
+
         SslHandler sslHandler = channel.getPipeline().get(SslHandler.class);
 
         if (sslHandler != null) {
             sslHandler.handshake().addListener(new ChannelFutureListener() {
-                
+
                 @Override
-                public void operationComplete(ChannelFuture future) throws Exception {
-                    if (future.isSuccess())
+                public void operationComplete(ChannelFuture handshakeFuture) throws Exception {
+                    if (handshakeFuture.isSuccess()){
+                        final AsyncHandler<T> asyncHandler = future.getAsyncHandler();
+                        if (asyncHandler instanceof AsyncHandlerExtensions)
+                            AsyncHandlerExtensions.class.cast(asyncHandler).onSslHandshakeCompleted();
+
                         writeRequest(channel);
+                    }
                     else
-                        onFutureFailure(channel, future.getCause());
+                        onFutureFailure(channel, handshakeFuture.getCause());
                 }
             });
-        
+
         } else {
             writeRequest(channel);
         }

src/test/java/com/ning/http/client/async/EventCollectingHandler.java

@@ -104,4 +104,9 @@ public void onRetry() {
     public void onDnsResolved() {
         firedEvents.add("DnsResolved");
     }
+
+    @Override
+    public void onSslHandshakeCompleted() {
+        firedEvents.add("SslHandshakeCompleted");
+    }
 }

src/test/java/com/ning/http/client/async/netty/NettyBasicHttpsTest.java

@@ -43,6 +43,7 @@ public void testNormalEventsFired() throws InterruptedException, TimeoutExceptio
                     "PoolConnection",
                     "OpenConnection",
                     "DnsResolved",
+                    "SslHandshakeCompleted",
                     "ConnectionOpen",
                     "SendRequest",
                     "HeaderWriteCompleted",