Make WebSocketUpgradeHandler extensible in a controlled manner, backport #1446

Author: slandelle

client/src/main/java/org/asynchttpclient/ws/WebSocketUpgradeHandler.java

@@ -48,30 +48,32 @@ public void bufferFrame(Runnable bufferedFrame) {
         bufferedFrames.add(bufferedFrame);
     }
 
-    @Override
-    public final void onThrowable(Throwable t) {
-        onFailure(t);
-    }
+    protected void onStatusReceived0(HttpResponseStatus responseStatus) throws Exception {}
 
-    public boolean touchSuccess() {
-        boolean prev = onSuccessCalled;
-        onSuccessCalled = true;
-        return prev;
-    }
+    protected void onHeadersReceived0(HttpResponseHeaders headers) throws Exception {}
 
-    @Override
-    public final State onBodyPartReceived(HttpResponseBodyPart bodyPart) throws Exception {
-        return State.CONTINUE;
-    }
+    protected void onBodyPartReceived0(HttpResponseBodyPart bodyPart) throws Exception {}
+
+    protected void onCompleted0() throws Exception {}
+
+    protected void onThrowable0(Throwable t) {}
 
     @Override
     public final State onStatusReceived(HttpResponseStatus responseStatus) throws Exception {
+        onStatusReceived0(responseStatus);
         status = responseStatus.getStatusCode();
         return status == SWITCHING_PROTOCOLS ? State.UPGRADE : State.ABORT;
     }
 
     @Override
     public final State onHeadersReceived(HttpResponseHeaders headers) throws Exception {
+        onHeadersReceived0(headers);
+        return State.CONTINUE;
+    }
+
+    @Override
+    public final State onBodyPartReceived(HttpResponseBodyPart bodyPart) throws Exception {
+        onBodyPartReceived0(bodyPart);
         return State.CONTINUE;
     }
 
@@ -85,9 +87,22 @@ public final WebSocket onCompleted() throws Exception {
             throw e;
         }
 
+        onCompleted0();
         return webSocket;
     }
 
+    @Override
+    public final void onThrowable(Throwable t) {
+        onThrowable0(t);
+        onFailure(t);
+    }
+
+    public boolean touchSuccess() {
+        boolean prev = onSuccessCalled;
+        onSuccessCalled = true;
+        return prev;
+    }
+
     @Override
     public final void onSuccess(WebSocket webSocket) {
         this.webSocket = webSocket;