Function New

tempodb/tempodb.go:174–241

New creates a new tempodb

(cfg *Config, cacheProvider cache.Provider, logger gkLog.Logger)

// New creates a new tempodb
func New(cfg *Config, cacheProvider cache.Provider, logger gkLog.Logger) (Reader, Writer, Compactor, error) {
	var rawR backend.RawReader
	var rawW backend.RawWriter
	var c backend.Compactor

	err := validateConfig(cfg)
	if err != nil {
		return nil, nil, nil, fmt.Errorf("invalid config while creating tempodb: %w", err)
	}

	switch cfg.Backend {
	case backend.Local:
		rawR, rawW, c, err = local.New(cfg.Local)
	case backend.GCS:
		rawR, rawW, c, err = gcs.New(cfg.GCS)
	case backend.S3:
		rawR, rawW, c, err = s3.New(cfg.S3)
	case backend.Azure:
		rawR, rawW, c, err = azure.New(cfg.Azure)
	default:
		err = fmt.Errorf("unknown backend %s", cfg.Backend)
	}

	if err != nil {
		return nil, nil, nil, err
	}

	// build a caching layer if we have a provider
	if cacheProvider != nil {
		legacyCache, roles, err := createLegacyCache(cfg, logger)
		if err != nil {
			return nil, nil, nil, err
		}

		// inject legacy cache into the cache provider for the roles
		for _, role := range roles {
			err = cacheProvider.AddCache(role, legacyCache)
			if err != nil {
				return nil, nil, nil, fmt.Errorf("error adding legacy cache to provider: %w", err)
			}
		}

		rawR, rawW, err = backend_cache.NewCache(&cfg.BloomCacheCfg, rawR, rawW, cacheProvider, logger)
		if err != nil {
			return nil, nil, nil, err
		}
	}

	r := backend.NewReader(rawR)
	w := backend.NewWriter(rawW)
	rw := &readerWriter{
		c:                       c,
		r:                       r,
		w:                       w,
		cfg:                     cfg,
		logger:                  logger,
		pool:                    pool.NewPool(cfg.Pool),
		blocklist:               blocklist.New(),