feat(querier): propagate all known alerting headers (#92873)

pkg/registry/apis/datasource/sub_query.go

@@ -4,14 +4,14 @@ import (
 	"context"
 	"fmt"
 	"net/http"
-	"strings"
 
 	"github.com/grafana/grafana-plugin-sdk-go/backend"
 	data "github.com/grafana/grafana-plugin-sdk-go/experimental/apis/data/v0alpha1"
+	query "github.com/grafana/grafana/pkg/apis/query/v0alpha1"
+	query_headers "github.com/grafana/grafana/pkg/registry/apis/query"
 	"k8s.io/apimachinery/pkg/runtime"
 	"k8s.io/apiserver/pkg/registry/rest"
 
-	query "github.com/grafana/grafana/pkg/apis/query/v0alpha1"
 	"github.com/grafana/grafana/pkg/web"
 )
 
@@ -74,29 +74,10 @@ func (r *subQueryREST) Connect(ctx context.Context, name string, opts runtime.Ob
 		ctx = backend.WithGrafanaConfig(ctx, pluginCtx.GrafanaConfig)
 		ctx = contextualMiddlewares(ctx)
 
-		// only forward expected headers, log unexpected ones
-		headers := make(map[string]string)
-		// headers are case insensitive, however some datasources still check for camel casing so we have to send them camel cased
-		expectedHeaders := map[string]string{
-			"fromalert":      "FromAlert",
-			"content-type":   "Content-Type",
-			"content-length": "Content-Length",
-			"user-agent":     "User-Agent",
-			"accept":         "Accept",
-		}
-		for k, v := range req.Header {
-			headerToSend, ok := expectedHeaders[strings.ToLower(k)]
-			if ok {
-				headers[headerToSend] = v[0]
-			} else {
-				r.builder.log.Warn("datasource received an unexpected header, ignoring it", "header", k)
-			}
-		}
-
 		rsp, err := r.builder.client.QueryData(ctx, &backend.QueryDataRequest{
 			Queries:       queries,
 			PluginContext: pluginCtx,
-			Headers:       headers,
+			Headers:       query_headers.ExtractKnownHeaders(req.Header),
 		})
 		if err != nil {
 			responder.Error(err)

pkg/registry/apis/datasource/sub_query_test.go

@@ -10,6 +10,7 @@ import (
 	"github.com/grafana/grafana-plugin-sdk-go/backend"
 	"github.com/grafana/grafana/pkg/apis/datasource/v0alpha1"
 	"github.com/grafana/grafana/pkg/infra/log"
+	"github.com/grafana/grafana/pkg/services/ngalert/models"
 	"github.com/stretchr/testify/require"
 	"k8s.io/apimachinery/pkg/runtime"
 )
@@ -32,14 +33,24 @@ func TestSubQueryConnect(t *testing.T) {
 
 	rr := httptest.NewRecorder()
 	req := httptest.NewRequest(http.MethodGet, "/some-path", nil)
-	req.Header.Set("fromAlert", "true")
+	req.Header.Set(models.FromAlertHeaderName, "true")
+	req.Header.Set(models.CacheSkipHeaderName, "true")
+	req.Header.Set("X-Rule-Uid", "abc")
+	req.Header.Set("X-Rule-Folder", "folder-1")
+	req.Header.Set("X-Rule-Source", "grafana-ruler")
+	req.Header.Set("X-Grafana-Org-Id", "1")
 	req.Header.Set("Content-Type", "application/json")
+	req.Header.Set("some-unexpected-header", "some-value")
 	handler.ServeHTTP(rr, req)
 
 	// test that headers are forwarded and cased appropriately
 	require.Equal(t, map[string]string{
-		"FromAlert":    "true",
-		"Content-Type": "application/json",
+		models.FromAlertHeaderName: "true",
+		models.CacheSkipHeaderName: "true",
+		"X-Rule-Uid":               "abc",
+		"X-Rule-Folder":            "folder-1",
+		"X-Rule-Source":            "grafana-ruler",
+		"X-Grafana-Org-Id":         "1",
 	}, *sqr.builder.client.(mockClient).lastCalledWithHeaders)
 }