Add proto annotation

c2FmZQ · Nov 30, 2024 · deabc0f · deabc0f
1 parent 30083c0
commit deabc0f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/proxy/websocket.go b/proxy/websocket.go
@@ -63,8 +63,8 @@ func (p *Proxy) webSocketHandler(cfg WebSocketConfig) http.Handler {
 		wc.SetAnnotation(startTimeKey, time.Now())
 		if conn, ok := req.Context().Value(connCtxKey).(anyConn); ok {
 			annotatedConn(conn).SetAnnotation(httpUpgradeKey, "websocket")
-			wc.SetAnnotation(httpUpgradeKey, "websocket")
 			wc.SetAnnotation(serverNameKey, connServerName(conn))
+			wc.SetAnnotation(protoKey, "websocket->tcp")
 			wc.SetAnnotation(modeKey, connMode(conn))
 		}
 		p.outConns.add(wc)