Skip to content

Commit 843e2da

Browse files
grobinson-grafanaperiklis
authored andcommitted
chore: small clean up of frontend.New (#16822)
1 parent a10a531 commit 843e2da

File tree

1 file changed

+14
-8
lines changed

1 file changed

+14
-8
lines changed

pkg/limits/frontend/frontend.go

+14-8
Original file line numberDiff line numberDiff line change
@@ -77,12 +77,19 @@ type Frontend struct {
7777

7878
// New returns a new Frontend.
7979
func New(cfg Config, ringName string, limitsRing ring.ReadRing, limits Limits, logger log.Logger, reg prometheus.Registerer) (*Frontend, error) {
80-
var servs []services.Service
80+
// Set up a client pool for the limits service. The frontend will use this
81+
// to make RPCs that get the current stream usage to checks per-tenant limits.
82+
clientPoolFactory := limits_client.NewPoolFactory(cfg.ClientConfig)
83+
clientPool := limits_client.NewPool(
84+
ringName,
85+
cfg.ClientConfig.PoolConfig,
86+
limitsRing,
87+
clientPoolFactory,
88+
logger,
89+
)
8190

82-
factory := limits_client.NewPoolFactory(cfg.ClientConfig)
83-
pool := limits_client.NewPool(ringName, cfg.ClientConfig.PoolConfig, limitsRing, factory, logger)
8491
rateLimiter := limiter.NewRateLimiter(newRateLimitsAdapter(limits), cfg.RecheckPeriod)
85-
streamUsage := NewRingStreamUsageGatherer(limitsRing, pool, logger)
92+
streamUsage := NewRingStreamUsageGatherer(limitsRing, clientPool, logger)
8693

8794
f := &Frontend{
8895
cfg: cfg,
@@ -93,17 +100,16 @@ func New(cfg Config, ringName string, limitsRing ring.ReadRing, limits Limits, l
93100
metrics: newMetrics(reg),
94101
}
95102

96-
var err error
97-
f.lifecycler, err = ring.NewLifecycler(cfg.LifecyclerConfig, f, RingName, RingKey, true, logger, reg)
103+
lifecycler, err := ring.NewLifecycler(cfg.LifecyclerConfig, f, RingName, RingKey, true, logger, reg)
98104
if err != nil {
99105
return nil, fmt.Errorf("failed to create %s lifecycler: %w", RingName, err)
100106
}
107+
f.lifecycler = lifecycler
101108
// Watch the lifecycler
102109
f.lifecyclerWatcher = services.NewFailureWatcher()
103110
f.lifecyclerWatcher.WatchService(f.lifecycler)
104111

105-
servs = append(servs, f.lifecycler)
106-
servs = append(servs, pool)
112+
servs := []services.Service{lifecycler, clientPool}
107113
mgr, err := services.NewManager(servs...)
108114
if err != nil {
109115
return nil, err

0 commit comments

Comments
 (0)