StakeSquid/ethereum-rpc-docker
1
0
Fork 0
Code Issues 1 Packages Projects Releases Wiki Activity

now with websockets

Browse Source
This commit is contained in:
Para Dox
2025-05-01 14:20:31 +07:00
parent 25502ae1a1
commit f2b642c20d
1 changed files with 26 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
benchmark-proxy/main.go
View File

@@ -492,12 +492,33 @@ func handleWebSocketRequest(w http.ResponseWriter, r *http.Request, backends []B
backendURL := strings.Replace(b.URL, "http://", "ws://", 1)
backendURL = strings.Replace(backendURL, "https://", "wss://", 1)
// Copy headers for the dialer
// Create a clean header map for the dialer
// Instead of copying all headers which can cause duplicates
header := http.Header{}
for name, values := range r.Header {
for _, value := range values {
header.Add(name, value)
}
// Only copy specific headers needed for the WebSocket connection
// and avoid the problematic ones like "Connection" and "Upgrade"
if host := r.Header.Get("Host"); host != "" {
header.Set("Host", host)
}
if origin := r.Header.Get("Origin"); origin != "" {
header.Set("Origin", origin)
}
if secWebSocketKey := r.Header.Get("Sec-WebSocket-Key"); secWebSocketKey != "" {
header.Set("Sec-WebSocket-Key", secWebSocketKey)
}
if secWebSocketVersion := r.Header.Get("Sec-WebSocket-Version"); secWebSocketVersion != "" {
header.Set("Sec-WebSocket-Version", secWebSocketVersion)
}
if secWebSocketProtocol := r.Header.Get("Sec-WebSocket-Protocol"); secWebSocketProtocol != "" {
header.Set("Sec-WebSocket-Protocol", secWebSocketProtocol)
}
if secWebSocketExtensions := r.Header.Get("Sec-WebSocket-Extensions"); secWebSocketExtensions != "" {
header.Set("Sec-WebSocket-Extensions", secWebSocketExtensions)
}
// Add user-agent if present
if userAgent := r.Header.Get("User-Agent"); userAgent != "" {
header.Set("User-Agent", userAgent)
}
startTime := time.Now()