Fixed to support websocket communication

Author: Thanus-Kumaar

controllers/kernel_controller.go

@@ -171,6 +171,7 @@ func (c *KernelController) KernelChannelsHandler(w http.ResponseWriter, r *http.
 	headers := http.Header{}
 	headers.Set("Authorization", "token "+gatewayToken)
 
+	c.Logger.Debug().Str("targetURL", targetURL).Interface("headers", headers).Msg("Attempting to dial kernel gateway")
 	// Connect to the Jupyter Kernel Gateway
 	kgConn, _, err := websocket.DefaultDialer.Dial(targetURL, headers)
 	if err != nil {

pkg/middleware/log_middleware.go

@@ -1,6 +1,9 @@
 package middleware
 
 import (
+	"bufio"
+	"fmt"
+	"net"
 	"net/http"
 	"time"
 
@@ -18,6 +21,14 @@ func (r *statusRecorder) WriteHeader(code int) {
 	r.ResponseWriter.WriteHeader(code)
 }
 
+func (r *statusRecorder) Hijack() (net.Conn, *bufio.ReadWriter, error) {
+	h, ok := r.ResponseWriter.(http.Hijacker)
+	if !ok {
+		return nil, nil, fmt.Errorf("hijack not supported")
+	}
+	return h.Hijack()
+}
+
 func RequestLogger(next http.Handler) http.Handler {
 	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {