Skip to content

Commit

Permalink
rules.NewGroup: Fix when no logger is passed (#15356)
Browse files Browse the repository at this point in the history
Signed-off-by: Arve Knudsen <[email protected]>
  • Loading branch information
aknuds1 authored Nov 21, 2024
1 parent 125a908 commit c2e28f2
Show file tree
Hide file tree
Showing 2 changed files with 20 additions and 7 deletions.
18 changes: 11 additions & 7 deletions rules/group.go
Original file line number Diff line number Diff line change
Expand Up @@ -99,9 +99,13 @@ type GroupOptions struct {

// NewGroup makes a new Group with the given name, options, and rules.
func NewGroup(o GroupOptions) *Group {
metrics := o.Opts.Metrics
opts := o.Opts
if opts == nil {
opts = &ManagerOptions{}
}
metrics := opts.Metrics
if metrics == nil {
metrics = NewGroupMetrics(o.Opts.Registerer)
metrics = NewGroupMetrics(opts.Registerer)
}

key := GroupKey(o.File, o.Name)
Expand All @@ -120,13 +124,13 @@ func NewGroup(o GroupOptions) *Group {
evalIterationFunc = DefaultEvalIterationFunc
}

concurrencyController := o.Opts.RuleConcurrencyController
concurrencyController := opts.RuleConcurrencyController
if concurrencyController == nil {
concurrencyController = sequentialRuleEvalController{}
}

if o.Opts.Logger == nil {
promslog.NewNopLogger()
if opts.Logger == nil {
opts.Logger = promslog.NewNopLogger()
}

return &Group{
Expand All @@ -137,12 +141,12 @@ func NewGroup(o GroupOptions) *Group {
limit: o.Limit,
rules: o.Rules,
shouldRestore: o.ShouldRestore,
opts: o.Opts,
opts: opts,
seriesInPreviousEval: make([]map[string]labels.Labels, len(o.Rules)),
done: make(chan struct{}),
managerDone: o.done,
terminated: make(chan struct{}),
logger: o.Opts.Logger.With("file", o.File, "group", o.Name),
logger: opts.Logger.With("file", o.File, "group", o.Name),
metrics: metrics,
evalIterationFunc: evalIterationFunc,
concurrencyController: concurrencyController,
Expand Down
9 changes: 9 additions & 0 deletions rules/group_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,9 +17,18 @@ import (
"testing"
"time"

"github.com/prometheus/common/promslog"
"github.com/stretchr/testify/require"
)

func TestNewGroup(t *testing.T) {
g := NewGroup(GroupOptions{
File: "test-file",
Name: "test-name",
})
require.Equal(t, promslog.NewNopLogger().With("file", "test-file", "group", "test-name"), g.logger)
}

func TestGroup_Equals(t *testing.T) {
tests := map[string]struct {
first *Group
Expand Down

0 comments on commit c2e28f2

Please sign in to comment.