Fix benchmarks

cloudflare · Aug 27, 2024 · bfb010e · bfb010e
1 parent 7d0d534
commit bfb010e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/internal/config/config.go b/internal/config/config.go
@@ -83,7 +83,7 @@ func (cfg Config) String() string {
 func (cfg *Config) GetChecksForEntry(ctx context.Context, gen *PrometheusGenerator, entry discovery.Entry) []checks.RuleChecker {
 	enabled := []checks.RuleChecker{}
 
-	defaultMatch := []Match{{State: defaultMatchStates(ctx.Value(CommandKey).(ContextCommandVal))}}
+	defaultMatch := []Match{{State: defaultMatchStates(commandFromContext(ctx))}}
 	proms := gen.ServersForPath(entry.Path.Name)
 
 	if entry.PathError != nil || entry.Rule.Error.Err != nil {
@@ -229,3 +229,10 @@ func parseDuration(d string) (time.Duration, error) {
 	}
 	return time.Duration(mdur), nil
 }
+
+func commandFromContext(ctx context.Context) (cmd ContextCommandVal) {
+	if val := ctx.Value(CommandKey); val != nil {
+		cmd = val.(ContextCommandVal)
+	}
+	return cmd
+}
diff --git a/internal/config/match.go b/internal/config/match.go
@@ -105,7 +105,7 @@ func (m Match) validate(allowEmpty bool) error {
 }
 
 func (m Match) IsMatch(ctx context.Context, path string, e discovery.Entry) bool {
-	cmd := ctx.Value(CommandKey).(ContextCommandVal)
+	cmd := commandFromContext(ctx)
 
 	if m.Command != nil {
 		if cmd != *m.Command {