mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-15 08:23:34 +01:00
be6da5053f
- Return meta-labels for the discovered targets via promutils.Labels instead of map[string]string. This improves the speed of generating meta-labels for discovered targets by up to 5x. - Remove memory allocations in hot paths during ScrapeWork generation. The ScrapeWork contains scrape settings for a single discovered target. This improves the service discovery speed by up to 2x.
78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
package gce
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promutils"
|
|
)
|
|
|
|
// SDCheckInterval defines interval for targets refresh.
|
|
var SDCheckInterval = flag.Duration("promscrape.gceSDCheckInterval", time.Minute, "Interval for checking for changes in gce. "+
|
|
"This works only if gce_sd_configs is configured in '-promscrape.config' file. "+
|
|
"See https://docs.victoriametrics.com/sd_configs.html#gce_sd_configs for details")
|
|
|
|
// SDConfig represents service discovery config for gce.
|
|
//
|
|
// See https://prometheus.io/docs/prometheus/latest/configuration/configuration/#gce_sd_config
|
|
type SDConfig struct {
|
|
Project string `yaml:"project"`
|
|
Zone ZoneYAML `yaml:"zone"`
|
|
Filter string `yaml:"filter,omitempty"`
|
|
// RefreshInterval time.Duration `yaml:"refresh_interval"`
|
|
// refresh_interval is obtained from `-promscrape.gceSDCheckInterval` command-line option.
|
|
Port *int `yaml:"port,omitempty"`
|
|
TagSeparator *string `yaml:"tag_separator,omitempty"`
|
|
}
|
|
|
|
// ZoneYAML holds info about zones.
|
|
type ZoneYAML struct {
|
|
Zones []string
|
|
}
|
|
|
|
// UnmarshalYAML implements yaml.Unmarshaler
|
|
func (z *ZoneYAML) UnmarshalYAML(unmarshal func(interface{}) error) error {
|
|
var v interface{}
|
|
if err := unmarshal(&v); err != nil {
|
|
return err
|
|
}
|
|
var zones []string
|
|
switch t := v.(type) {
|
|
case string:
|
|
zones = []string{t}
|
|
case []interface{}:
|
|
for _, vv := range t {
|
|
zone, ok := vv.(string)
|
|
if !ok {
|
|
return fmt.Errorf("unexpected zone type detected: %T; contents: %#v", vv, vv)
|
|
}
|
|
zones = append(zones, zone)
|
|
}
|
|
default:
|
|
return fmt.Errorf("unexpected type unmarshaled for ZoneYAML: %T; contents: %#v", v, v)
|
|
}
|
|
z.Zones = zones
|
|
return nil
|
|
}
|
|
|
|
// MarshalYAML implements yaml.Marshaler
|
|
func (z ZoneYAML) MarshalYAML() (interface{}, error) {
|
|
return z.Zones, nil
|
|
}
|
|
|
|
// GetLabels returns gce labels according to sdc.
|
|
func (sdc *SDConfig) GetLabels(baseDir string) ([]*promutils.Labels, error) {
|
|
cfg, err := getAPIConfig(sdc)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot get API config: %w", err)
|
|
}
|
|
ms := getInstancesLabels(cfg)
|
|
return ms, nil
|
|
}
|
|
|
|
// MustStop stops further usage for sdc.
|
|
func (sdc *SDConfig) MustStop() {
|
|
configMap.Delete(sdc)
|
|
}
|