From 68ed467231839e7b1ace8b801452970ffd05f118 Mon Sep 17 00:00:00 2001
From: Darren Shepherd <darren@rancher.com>
Date: Fri, 28 Feb 2020 09:30:25 -0700
Subject: [PATCH] Ensure redirector supports Hijacking if possible

---
 pkg/schemaserver/urlbuilder/redirect.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/pkg/schemaserver/urlbuilder/redirect.go b/pkg/schemaserver/urlbuilder/redirect.go
index 6ab3098..d4a765c 100644
--- a/pkg/schemaserver/urlbuilder/redirect.go
+++ b/pkg/schemaserver/urlbuilder/redirect.go
@@ -18,6 +18,9 @@ func RedirectRewrite(next http.Handler) http.Handler {
 			ResponseWriter: rw,
 			prefix:         prefix,
 		}
+		if h, ok := rw.(http.Hijacker); ok {
+			r.Hijacker = h
+		}
 		next.ServeHTTP(r, req)
 		r.Close()
 	})
@@ -25,6 +28,7 @@ func RedirectRewrite(next http.Handler) http.Handler {
 
 type redirector struct {
 	http.ResponseWriter
+	http.Hijacker
 	prefix     string
 	from, to   string
 	tempBuffer *bytes.Buffer