github.com/m3db/m3@v1.5.1-0.20231129193456-75a402aa583b/src/cmd/services/m3query/config/config.go

// Copyright (c) 2018 Uber Technologies, Inc.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.

package config

import (
	"errors"
	"math"
	"time"

	etcdclient "github.com/m3db/m3/src/cluster/client/etcd"
	"github.com/m3db/m3/src/cluster/placement"
	"github.com/m3db/m3/src/cmd/services/m3coordinator/downsample"
	ingestm3msg "github.com/m3db/m3/src/cmd/services/m3coordinator/ingest/m3msg"
	"github.com/m3db/m3/src/cmd/services/m3coordinator/server/m3msg"
	"github.com/m3db/m3/src/metrics/aggregation"
	"github.com/m3db/m3/src/query/api/v1/handler/prometheus/handleroptions"
	"github.com/m3db/m3/src/query/graphite/graphite"
	"github.com/m3db/m3/src/query/models"
	"github.com/m3db/m3/src/query/storage"
	"github.com/m3db/m3/src/query/storage/m3"
	"github.com/m3db/m3/src/query/storage/m3/consolidators"
	"github.com/m3db/m3/src/query/storage/m3/storagemetadata"
	xconfig "github.com/m3db/m3/src/x/config"
	"github.com/m3db/m3/src/x/debug/config"
	"github.com/m3db/m3/src/x/instrument"
	xlog "github.com/m3db/m3/src/x/log"
	"github.com/m3db/m3/src/x/opentracing"
	xtime "github.com/m3db/m3/src/x/time"
)

// BackendStorageType is an enum for different backends.
type BackendStorageType string

const (
	// GRPCStorageType is for backends which only support grpc endpoints.
	GRPCStorageType BackendStorageType = "grpc"
	// M3DBStorageType is for m3db backend.
	M3DBStorageType BackendStorageType = "m3db"
	// NoopEtcdStorageType is for a noop backend which returns empty results for
	// any query and blackholes any writes, but requires that a valid etcd cluster
	// is defined and can be connected to. Primarily used for standalone
	// coordinators used only to serve m3admin APIs.
	NoopEtcdStorageType BackendStorageType = "noop-etcd"

	// PromRemoteStorageType is a type of storage that is backed by Prometheus Remote Write compatible API.
	PromRemoteStorageType BackendStorageType = "prom-remote"

	defaultListenAddress = "0.0.0.0:7201"

	defaultCarbonIngesterListenAddress = "0.0.0.0:7204"

	defaultQueryTimeout = 30 * time.Second

	defaultPrometheusMaxSamplesPerQuery = 100000000
)

var (
	defaultLogging = xlog.Configuration{
		Level: "info",
	}
	defaultMetricsSanitization        = instrument.PrometheusMetricSanitization
	defaultMetricsExtendedMetricsType = instrument.NoExtendedMetrics
	defaultMetrics                    = instrument.MetricsConfiguration{
		RootScope: &instrument.ScopeConfiguration{
			Prefix: "coordinator",
		},
		PrometheusReporter: &instrument.PrometheusConfiguration{
			HandlerPath: "/metrics",
			// Default to coordinator (until https://github.com/m3db/m3/issues/682 is resolved)
			ListenAddress: "0.0.0.0:7203",
		},
		Sanitization:    &defaultMetricsSanitization,
		SamplingRate:    1.0,
		ExtendedMetrics: &defaultMetricsExtendedMetricsType,
	}

	// 5m is the default lookback in Prometheus.
	defaultLookbackDuration = 5 * time.Minute

	defaultCarbonIngesterAggregationType = aggregation.Mean

	// By default, cap total series to prevent results of
	// extremely large sizes consuming too much memory.
	defaultStorageQuerySeriesLimit = 100_000
	defaultStorageQueryDocsLimit   = 0 // Default OFF.

	// By default, raise errors instead of truncating results so
	// users do not experience see unexpected results.
	defaultRequireExhaustive = true

	defaultWriteWorkerPool = xconfig.WorkerPoolPolicy{
		GrowOnDemand:          true,
		Size:                  4096,
		KillWorkerProbability: 0.001,
	}

	// By default, return up to 4 metric metadata stats per request.
	defaultMaxMetricMetadataStats = 4
)

// Configuration is the configuration for the query service.
type Configuration struct {
	// Metrics configuration.
	Metrics *instrument.MetricsConfiguration `yaml:"metrics"`

	// Logging configuration.
	Logging *xlog.Configuration `yaml:"logging"`

	// Tracing configures opentracing. If not provided, tracing is disabled.
	Tracing opentracing.TracingConfiguration `yaml:"tracing"`

	// Clusters is the DB cluster configurations for read, write and
	// query endpoints.
	Clusters m3.ClustersStaticConfiguration `yaml:"clusters"`

	// LocalConfiguration is the local embedded configuration if running
	// coordinator embedded in the DB.
	Local *LocalConfiguration `yaml:"local"`

	// ClusterManagement for placement, namespaces and database management
	// endpoints.
	ClusterManagement ClusterManagementConfiguration `yaml:"clusterManagement"`

	// PrometheusRemoteBackend configures prometheus remote write backend.
	// Used only when backend property is "prom-remote"
	PrometheusRemoteBackend *PrometheusRemoteBackendConfiguration `yaml:"prometheusRemoteBackend"`

	// ListenAddress is the server listen address.
	ListenAddress *string `yaml:"listenAddress"`

	// Filter is the read/write/complete tags filter configuration.
	Filter FilterConfiguration `yaml:"filter"`

	// RPC is the RPC configuration.
	RPC *RPCConfiguration `yaml:"rpc"`

	// HTTP is the HTTP configuration.
	HTTP HTTPConfiguration `yaml:"http"`

	// Backend is the backend store for query service.
	Backend BackendStorageType `yaml:"backend"`

	// TagOptions is the tag configuration options.
	TagOptions TagOptionsConfiguration `yaml:"tagOptions"`

	// ReadWorkerPool is the worker pool policy for read requests.
	ReadWorkerPool xconfig.WorkerPoolPolicy `yaml:"readWorkerPoolPolicy"`

	// WriteWorkerPool is the worker pool policy for write requests.
	WriteWorkerPool *xconfig.WorkerPoolPolicy `yaml:"writeWorkerPoolPolicy"`

	// WriteForwarding is the write forwarding options.
	WriteForwarding WriteForwardingConfiguration `yaml:"writeForwarding"`

	// Downsample configures how the metrics should be downsampled.
	Downsample downsample.Configuration `yaml:"downsample"`

	// Ingest is the ingest server.
	Ingest *IngestConfiguration `yaml:"ingest"`

	// Carbon is the carbon configuration.
	Carbon *CarbonConfiguration `yaml:"carbon"`

	// Middleware is middleware-specific configuration.
	Middleware MiddlewareConfiguration `yaml:"middleware"`

	// Query is the query configuration.
	Query QueryConfiguration `yaml:"query"`

	// Limits specifies limits on per-query resource usage.
	Limits LimitsConfiguration `yaml:"limits"`

	// LookbackDuration determines the lookback duration for queries
	LookbackDuration *time.Duration `yaml:"lookbackDuration"`

	// ResultOptions are the results options for query.
	ResultOptions ResultOptions `yaml:"resultOptions"`

	// DeprecatedExperimental is the configuration for the experimental API group. It is not used anymore
	// and only kept for backwards-support with older configuration files.
	DeprecatedExperimental ExperimentalAPIConfiguration `yaml:"experimental"`

	// StoreMetricsType controls if metrics type is stored or not.
	StoreMetricsType *bool `yaml:"storeMetricsType"`

	// MultiProcess is the multi-process configuration.
	MultiProcess MultiProcessConfiguration `yaml:"multiProcess"`

	// Debug configuration.
	Debug config.DebugConfiguration `yaml:"debug"`
}

// ListenAddressOrDefault returns the listen address or default.
func (c *Configuration) ListenAddressOrDefault() string {
	if c.ListenAddress != nil {
		return *c.ListenAddress
	}

	return defaultListenAddress
}

// LoggingOrDefault returns the logging config or default.
func (c *Configuration) LoggingOrDefault() xlog.Configuration {
	if c.Logging != nil {
		return *c.Logging
	}

	return defaultLogging
}

// MetricsOrDefault returns the metrics config or default.
func (c *Configuration) MetricsOrDefault() *instrument.MetricsConfiguration {
	if c.Metrics != nil {
		return c.Metrics
	}

	return &defaultMetrics
}

// WriteWorkerPoolOrDefault returns the write worker pool config or default.
func (c *Configuration) WriteWorkerPoolOrDefault() xconfig.WorkerPoolPolicy {
	if c.WriteWorkerPool != nil {
		return *c.WriteWorkerPool
	}

	return defaultWriteWorkerPool
}

// WriteForwardingConfiguration is the write forwarding configuration.
type WriteForwardingConfiguration struct {
	PromRemoteWrite handleroptions.PromWriteHandlerForwardingOptions `yaml:"promRemoteWrite"`
}

// Filter is a query filter type.
type Filter string

const (
	// FilterLocalOnly is a filter that specifies local only storage should be used.
	FilterLocalOnly Filter = "local_only"
	// FilterRemoteOnly is a filter that specifies remote only storage should be used.
	FilterRemoteOnly Filter = "remote_only"
	// FilterAllowAll is a filter that specifies all storages should be used.
	FilterAllowAll Filter = "allow_all"
	// FilterAllowNone is a filter that specifies no storages should be used.
	FilterAllowNone Filter = "allow_none"
)

// FilterConfiguration is the filters for write/read/complete tags storage filters.
type FilterConfiguration struct {
	Read         Filter `yaml:"read"`
	Write        Filter `yaml:"write"`
	CompleteTags Filter `yaml:"completeTags"`
}

// ResultOptions are the result options for query.
type ResultOptions struct {
	// KeepNaNs keeps NaNs before returning query results.
	// The default is false, which matches Prometheus
	KeepNaNs bool `yaml:"keepNans"`
}

// QueryConfiguration is the query configuration.
type QueryConfiguration struct {
	// Timeout is the query timeout.
	Timeout *time.Duration `yaml:"timeout"`
	// DefaultEngine is the default query engine.
	DefaultEngine string `yaml:"defaultEngine"`
	// ConsolidationConfiguration are configs for consolidating fetched queries.
	ConsolidationConfiguration ConsolidationConfiguration `yaml:"consolidation"`
	// Prometheus is prometheus client configuration.
	Prometheus PrometheusQueryConfiguration `yaml:"prometheus"`
	// RestrictTags is an optional configuration that can be set to restrict
	// all queries with certain tags by.
	RestrictTags *RestrictTagsConfiguration `yaml:"restrictTags"`
	// RequireLabelsEndpointStartEndTime requires requests to /label(s) endpoints
	// to specify a start and end time to prevent unbounded queries.
	RequireLabelsEndpointStartEndTime bool `yaml:"requireLabelsEndpointStartEndTime"`
	// RequireSeriesEndpointStartEndTime requires requests to /series endpoint
	// to specify a start and end time to prevent unbounded queries.
	RequireSeriesEndpointStartEndTime bool `yaml:"requireSeriesEndpointStartEndTime"`
}

// TimeoutOrDefault returns the configured timeout or default value.
func (c QueryConfiguration) TimeoutOrDefault() time.Duration {
	if v := c.Timeout; v != nil {
		return *v
	}
	return defaultQueryTimeout
}

// RestrictTagsAsStorageRestrictByTag returns restrict tags as
// storage options to restrict all queries by default.
func (c QueryConfiguration) RestrictTagsAsStorageRestrictByTag() (*storage.RestrictByTag, bool, error) {
	if c.RestrictTags == nil {
		return nil, false, nil
	}

	var (
		cfg    = *c.RestrictTags
		result = handleroptions.StringTagOptions{
			Restrict: make([]handleroptions.StringMatch, 0, len(cfg.Restrict)),
			Strip:    cfg.Strip,
		}
	)
	for _, elem := range cfg.Restrict {
		value := handleroptions.StringMatch(elem)
		result.Restrict = append(result.Restrict, value)
	}

	opts, err := result.StorageOptions()
	if err != nil {
		return nil, false, err
	}

	return opts, true, nil
}

// RestrictTagsConfiguration applies tag restriction to all queries.
type RestrictTagsConfiguration struct {
	Restrict []StringMatch `yaml:"match"`
	Strip    []string      `yaml:"strip"`
}

// StringMatch is an easy to use representation of models.Matcher.
type StringMatch struct {
	Name  string `yaml:"name"`
	Type  string `yaml:"type"`
	Value string `yaml:"value"`
}

// ConsolidationConfiguration are configs for consolidating fetched queries.
type ConsolidationConfiguration struct {
	// MatchType determines the options by which series should match.
	MatchType consolidators.MatchType `yaml:"matchType"`
}

// PrometheusQueryConfiguration is the prometheus query engine configuration.
type PrometheusQueryConfiguration struct {
	// MaxSamplesPerQuery is the limit on fetched samples per query.
	MaxSamplesPerQuery *int `yaml:"maxSamplesPerQuery"`

	// Convert configures Prometheus time series conversions.
	Convert *PrometheusConvertConfiguration `yaml:"convert"`
}

// ConvertOptionsOrDefault creates storage.PromConvertOptions based on the given configuration.
func (c PrometheusQueryConfiguration) ConvertOptionsOrDefault() storage.PromConvertOptions {
	opts := storage.NewPromConvertOptions()

	if v := c.Convert; v != nil {
		if value := v.ResolutionThresholdForCounterNormalization; value != nil {
			opts = opts.SetResolutionThresholdForCounterNormalization(*value)
		}

		opts = opts.SetValueDecreaseTolerance(v.ValueDecreaseTolerance)

		// Default to max time so that it's always applicable if value
		// decrease tolerance is non-zero.
		toleranceUntil := xtime.UnixNano(math.MaxInt64)
		if value := v.ValueDecreaseToleranceUntil; value != nil {
			toleranceUntil = xtime.ToUnixNano(*value)
		}
		opts = opts.SetValueDecreaseToleranceUntil(toleranceUntil)
	}

	return opts
}

// PrometheusConvertConfiguration configures Prometheus time series conversions.
type PrometheusConvertConfiguration struct {
	// ResolutionThresholdForCounterNormalization sets the resolution threshold starting from which
	// Prometheus counter normalization is performed in order to avoid Prometheus counter
	// extrapolation artifacts.
	ResolutionThresholdForCounterNormalization *time.Duration `yaml:"resolutionThresholdForCounterNormalization"`

	// ValueDecreaseTolerance allows for setting a specific amount of tolerance
	// to avoid returning a decrease if it's below a certain tolerance.
	// This is useful for applications that have precision issues emitting
	// monotonic increasing data and will accidentally make it seem like the
	// counter value decreases when it hasn't changed.
	ValueDecreaseTolerance float64 `yaml:"valueDecreaseTolerance"`

	// ValueDecreaseToleranceUntil allows for setting a time threshold on
	// which to apply the conditional value decrease threshold.
	ValueDecreaseToleranceUntil *time.Time `yaml:"valueDecreaseToleranceUntil"`
}

// MaxSamplesPerQueryOrDefault returns the max samples per query or default.
func (c PrometheusQueryConfiguration) MaxSamplesPerQueryOrDefault() int {
	if v := c.MaxSamplesPerQuery; v != nil {
		return *v
	}

	return defaultPrometheusMaxSamplesPerQuery
}

// LimitsConfiguration represents limitations on resource usage in the query
// instance. Limits are split between per-query and global limits.
type LimitsConfiguration struct {
	// PerQuery configures limits which apply to each query individually.
	PerQuery PerQueryLimitsConfiguration `yaml:"perQuery"`
}

// PerQueryLimitsConfiguration represents limits on resource usage within a
// single query. Zero or negative values imply no limit.
type PerQueryLimitsConfiguration struct {
	// MaxFetchedSeries limits the number of time series returned for any given
	// individual storage node per query, before returning result to query
	// service.
	MaxFetchedSeries int `yaml:"maxFetchedSeries"`

	// InstanceMultiple increases the per database instance series limit.
	// The series limit per database instance is calculated as:
	//
	// InstanceSeriesLimit = MaxFetchesSeries / (instances per replica) * InstanceMultiple.
	//
	// A value > 1 allows a buffer in case data is not uniformly sharded across instances in a replica.
	// If set to 0 the feature is disabled and the MaxFetchedSeries is used as the limit for database instance.
	// For large clusters, enabling this feature can dramatically decrease the amount of wasted series read from a
	// single database instance.
	InstanceMultiple float32 `yaml:"instanceMultiple"`

	// MaxFetchedDocs limits the number of index documents matched for any given
	// individual storage node per query, before returning result to query
	// service.
	MaxFetchedDocs int `yaml:"maxFetchedDocs"`

	// MaxFetchedRange limits the time range of index documents matched for any given
	// individual storage node per query, before returning result to query
	// service.
	MaxFetchedRange time.Duration `yaml:"maxFetchedRange"`

	// RequireExhaustive results in an error if the query exceeds any limit.
	RequireExhaustive *bool `yaml:"requireExhaustive"`

	// MaxMetricMetadataStats limits the number of metric metadata stats to return
	// as a response header after a query. If unset, defaults to 4. If set to zero,
	// no metric metadata stats will be returned as a response header.
	MaxMetricMetadataStats *int `yaml:"maxMetricMetadataStats"`
}

// AsFetchOptionsBuilderLimitsOptions converts this configuration to
// handleroptions.FetchOptionsBuilderLimitsOptions.
func (l *PerQueryLimitsConfiguration) AsFetchOptionsBuilderLimitsOptions() handleroptions.FetchOptionsBuilderLimitsOptions {
	seriesLimit := defaultStorageQuerySeriesLimit
	if v := l.MaxFetchedSeries; v > 0 {
		seriesLimit = v
	}

	docsLimit := defaultStorageQueryDocsLimit
	if v := l.MaxFetchedDocs; v > 0 {
		docsLimit = v
	}

	requireExhaustive := defaultRequireExhaustive
	if r := l.RequireExhaustive; r != nil {
		requireExhaustive = *r
	}

	maxMetricMetadataStats := defaultMaxMetricMetadataStats
	if v := l.MaxMetricMetadataStats; v != nil {
		maxMetricMetadataStats = *v
	}

	return handleroptions.FetchOptionsBuilderLimitsOptions{
		SeriesLimit:            seriesLimit,
		InstanceMultiple:       l.InstanceMultiple,
		DocsLimit:              docsLimit,
		RangeLimit:             l.MaxFetchedRange,
		RequireExhaustive:      requireExhaustive,
		MaxMetricMetadataStats: maxMetricMetadataStats,
	}
}

// IngestConfiguration is the configuration for ingestion server.
type IngestConfiguration struct {
	// Ingester is the configuration for storage based ingester.
	Ingester ingestm3msg.Configuration `yaml:"ingester"`

	// M3Msg is the configuration for m3msg server.
	M3Msg m3msg.Configuration `yaml:"m3msg"`
}

