Perform post-handshake drain after setting connection state

This is to make sure that session live events will be delivered before
any incoming requests.

1 files changed, 1 insertions, 1 deletions

diff --git a/jrt/src/com/yahoo/jrt/Connection.java b/jrt/src/com/yahoo/jrt/Connection.java
index e85e5801284..76ef831850e 100644
--- a/jrt/src/com/yahoo/jrt/Connection.java
+++ b/jrt/src/com/yahoo/jrt/Connection.java
@@ -220,10 +220,10 @@ class Connection extends Target {
             if (socket.getMinimumReadBufferSize() > readSize) {
                 readSize = socket.getMinimumReadBufferSize();
             }
+            setState(CONNECTED);
             while (socket.drain(input.getChannelWritable(readSize)) > 0) {
                 handlePackets();
             }
-            setState(CONNECTED);
             break;
         case NEED_READ:
             enableRead();