Function New

modules/generator/registry/registry.go:126–170

New creates a ManagedRegistry. This Registry will scrape itself, write samples into an appender and remove stale series.

(cfg *Config, overrides Overrides, tenant string, appendable storage.Appendable, logger log.Logger, limiter Limiter)

// New creates a ManagedRegistry. This Registry will scrape itself, write samples into an appender
// and remove stale series.
func New(cfg *Config, overrides Overrides, tenant string, appendable storage.Appendable, logger log.Logger, limiter Limiter) *ManagedRegistry {
	instanceCtx, cancel := context.WithCancel(context.Background())

	externalLabels := make(map[string]string)
	for k, v := range cfg.ExternalLabels {
		externalLabels[k] = v
	}
	hostname, _ := os.Hostname()
	externalLabels["__metrics_gen_instance"] = hostname

	if cfg.InjectTenantIDAs != "" {
		externalLabels[cfg.InjectTenantIDAs] = tenant
	}

	drainSanitizer := NewDrainSanitizer(tenant, overrides.MetricsGeneratorSpanNameSanitization, cfg.StaleDuration)
	perLabelLimiter := NewPerLabelLimiter(tenant, overrides.MetricsGeneratorMaxCardinalityPerLabel, cfg.StaleDuration)

	r := &ManagedRegistry{
		onShutdown: cancel,

		cfg:            cfg,
		overrides:      overrides,
		tenant:         tenant,
		externalLabels: externalLabels,

		metrics: map[string]metric{},

		appendable:      appendable,
		sanitizer:       drainSanitizer,
		perLabelLimiter: perLabelLimiter,
		limiter:         limiter,
		entityDemand:    NewCardinality(cfg.StaleDuration, removeStaleSeriesInterval),

		logger:                 logger,
		limitLogger:            tempo_log.NewRateLimitedLogger(1, level.Warn(logger)),
		metricEntityDemand:     metricEntityDemand.WithLabelValues(tenant),
		metricSeriesDemand:     metricSeriesDemand.WithLabelValues(tenant),
		metricTotalCollections: metricTotalCollections.WithLabelValues(tenant),
	}

	go job(instanceCtx, r.CollectMetrics, r.collectionInterval)
	go job(instanceCtx, r.removeStaleSeries, constantInterval(removeStaleSeriesInterval))

	return r
}

func (r *ManagedRegistry) NewLabelBuilder() LabelBuilder {
	return NewLabelBuilder(r.cfg.MaxLabelNameLength, r.cfg.MaxLabelValueLength, r.sanitizer, r.perLabelLimiter)