// CarbonConfiguration is the configuration for the carbon server.
type CarbonConfiguration struct {
	// Ingester if set defines an ingester to run for carbon.
	Ingester *CarbonIngesterConfiguration `yaml:"ingester"`
	// LimitsFind sets the limits configuration for find queries.
	LimitsFind *LimitsConfiguration `yaml:"limitsFind"`
	// LimitsRender sets the limits configuration for render queries.
	LimitsRender *LimitsConfiguration `yaml:"limitsRender"`
	// AggregateNamespacesAllData configures whether all aggregate
	// namespaces contain entire copies of the data set.
	// This affects whether queries can be optimized or not, if false
	// they cannot be since it's unclear if data matching an expression
	// sits in one or many or none of the aggregate namespaces so all
	// must be queried, but if true then it can be determined based
	// on the query range whether a single namespace can fulfill the
	// entire query and if so to only fetch from that one aggregated namespace.
	AggregateNamespacesAllData bool `yaml:"aggregateNamespacesAllData"`
	// ShiftTimeStart sets a constant time to shift start by.
	ShiftTimeStart time.Duration `yaml:"shiftTimeStart"`
	// ShiftTimeEnd sets a constant time to shift end by.
	ShiftTimeEnd time.Duration `yaml:"shiftTimeEnd"`
	// ShiftStepsStart sets a constant set of steps to shift start by.
	ShiftStepsStart int `yaml:"shiftStepsStart"`
	// ShiftStepsEnd sets a constant set of steps to shift end by.
	ShiftStepsEnd int `yaml:"shiftStepsEnd"`
	// ShiftStepsStartWhenAtResolutionBoundary sets a constant set of steps to
	// shift start by if and only if the start is an exact match to the
	// resolution boundary of a query.
	ShiftStepsStartWhenAtResolutionBoundary *int `yaml:"shiftStepsStartWhenAtResolutionBoundary"`
	// ShiftStepsEndWhenAtResolutionBoundary sets a constant set of steps to
	// shift end by if and only if the end is an exact match to the
	// resolution boundary of a query.
	ShiftStepsEndWhenAtResolutionBoundary *int `yaml:"shiftStepsEndWhenAtResolutionBoundary"`
	// ShiftStepsStartWhenEndAtResolutionBoundary sets a constant set of steps to
	// shift start by if and only if the end is an exact match to the resolution boundary
	// of a query AND the start is not an exact match to the resolution boundary.
	ShiftStepsStartWhenEndAtResolutionBoundary *int `yaml:"shiftStepsStartWhenEndAtResolutionBoundary"`
	// ShiftStepsEndWhenStartAtResolutionBoundary sets a constant set of steps to
	// shift end by if and only if the start is an exact match to the resolution boundary
	// of a query AND the end is not an exact match to the resolution boundary.
	ShiftStepsEndWhenStartAtResolutionBoundary *int `yaml:"shiftStepsEndWhenStartAtResolutionBoundary"`
	// RenderPartialStart sets whether to render partial datapoints when
	// the start time is between a datapoint's resolution step size.
	RenderPartialStart bool `yaml:"renderPartialStart"`
	// RenderPartialEnd sets whether to render partial datapoints when
	// the end time is between a datapoint's resolution step size.
	RenderPartialEnd bool `yaml:"renderPartialEnd"`
	// RenderSeriesAllNaNs will render series that have only NaNs for entire
	// output instead of returning an empty array of datapoints.
	RenderSeriesAllNaNs bool `yaml:"renderSeriesAllNaNs"`
	// CompileEscapeAllNotOnlyQuotes will escape all characters when using a backslash
	// in a quoted string rather than just reserving for escaping quotes.
	CompileEscapeAllNotOnlyQuotes bool `yaml:"compileEscapeAllNotOnlyQuotes"`
	// FindResultsIncludeBothExpandableAndLeaf will include both an expandable
	// node and a leaf node if there is a duplicate path node that is both an
	// expandable node and a leaf node.
	FindResultsIncludeBothExpandableAndLeaf bool `yaml:"findResultsIncludeBothExpandableAndLeaf"`
}

// MiddlewareConfiguration is middleware-specific configuration.
type MiddlewareConfiguration struct {
	// Logging configures the logging middleware.
	Logging LoggingMiddlewareConfiguration `yaml:"logging"`
	// Metrics configures the metrics middleware.
	Metrics MetricsMiddlewareConfiguration `yaml:"metrics"`
	// Prometheus configures prometheus-related middleware.
	Prometheus PrometheusMiddlewareConfiguration `yaml:"prometheus"`
}

// LoggingMiddlewareConfiguration configures the logging middleware.
type LoggingMiddlewareConfiguration struct {
	// Threshold defines the latency threshold for logging the response. If zero, the default of 1s is used. To disable
	// response logging set Disabled.
	Threshold time.Duration
	// Disabled turns off response logging by default for endpoints.
	Disabled bool
}

// MetricsMiddlewareConfiguration configures the metrics middleware.
type MetricsMiddlewareConfiguration struct {
	// QueryEndpointsClassification contains the configuration for sizing queries to
	// the query and query_range Prometheus endpoints.
	QueryEndpointsClassification QueryClassificationConfig `yaml:"queryEndpointsClassification"`
	// LabelEndpointsClassification contains the configuration for sizing queries to
	// the label names and label values Prometheus endpoints.
	LabelEndpointsClassification QueryClassificationConfig `yaml:"labelEndpointsClassification"`
	// AddStatusToLatencies will add a tag with the query's response code to
	// middleware latency metrics.
	// NB: Setting this to true will increase cardinality by the number of
	// expected response codes (likely around ~10).
	AddStatusToLatencies bool `yaml:"addStatusToLatencies"`
}

// QueryClassificationConfig contains the buckets used to group a query into a bucket for
// the sake of understanding the size of the query based on a specific dimension. Currently,
// we have two sets of buckets: results and duration. The results buckets help us understand
// the size of the query based on the number of results returned whereas the duration buckets help
// us understand the size of the query based on the time range of the query. Dimension values are
// rounded down to the nearest bucket. If the value is smaller than all buckets, then it is
// allocated to the first bucket. Buckets are expected to be ordered in ascending order.
type QueryClassificationConfig struct {
	// ResultsBuckets contains the buckets to be compared with the number of results (e.g. number of
	// time series or labels) returned by a specific endpoint.
	ResultsBuckets []int `yaml:"resultsBuckets"`
	// DurationBuckets contains the buckets to be compared with time range of a query for a
	// specific endpoint.
	DurationBuckets []time.Duration `yaml:"durationBuckets"`
}

// Enabled returns true if classification buckets were specified.
func (q *QueryClassificationConfig) Enabled() bool {
	return len(q.DurationBuckets) > 0 || len(q.ResultsBuckets) > 0
}

// PrometheusMiddlewareConfiguration configures the range rewriting middleware.
type PrometheusMiddlewareConfiguration struct {
	// ResolutionMultiplier is the multiple that will be applied to the range if it's determined
	// that it needs to be updated. If this value is greater than 0, the range in a query will be
	// updated if the namespaces used to service the request have resolution(s)
	// that are greater than the range. The range will be updated to the largest resolution
	// of the namespaces to service the request * the multiplier specified here. If this multiplier
	// is 0, then this feature is disabled.
	ResolutionMultiplier int `yaml:"resolutionMultiplier"`
}

// CarbonIngesterConfiguration is the configuration struct for carbon ingestion.
type CarbonIngesterConfiguration struct {
	ListenAddress  string                             `yaml:"listenAddress"`
	MaxConcurrency int                                `yaml:"maxConcurrency"`
	Rewrite        CarbonIngesterRewriteConfiguration `yaml:"rewrite"`
	Rules          []CarbonIngesterRuleConfiguration  `yaml:"rules"`
}

// CarbonIngesterRewriteConfiguration is the configuration for rewriting
// metrics at ingestion.
type CarbonIngesterRewriteConfiguration struct {
	// Cleanup will perform:
	// - Trailing/leading dot elimination.
	// - Double dot elimination.
	// - Irregular char replacement with underscores (_), currently irregular
	//   is defined as not being in [0-9a-zA-Z-_:#].
	Cleanup bool `yaml:"cleanup"`
}

// LookbackDurationOrDefault validates the LookbackDuration
func (c Configuration) LookbackDurationOrDefault() (time.Duration, error) {
	if c.LookbackDuration == nil {
		return defaultLookbackDuration, nil
	}

	v := *c.LookbackDuration
	if v < 0 {
		return 0, errors.New("lookbackDuration must be > 0")
	}

	return v, nil
}

// ListenAddressOrDefault returns the specified carbon ingester listen address if provided, or the
// default value if not.
func (c *CarbonIngesterConfiguration) ListenAddressOrDefault() string {
	if c.ListenAddress != "" {
		return c.ListenAddress
	}

	return defaultCarbonIngesterListenAddress
}

// RulesOrDefault returns the specified carbon ingester rules if provided, or generates reasonable
// defaults using the provided aggregated namespaces if not.
func (c *CarbonIngesterConfiguration) RulesOrDefault(namespaces m3.ClusterNamespaces) []CarbonIngesterRuleConfiguration {
	if len(c.Rules) > 0 {
		return c.Rules
	}

	if namespaces.NumAggregatedClusterNamespaces() == 0 {
		return nil
	}

	// Default to fanning out writes for all metrics to all aggregated namespaces if any exists.
	policies := make([]CarbonIngesterStoragePolicyConfiguration, 0, len(namespaces))
	for _, ns := range namespaces {
		if ns.Options().Attributes().MetricsType == storagemetadata.AggregatedMetricsType {
			policies = append(policies, CarbonIngesterStoragePolicyConfiguration{
				Resolution: ns.Options().Attributes().Resolution,
				Retention:  ns.Options().Attributes().Retention,
			})
		}
	}

	if len(policies) == 0 {
		return nil
	}

	// Create a single catch-all rule with a policy for each of the aggregated namespaces we
	// enumerated above.
	aggregationEnabled := true
	return []CarbonIngesterRuleConfiguration{
		{
			Pattern: graphite.MatchAllPattern,
			Aggregation: CarbonIngesterAggregationConfiguration{
				Enabled: &aggregationEnabled,
				Type:    &defaultCarbonIngesterAggregationType,
			},
			Policies: policies,
		},
	}
}

