Skip to content

Commit

Permalink
move "memberlist" to a const
Browse files Browse the repository at this point in the history
Signed-off-by: Trevor Whitney <trevorjwhitney@gmail.com>
  • Loading branch information
trevorwhitney committed Oct 7, 2021
1 parent e424763 commit 0c71dba
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 10 deletions.
8 changes: 5 additions & 3 deletions pkg/loki/config_wrapper.go
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,8 @@ func (c *ConfigWrapper) Clone() flagext.Registerer {
}(*c)
}

const memberlistStr = "memberlist"

// ApplyDynamicConfig satisfies WithCommonCloneable interface, and applies all rules for setting Loki
// config values from the common section of the Loki config file.
// This method's purpose is to simplify Loki's config in an opinionated way so that Loki can be run
Expand Down Expand Up @@ -70,9 +72,9 @@ func (c *ConfigWrapper) ApplyDynamicConfig() cfg.Source {
}

if len(r.MemberlistKV.JoinMembers) > 0 {
r.Ingester.LifecyclerConfig.RingConfig.KVStore.Store = "memberlist"
r.Distributor.DistributorRing.KVStore.Store = "memberlist"
r.Ruler.Ring.KVStore.Store = "memberlist"
r.Ingester.LifecyclerConfig.RingConfig.KVStore.Store = memberlistStr
r.Distributor.DistributorRing.KVStore.Store = memberlistStr
r.Ruler.Ring.KVStore.Store = memberlistStr
}

return nil
Expand Down
14 changes: 7 additions & 7 deletions pkg/loki/config_wrapper_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -108,9 +108,9 @@ memberlist:

config, _ := testContext(configFileString, nil)

assert.EqualValues(t, "memberlist", config.Ingester.LifecyclerConfig.RingConfig.KVStore.Store)
assert.EqualValues(t, "memberlist", config.Distributor.DistributorRing.KVStore.Store)
assert.EqualValues(t, "memberlist", config.Ruler.Ring.KVStore.Store)
assert.EqualValues(t, memberlistStr, config.Ingester.LifecyclerConfig.RingConfig.KVStore.Store)
assert.EqualValues(t, memberlistStr, config.Distributor.DistributorRing.KVStore.Store)
assert.EqualValues(t, memberlistStr, config.Ruler.Ring.KVStore.Store)
})

t.Run("explicit ring configs provided via config file are preserved", func(t *testing.T) {
Expand All @@ -127,8 +127,8 @@ distributor:

assert.EqualValues(t, "etcd", config.Distributor.DistributorRing.KVStore.Store)

assert.EqualValues(t, "memberlist", config.Ingester.LifecyclerConfig.RingConfig.KVStore.Store)
assert.EqualValues(t, "memberlist", config.Ruler.Ring.KVStore.Store)
assert.EqualValues(t, memberlistStr, config.Ingester.LifecyclerConfig.RingConfig.KVStore.Store)
assert.EqualValues(t, memberlistStr, config.Ruler.Ring.KVStore.Store)
})

t.Run("explicit ring configs provided via command line are preserved", func(t *testing.T) {
Expand All @@ -141,8 +141,8 @@ memberlist:

assert.EqualValues(t, "inmemory", config.Ruler.Ring.KVStore.Store)

assert.EqualValues(t, "memberlist", config.Ingester.LifecyclerConfig.RingConfig.KVStore.Store)
assert.EqualValues(t, "memberlist", config.Distributor.DistributorRing.KVStore.Store)
assert.EqualValues(t, memberlistStr, config.Ingester.LifecyclerConfig.RingConfig.KVStore.Store)
assert.EqualValues(t, memberlistStr, config.Distributor.DistributorRing.KVStore.Store)
})
})
}

0 comments on commit 0c71dba

Please sign in to comment.