Support websocket subprotocl. (#1092)

* Support websocket subprotocl. * use HTTPHeader()
easegress-io · Sep 27, 2023 · 9ad9310 · 9ad9310
1 parent 0722e3b
commit 9ad9310
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/filters/proxies/httpproxy/wspool.go b/pkg/filters/proxies/httpproxy/wspool.go
@@ -189,6 +189,10 @@ func (sp *WebSocketServerPool) handle(ctx *context.Context) (result string) {
  InsecureSkipVerify: sp.spec.InsecureSkipVerify,
  OriginPatterns: sp.spec.OriginPatterns,
  }
+ subProtocol := req.HTTPHeader().Get("Sec-WebSocket-Protocol")
+ if subProtocol != "" {
+ opts.Subprotocols = []string{subProtocol}
+ }
  clntConn, err := websocket.Accept(stdw, req.Std(), opts)
  if err != nil {
  logger.Errorf("%s: failed to establish client connection: %v", sp.Name, err)