// CarbonIngesterRuleConfiguration is the configuration struct for a carbon
// ingestion rule.
type CarbonIngesterRuleConfiguration struct {
	Pattern     string                                     `yaml:"pattern"`
	Contains    string                                     `yaml:"contains"`
	Continue    bool                                       `yaml:"continue"`
	Aggregation CarbonIngesterAggregationConfiguration     `yaml:"aggregation"`
	Policies    []CarbonIngesterStoragePolicyConfiguration `yaml:"policies"`
}

// CarbonIngesterAggregationConfiguration is the configuration struct
// for the aggregation for a carbon ingest rule's storage policy.
type CarbonIngesterAggregationConfiguration struct {
	Enabled *bool             `yaml:"enabled"`
	Type    *aggregation.Type `yaml:"type"`
}

// EnabledOrDefault returns whether aggregation should be enabled based on the provided configuration,
// or a default value otherwise.
func (c *CarbonIngesterAggregationConfiguration) EnabledOrDefault() bool {
	if c.Enabled != nil {
		return *c.Enabled
	}

	return true
}

// TypeOrDefault returns the aggregation type that should be used based on the provided configuration,
// or a default value otherwise.
func (c *CarbonIngesterAggregationConfiguration) TypeOrDefault() aggregation.Type {
	if c.Type != nil {
		return *c.Type
	}

	return defaultCarbonIngesterAggregationType
}

// CarbonIngesterStoragePolicyConfiguration is the configuration struct for
// a carbon rule's storage policies.
type CarbonIngesterStoragePolicyConfiguration struct {
	Resolution time.Duration `yaml:"resolution" validate:"nonzero"`
	Retention  time.Duration `yaml:"retention" validate:"nonzero"`
}

// LocalConfiguration is the local embedded configuration if running
// coordinator embedded in the DB.
type LocalConfiguration struct {
	// Namespaces is the list of namespaces that the local embedded DB has.
	Namespaces []m3.ClusterStaticNamespaceConfiguration `yaml:"namespaces"`
}

// ClusterManagementConfiguration is configuration for the placement,
// namespaces and database management endpoints (optional).
type ClusterManagementConfiguration struct {
	// Etcd is the client configuration for etcd.
	Etcd *etcdclient.Configuration `yaml:"etcd"`

	// Placement is the cluster placement configuration.
	Placement placement.Configuration `yaml:"placement"`
}

// RemoteConfigurations is a set of remote host configurations.
type RemoteConfigurations []RemoteConfiguration

// RemoteConfiguration is the configuration for a single remote host.
type RemoteConfiguration struct {
	// Name is the name for the remote zone.
	Name string `yaml:"name"`
	// RemoteListenAddresses is the remote listen addresses to call for remote
	// coordinator calls in the remote zone.
	RemoteListenAddresses []string `yaml:"remoteListenAddresses"`
	// ErrorBehavior overrides the default error behavior for this host.
	//
	// NB: defaults to warning on error.
	ErrorBehavior *storage.ErrorBehavior `yaml:"errorBehavior"`
}

// RPCConfiguration is the RPC configuration for the coordinator for
// the GRPC server used for remote coordinator to coordinator calls.
type RPCConfiguration struct {
	// Enabled determines if coordinator RPC is enabled for remote calls.
	//
	// NB: this is no longer necessary to set to true if RPC is desired; enabled
	// status is inferred based on which other options are provided;
	// this remains for back-compat, and for disabling any existing RPC options.
	Enabled *bool `yaml:"enabled"`

	// ListenAddress is the RPC server listen address.
	ListenAddress string `yaml:"listenAddress"`

	// Remotes are the configuration settings for remote coordinator zones.
	Remotes RemoteConfigurations `yaml:"remotes"`

	// RemoteListenAddresses is the remote listen addresses to call for
	// remote coordinator calls.
	//
	// NB: this is deprecated in favor of using RemoteZones, as setting
	// RemoteListenAddresses will only allow for a single remote zone to be used.
	RemoteListenAddresses []string `yaml:"remoteListenAddresses"`

	// ErrorBehavior overrides the default error behavior for all rpc hosts.
	//
	// NB: defaults to warning on error.
	ErrorBehavior *storage.ErrorBehavior `yaml:"errorBehavior"`

	// ReflectionEnabled will enable reflection on the GRPC server, useful
	// for testing connectivity with grpcurl, etc.
	ReflectionEnabled bool `yaml:"reflectionEnabled"`
}

// PrometheusRemoteBackendConfiguration configures prometheus remote write backend.
type PrometheusRemoteBackendConfiguration struct {
	Endpoints       []PrometheusRemoteBackendEndpointConfiguration `yaml:"endpoints"`
	RequestTimeout  *time.Duration                                 `yaml:"requestTimeout"`
	ConnectTimeout  *time.Duration                                 `yaml:"connectTimeout"`
	KeepAlive       *time.Duration                                 `yaml:"keepAlive"`
	IdleConnTimeout *time.Duration                                 `yaml:"idleConnTimeout"`
	MaxIdleConns    *int                                           `yaml:"maxIdleConns"`
}

