Fixes cloudflare#1273

zemaj · Jun 13, 2024 · f76b872 · f76b872
1 parent a16532d
commit f76b872
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/proxy/proxy.go b/proxy/proxy.go
@@ -199,6 +199,16 @@ func (p *Proxy) proxyHTTPRequest(
 		roundTripReq.Header.Set("Connection", "keep-alive")
 	}
 
+    // Handle GOAWAY frame to correctly retry a request
+    if roundTripReq.Body != nil {
+        roundTripReq.GetBody = func() (io.ReadCloser, err error) {
+            if err.Error() == "http2: Transport received Server's graceful shutdown GOAWAY" {
+                return roundTripReq.Body, nil
+            }
+            return nil, err
+        }
+    }
+
 	// Set the User-Agent as an empty string if not provided to avoid inserting golang default UA
 	if roundTripReq.Header.Get("User-Agent") == "" {
 		roundTripReq.Header.Set("User-Agent", "")