Support forwarding of websockets

alexellis · alexellis · commit 3ec7549caaf7 · 2025-02-27T08:43:21.000Z
Signed-off-by: Alex Ellis (OpenFaaS Ltd) &lt;alexellis2@gmail.com&gt;
diff --git a/executor/http_runner.go b/executor/http_runner.go
@@ -132,7 +132,8 @@ func (f *HTTPFunctionRunner) Run(req FunctionRequest, contentLength int64, r *ht
 	}
 	defer cancel()
 
-	if strings.HasPrefix(r.Header.Get("Accept"), "text/event-stream") {
+	if strings.HasPrefix(r.Header.Get("Accept"), "text/event-stream") ||
+		r.Header.Get("Upgrade") == "websocket" {
 		ww := fhttputil.NewHttpWriteInterceptor(w)
 
 		f.ReverseProxy.ServeHTTP(w, r)

Original file line number	Diff line number	Diff line change
`@@ -132,7 +132,8 @@ func (f HTTPFunctionRunner) Run(req FunctionRequest, contentLength int64, r ht`
`132`	`132`	`}`
`133`	`133`	`defer cancel()`
`134`	`134`
`135`		`- if strings.HasPrefix(r.Header.Get("Accept"), "text/event-stream") {`
	`135`	`+ if strings.HasPrefix(r.Header.Get("Accept"), "text/event-stream") \|\|`
	`136`	`+ r.Header.Get("Upgrade") == "websocket" {`
`136`	`137`	`ww := fhttputil.NewHttpWriteInterceptor(w)`
`137`	`138`
`138`	`139`	`f.ReverseProxy.ServeHTTP(w, r)`