// PrometheusRemoteBackendEndpointConfiguration configures single endpoint.
type PrometheusRemoteBackendEndpointConfiguration struct {
	Name    string `yaml:"name"`
	Address string `yaml:"address"`
	// When nil all unaggregated data will be sent to this endpoint.
	StoragePolicy *PrometheusRemoteBackendStoragePolicyConfiguration `yaml:"storagePolicy"`
}

// PrometheusRemoteBackendStoragePolicyConfiguration configures storage policy for single endpoint.
type PrometheusRemoteBackendStoragePolicyConfiguration struct {
	Resolution time.Duration `yaml:"resolution" validate:"nonzero"`
	Retention  time.Duration `yaml:"retention" validate:"nonzero"`

	// Downsample is downsampling options to be used with this storage policy.
	Downsample *m3.DownsampleClusterStaticNamespaceConfiguration `yaml:"downsample"`
}

// HTTPConfiguration is the HTTP configuration for configuring
// the HTTP server used by the coordinator to serve incoming requests.
type HTTPConfiguration struct {
	// EnableH2C enables support for the HTTP/2 cleartext protocol. H2C
	// enables the use of HTTP/2 without requiring TLS.
	EnableH2C bool `yaml:"enableH2C"`
}

// TagOptionsConfiguration is the configuration for shared tag options
// Currently only name, but can expand to cover deduplication settings, or other
// relevant options.
type TagOptionsConfiguration struct {
	// MetricName specifies the tag name that corresponds to the metric's name tag
	// If not provided, defaults to `__name__`.
	MetricName string `yaml:"metricName"`

	// BucketName specifies the tag name that corresponds to the metric's bucket.
	// If not provided, defaults to `le`.
	BucketName string `yaml:"bucketName"`

	// Scheme determines the default ID generation scheme. Defaults to TypeQuoted.
	Scheme models.IDSchemeType `yaml:"idScheme"`

	// Filters are optional tag filters, removing all series with tags
	// matching the filter from computations.
	Filters []TagFilter `yaml:"filters"`

	// AllowTagNameDuplicates allows for duplicate tags to appear on series.
	AllowTagNameDuplicates bool `yaml:"allowTagNameDuplicates"`

	// AllowTagValueEmpty allows for empty tags to appear on series.
	AllowTagValueEmpty bool `yaml:"allowTagValueEmpty"`
}

// TagFilter is a tag filter.
type TagFilter struct {
	// Values are the values to filter.
	//
	// NB:If this is unset, all series containing
	// a tag with given `Name` are filtered.
	Values []string `yaml:"values"`
	// Name is the tag name.
	Name string `yaml:"name"`
}

// TagOptionsFromConfig translates tag option configuration into tag options.
func TagOptionsFromConfig(cfg TagOptionsConfiguration) (models.TagOptions, error) {
	opts := models.NewTagOptions()
	name := cfg.MetricName
	if name != "" {
		opts = opts.SetMetricName([]byte(name))
	}

	bucket := cfg.BucketName
	if bucket != "" {
		opts = opts.SetBucketName([]byte(bucket))
	}

	if cfg.Scheme == models.TypeDefault {
		// Default to quoted if unspecified.
		cfg.Scheme = models.TypeQuoted
	}

	opts = opts.SetIDSchemeType(cfg.Scheme)
	if err := opts.Validate(); err != nil {
		return nil, err
	}

	if cfg.Filters != nil {
		filters := make([]models.Filter, 0, len(cfg.Filters))
		for _, filter := range cfg.Filters {
			values := make([][]byte, 0, len(filter.Values))
			for _, strVal := range filter.Values {
				values = append(values, []byte(strVal))
			}

			filters = append(filters, models.Filter{
				Name:   []byte(filter.Name),
				Values: values,
			})
		}

		opts = opts.SetFilters(filters)
	}

	opts = opts.SetAllowTagNameDuplicates(cfg.AllowTagNameDuplicates)
	opts = opts.SetAllowTagValueEmpty(cfg.AllowTagValueEmpty)

	return opts, nil
}

// ExperimentalAPIConfiguration is the configuration for the experimental API group.
type ExperimentalAPIConfiguration struct {
	Enabled bool `yaml:"enabled"`
}

// MultiProcessConfiguration is the multi-process configuration which
// allows running multiple sub-processes of an instance reusing the
// same listen ports.
type MultiProcessConfiguration struct {
	// Enabled is whether to enable multi-process execution.
	Enabled bool `yaml:"enabled"`
	// Count is the number of sub-processes to run, leave zero
	// to auto-detect based on number of CPUs.
	Count int `yaml:"count" validate:"min=0"`
	// PerCPU is the factor of processes to run per CPU, leave
	// zero to use the default of 0.5 per CPU (i.e. one process for
	// every two CPUs).
	PerCPU float64 `yaml:"perCPU" validate:"min=0.0, max=1.0"`
	// GoMaxProcs if set will explicitly set the child GOMAXPROCs env var.
	GoMaxProcs int `yaml:"goMaxProcs"`
}