Add websocket handshake request header to metadata

minekube · Mar 3, 2022 · 03d336f · 03d336f
1 parent 0207f68
commit 03d336f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ws/server.go b/ws/server.go
@@ -148,5 +148,9 @@ func pingLoop(ctx context.Context, d time.Duration, ws *websocket.Conn) {
 type httpRequestContextKey struct{}
 
 func withRequest(ctx context.Context, r *http.Request) context.Context {
+	// Add websocket handshake request header to metadata
+	md, _ := metadata.FromIncomingContext(ctx)
+	ctx = metadata.NewIncomingContext(ctx, metadata.Join(md, metadata.MD(r.Header)))
+
 	return context.WithValue(ctx, httpRequestContextKey{}, r)
 }