Method GetPrometheusRules

integration/e2ecortex/client.go:729–783

GetPrometheusRules fetches the rules from the Prometheus endpoint /api/v1/rules.

(filter RuleFilter)

// GetPrometheusRules fetches the rules from the Prometheus endpoint /api/v1/rules.
func (c *Client) GetPrometheusRules(filter RuleFilter) ([]*ruler.RuleGroup, string, error) {
	// Create HTTP request

	req, err := http.NewRequest("GET", fmt.Sprintf("http://%s/api/prom/api/v1/rules", c.rulerAddress), nil)
	if err != nil {
		return nil, "", err
	}
	req.Header.Set("X-Scope-OrgID", c.orgID)

	urlValues := req.URL.Query()
	addQueryParams(urlValues, "file[]", filter.Namespaces...)
	addQueryParams(urlValues, "rule_name[]", filter.RuleNames...)
	addQueryParams(urlValues, "rule_group[]", filter.RuleGroupNames...)
	addQueryParams(urlValues, "type", filter.RuleType)
	addQueryParams(urlValues, "exclude_alerts", filter.ExcludeAlerts)
	if filter.MaxRuleGroup > 0 {
		addQueryParams(urlValues, "group_limit", strconv.Itoa(filter.MaxRuleGroup))
	}
	if filter.NextToken != "" {
		addQueryParams(urlValues, "group_next_token", filter.NextToken)
	}
	req.URL.RawQuery = urlValues.Encode()

	ctx, cancel := context.WithTimeout(context.Background(), c.timeout)
	defer cancel()

	// Execute HTTP request
	res, err := c.httpClient.Do(req.WithContext(ctx))
	if err != nil {
		return nil, "", err
	}
	defer res.Body.Close()

	body, err := io.ReadAll(res.Body)
	if err != nil {
		return nil, "", err
	}

	// Decode the response.
	type response struct {
		Status string              `json:"status"`
		Data   ruler.RuleDiscovery `json:"data"`
	}

	decoded := &response{}
	if err := json.Unmarshal(body, decoded); err != nil {
		return nil, "", err
	}

	if decoded.Status != "success" {
		return nil, "", fmt.Errorf("unexpected response status '%s'", decoded.Status)
	}

	return decoded.Data.RuleGroups, decoded.Data.GroupNextToken, nil
}

// GetRuleGroups gets the configured rule groups from the ruler.
func (c *Client) GetRuleGroups() (map[string][]rulefmt.RuleGroup, error) {