Skip to content

Commit 5dac145

Browse files
committed
fix per PR comments
Signed-off-by: Jacob Lisi <[email protected]>
1 parent 10b4f84 commit 5dac145

File tree

3 files changed

+14
-9
lines changed

3 files changed

+14
-9
lines changed

pkg/ruler/api.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -129,14 +129,14 @@ func respondError(logger log.Logger, w http.ResponseWriter, msg string) {
129129
func (r *Ruler) rules(w http.ResponseWriter, req *http.Request) {
130130
logger := util.WithContext(req.Context(), util.Logger)
131131
userID, ctx, err := user.ExtractOrgIDFromHTTPRequest(req)
132-
if err != nil {
132+
if err != nil || userID == "" {
133133
level.Error(logger).Log("msg", "error extracting org id from context", "err", err)
134134
respondError(logger, w, "no valid org id found")
135135
return
136136
}
137137

138138
w.Header().Set("Content-Type", "application/json")
139-
rgs, err := r.GetRules(ctx, userID)
139+
rgs, err := r.GetRules(ctx)
140140

141141
if err != nil {
142142
respondError(logger, w, err.Error())
@@ -221,14 +221,14 @@ func (r *Ruler) rules(w http.ResponseWriter, req *http.Request) {
221221
func (r *Ruler) alerts(w http.ResponseWriter, req *http.Request) {
222222
logger := util.WithContext(req.Context(), util.Logger)
223223
userID, ctx, err := user.ExtractOrgIDFromHTTPRequest(req)
224-
if err != nil {
224+
if err != nil || userID == "" {
225225
level.Error(logger).Log("msg", "error extracting org id from context", "err", err)
226226
respondError(logger, w, "no valid org id found")
227227
return
228228
}
229229

230230
w.Header().Set("Content-Type", "application/json")
231-
rgs, err := r.GetRules(ctx, userID)
231+
rgs, err := r.GetRules(ctx)
232232

233233
if err != nil {
234234
respondError(logger, w, err.Error())

pkg/ruler/ruler.go

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -472,9 +472,14 @@ func (r *Ruler) newManager(ctx context.Context, userID string) (*promRules.Manag
472472

473473
// GetRules retrieves the running rules from this ruler and all running rulers in the ring if
474474
// sharding is enabled
475-
func (r *Ruler) GetRules(ctx context.Context, userID string) ([]*GroupStateDesc, error) {
475+
func (r *Ruler) GetRules(ctx context.Context) ([]*GroupStateDesc, error) {
476+
userID, err := user.ExtractOrgID(ctx)
477+
if err != nil {
478+
return nil, fmt.Errorf("no user id found in context")
479+
}
480+
476481
if r.cfg.EnableSharding {
477-
return r.getShardedRules(ctx, userID)
482+
return r.getShardedRules(ctx)
478483
}
479484

480485
return r.getLocalRules(userID)
@@ -564,7 +569,7 @@ func (r *Ruler) getLocalRules(userID string) ([]*GroupStateDesc, error) {
564569
return groupDescs, nil
565570
}
566571

567-
func (r *Ruler) getShardedRules(ctx context.Context, userID string) ([]*GroupStateDesc, error) {
572+
func (r *Ruler) getShardedRules(ctx context.Context) ([]*GroupStateDesc, error) {
568573
rulers, err := r.ring.GetAll()
569574
if err != nil {
570575
return nil, err

pkg/ruler/rules/rules.proto

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ option (gogoproto.unmarshaler_all) = true;
1414

1515
// RuleGroupDesc is a proto representation of a cortex rule group
1616
message RuleGroupDesc {
17-
reserved 5;
17+
reserved 5, 7, 8;
1818
string name = 1;
1919
string namespace = 2;
2020
google.protobuf.Duration interval = 3
@@ -25,7 +25,7 @@ message RuleGroupDesc {
2525

2626
// RuleDesc is a proto representation of a Prometheus Rule
2727
message RuleDesc {
28-
reserved 7 to 10;
28+
reserved 7 to 12;
2929
string expr = 1;
3030
string record = 2;
3131
string alert = 3;

0 commit comments

Comments
 (0)