diff --git a/internal/api/modules/amp/proxy.go b/internal/api/modules/amp/proxy.go index c460a0d60..b7d10760b 100644 --- a/internal/api/modules/amp/proxy.go +++ b/internal/api/modules/amp/proxy.go @@ -3,6 +3,8 @@ package amp import ( "bytes" "compress/gzip" + "context" + "errors" "fmt" "io" "net/http" @@ -188,6 +190,10 @@ func createReverseProxy(upstreamURL string, secretSource SecretSource) (*httputi // Error handler for proxy failures proxy.ErrorHandler = func(rw http.ResponseWriter, req *http.Request, err error) { + // Client-side cancellations are common during polling; suppress logging in this case + if errors.Is(err, context.Canceled) { + return + } log.Errorf("amp upstream proxy error for %s %s: %v", req.Method, req.URL.Path, err) rw.Header().Set("Content-Type", "application/json") rw.WriteHeader(http.StatusBadGateway) diff --git a/internal/api/modules/amp/proxy_test.go b/internal/api/modules/amp/proxy_test.go index ff23e3986..32f5d8605 100644 --- a/internal/api/modules/amp/proxy_test.go +++ b/internal/api/modules/amp/proxy_test.go @@ -493,6 +493,30 @@ func TestReverseProxy_ErrorHandler(t *testing.T) { } } +func TestReverseProxy_ErrorHandler_ContextCanceled(t *testing.T) { + // Test that context.Canceled errors return 499 without generic error response + proxy, err := createReverseProxy("http://example.com", NewStaticSecretSource("")) + if err != nil { + t.Fatal(err) + } + + // Create a canceled context to trigger the cancellation path + ctx, cancel := context.WithCancel(context.Background()) + cancel() // Cancel immediately + + req := httptest.NewRequest(http.MethodGet, "/test", nil).WithContext(ctx) + rr := httptest.NewRecorder() + + // Directly invoke the ErrorHandler with context.Canceled + proxy.ErrorHandler(rr, req, context.Canceled) + + // Body should be empty for canceled requests (no JSON error response) + body := rr.Body.Bytes() + if len(body) > 0 { + t.Fatalf("expected empty body for canceled context, got: %s", body) + } +} + func TestReverseProxy_FullRoundTrip_Gzip(t *testing.T) { // Upstream returns gzipped JSON without Content-Encoding header upstream := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {