Add query priority for all operations

Signed-off-by: Essam Eldaly <[email protected]>

Author: eeldaly

pkg/querier/tripperware/query_attribute_matcher.go

@@ -22,6 +22,7 @@ func rejectQueryOrSetPriority(r *http.Request, now time.Time, lookbackDelta time
 		return nil
 	}
 	op := getOperation(r)
+	reqStats := stats.FromContext(r.Context())
 
 	if op == "query" || op == "query_range" {
 		query := r.FormValue("query")
@@ -40,7 +41,6 @@ func rejectQueryOrSetPriority(r *http.Request, now time.Time, lookbackDelta time
 			}
 		}
 
-		reqStats := stats.FromContext(r.Context())
 		reqStats.SetDataSelectMaxTime(maxTime)
 		reqStats.SetDataSelectMinTime(minTime)
 
@@ -55,6 +55,16 @@ func rejectQueryOrSetPriority(r *http.Request, now time.Time, lookbackDelta time
 			}
 			reqStats.SetPriority(queryPriority.DefaultPriority)
 		}
+	} else {
+		if queryPriority := limits.QueryPriority(userStr); queryPriority.Enabled && len(queryPriority.Priorities) != 0 {
+			for _, priority := range queryPriority.Priorities {
+				for _, attribute := range priority.QueryAttributes {
+					if matchAttributeForMetadataQuery(attribute, op, r, now) {
+						reqStats.SetPriority(priority.Priority)
+					}
+				}
+			}
+		}
 	}
 
 	if queryReject := limits.QueryRejection(userStr); queryReject.Enabled && (op == "series" || op == "labels" || op == "label_values") {
@@ -81,6 +91,8 @@ func getOperation(r *http.Request) string {
 		return "labels"
 	case strings.HasSuffix(r.URL.Path, "/values"):
 		return "label_values"
+	case strings.HasSuffix(r.URL.Path, "/metadata"):
+		return "metadata"
 	default:
 		return "other"
 	}