mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-11 14:53:49 +00:00
58 lines
2.1 KiB
Go
58 lines
2.1 KiB
Go
|
package consulagent
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promauth"
|
||
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promutils"
|
||
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/proxy"
|
||
|
|
)
|
||
|
|
|
||
|
|
// SDConfig represents service discovery config for Consul Agent.
|
||
|
|
//
|
||
|
|
// See https://grafana.com/docs/loki/latest/clients/promtail/configuration/#consulagent_sd_config
|
||
|
|
type SDConfig struct {
|
||
|
|
Server string `yaml:"server,omitempty"`
|
||
|
|
Token *promauth.Secret `yaml:"token"`
|
||
|
|
Datacenter string `yaml:"datacenter"`
|
||
|
|
|
||
|
|
// Namespace only supported at enterprise consul.
|
||
|
|
// https://www.consul.io/docs/enterprise/namespaces
|
||
|
|
Namespace string `yaml:"namespace,omitempty"`
|
||
|
|
|
||
|
|
Scheme string `yaml:"scheme,omitempty"`
|
||
|
|
Username string `yaml:"username"`
|
||
|
|
Password *promauth.Secret `yaml:"password"`
|
||
|
|
HTTPClientConfig promauth.HTTPClientConfig `yaml:",inline"`
|
||
|
|
ProxyURL *proxy.URL `yaml:"proxy_url,omitempty"`
|
||
|
|
ProxyClientConfig promauth.ProxyClientConfig `yaml:",inline"`
|
||
|
|
Services []string `yaml:"services,omitempty"`
|
||
|
|
TagSeparator *string `yaml:"tag_separator,omitempty"`
|
||
|
|
// See https://developer.hashicorp.com/consul/api-docs/features/filtering
|
||
|
|
// list of supported filters https://developer.hashicorp.com/consul/api-docs/catalog#filtering-1
|
||
|
|
Filter string `yaml:"filter,omitempty"`
|
||
|
|
|
||
|
|
// RefreshInterval time.Duration `yaml:"refresh_interval"`
|
||
|
|
// refresh_interval is obtained from `-promscrape.consulagentSDCheckInterval` command-line option.
|
||
|
|
}
|
||
|
|
|
||
|
|
// GetLabels returns Consul labels according to sdc.
|
||
|
|
func (sdc *SDConfig) GetLabels(baseDir string) ([]*promutils.Labels, error) {
|
||
|
|
cfg, err := getAPIConfig(sdc, baseDir)
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("cannot get API config: %w", err)
|
||
|
|
}
|
||
|
|
ms := getServiceNodesLabels(cfg)
|
||
|
|
return ms, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// MustStop stops further usage for sdc.
|
||
|
|
func (sdc *SDConfig) MustStop() {
|
||
|
|
v := configMap.Delete(sdc)
|
||
|
|
if v != nil {
|
||
|
|
// v can be nil if GetLabels wasn't called yet.
|
||
|
|
cfg := v.(*apiConfig)
|
||
|
|
cfg.mustStop()
|
||
|
|
}
|
||
|
|
}
|