log grafana headers along with query request before query execution

Signed-off-by: Ben Ye <[email protected]>

Author: yeya24

pkg/frontend/transport/handler.go

@@ -228,13 +228,7 @@ func (f *Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	// Log request
 	if f.cfg.QueryStatsEnabled {
 		queryString = f.parseRequestQueryString(r, buf)
-		logMessage := append([]interface{}{
-			"msg", "query request",
-			"component", "query-frontend",
-			"method", r.Method,
-			"path", r.URL.Path,
-		}, formatQueryString(queryString)...)
-		level.Info(util_log.WithContext(r.Context(), f.log)).Log(logMessage...)
+		f.logQueryRequest(r, queryString)
 	}
 
 	startTime := time.Now()
@@ -306,6 +300,23 @@ func formatGrafanaStatsFields(r *http.Request) []interface{} {
 	return fields
 }
 
+// logQueryRequest logs query request before query execution.
+func (f *Handler) logQueryRequest(r *http.Request, queryString url.Values) {
+	logMessage := []interface{}{
+		"msg", "query request",
+		"component", "query-frontend",
+		"method", r.Method,
+		"path", r.URL.Path,
+	}
+	grafanaFields := formatGrafanaStatsFields(r)
+	if len(grafanaFields) > 0 {
+		logMessage = append(logMessage, grafanaFields...)
+	}
+	logMessage = append(logMessage, formatQueryString(queryString)...)
+
+	level.Info(util_log.WithContext(r.Context(), f.log)).Log(logMessage...)
+}
+
 // reportSlowQuery reports slow queries.
 func (f *Handler) reportSlowQuery(r *http.Request, queryString url.Values, queryResponseTime time.Duration) {
 	logMessage := []interface{}{