loki/pkg/runtime/config.go

package runtime

type Config struct {
	LogStreamCreation     bool `yaml:"log_stream_creation"`
	LogPushRequest        bool `yaml:"log_push_request"`
	LogPushRequestStreams bool `yaml:"log_push_request_streams"`
}

// TenantConfig is a function that returns configs for given tenant, or
// nil, if there are no tenant-specific configs.
type TenantConfig func(userID string) *Config

// TenantConfigs periodically fetch a set of per-user configs, and provides convenience
// functions for fetching the correct value.
type TenantConfigs struct {
	defaultConfig *Config
	tenantConfig  TenantConfig
}

// DefaultTenantConfigs creates and returns a new TenantConfigs with the defaults populated.
func DefaultTenantConfigs() *TenantConfigs {
	return &TenantConfigs{
		defaultConfig: &Config{},
		tenantConfig:  nil,
	}
}

// NewTenantConfig makes a new TenantConfigs
func NewTenantConfigs(tenantConfig TenantConfig) (*TenantConfigs, error) {
	return &TenantConfigs{
		defaultConfig: DefaultTenantConfigs().defaultConfig,
		tenantConfig:  tenantConfig,
	}, nil
}

func (o *TenantConfigs) getOverridesForUser(userID string) *Config {
	if o.tenantConfig != nil {
		l := o.tenantConfig(userID)
		if l != nil {
			return l
		}
	}
	return o.defaultConfig
}

func (o *TenantConfigs) LogStreamCreation(userID string) bool {
	return o.getOverridesForUser(userID).LogStreamCreation
}

func (o *TenantConfigs) LogPushRequest(userID string) bool {
	return o.getOverridesForUser(userID).LogPushRequest
}

func (o *TenantConfigs) LogPushRequestStreams(userID string) bool {
	return o.getOverridesForUser(userID).LogPushRequestStreams
}