Add filtering by health and state in list rules

Signed-off-by: Eunice Kim <[email protected]>

Author: euniceek

pkg/ruler/api.go

@@ -135,11 +135,25 @@ func (a *API) PrometheusRules(w http.ResponseWriter, req *http.Request) {
 		return
 	}
 
+	state := strings.ToLower(req.URL.Query().Get("state"))
+	if state != "" && state != firingStateFilter && state != pendingStateFilter && state != inactiveStateFilter {
+		util_api.RespondError(logger, w, v1.ErrBadData, fmt.Sprintf("unsupported state value %q", state), http.StatusBadRequest)
+		return
+	}
+
+	health := strings.ToLower(req.URL.Query().Get("health"))
+	if health != "" && health != unknownHealthFilter && health != okHealthFilter && health != errHealthFilter {
+		util_api.RespondError(logger, w, v1.ErrBadData, fmt.Sprintf("unsupported health value %q", health), http.StatusBadRequest)
+		return
+	}
+
 	rulesRequest := RulesRequest{
 		RuleNames:      req.Form["rule_name[]"],
 		RuleGroupNames: req.Form["rule_group[]"],
 		Files:          req.Form["file[]"],
 		Type:           typ,
+		State:          state,
+		Health:         health,
 	}
 
 	w.Header().Set("Content-Type", "application/json")

pkg/ruler/ruler.go

@@ -73,6 +73,14 @@ const (
 
 	alertingRuleFilter  string = "alert"
 	recordingRuleFilter string = "record"
+
+	firingStateFilter   string = "firing"
+	pendingStateFilter  string = "pending"
+	inactiveStateFilter string = "inactive"
+
+	unknownHealthFilter string = "unknown"
+	okHealthFilter      string = "ok"
+	errHealthFilter     string = "err"
 )
 
 type DisabledRuleGroupErr struct {
@@ -874,9 +882,11 @@ func (r *Ruler) getLocalRules(userID string, rulesRequest RulesRequest, includeB
 	ruleGroupNameSet := sliceToSet(rulesRequest.RuleGroupNames)
 	fileSet := sliceToSet(rulesRequest.Files)
 	ruleType := rulesRequest.Type
+	alertState := rulesRequest.State
+	health := rulesRequest.Health
 
 	returnAlerts := ruleType == "" || ruleType == alertingRuleFilter
-	returnRecording := ruleType == "" || ruleType == recordingRuleFilter
+	returnRecording := (ruleType == "" || ruleType == recordingRuleFilter) && alertState == ""
 
 	for _, group := range groups {
 		// The mapped filename is url path escaped encoded to make handling `/` characters easier
@@ -915,6 +925,9 @@ func (r *Ruler) getLocalRules(userID string, rulesRequest RulesRequest, includeB
 					continue
 				}
 			}
+			if !returnByHealth(health, string(r.Health())) {
+				continue
+			}
 			lastError := ""
 			if r.LastError() != nil {
 				lastError = r.LastError().Error()
@@ -926,6 +939,9 @@ func (r *Ruler) getLocalRules(userID string, rulesRequest RulesRequest, includeB
 				if !returnAlerts {
 					continue
 				}
+				if !returnByState(alertState, rule.State().String()) {
+					continue
+				}
 				alerts := []*AlertStateDesc{}
 				for _, a := range rule.ActiveAlerts() {
 					alerts = append(alerts, &AlertStateDesc{
@@ -1296,3 +1312,11 @@ func (r *Ruler) ListAllRules(w http.ResponseWriter, req *http.Request) {
 	close(iter)
 	<-done
 }
+
+func returnByState(requestState string, alertState string) bool {
+	return requestState == "" || requestState == alertState
+}
+
+func returnByHealth(requestHealth string, ruleHealth string) bool {
+	return requestHealth == "" || requestHealth == ruleHealth
+}

pkg/ruler/ruler.pb.go

@@ -24,6 +24,8 @@ message RulesRequest {
   repeated string ruleGroupNames = 2;
   repeated string files = 3;
   string type = 4;
+  string state = 5;
+  string health = 6;
 }
 
 message RulesResponse {