mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2025-02-09 15:27:11 +00:00
276989716f
This commit adds support for [Marathon](https://mesosphere.github.io/marathon/)
service discovery to the scrape configuration.
The following flag is introduced:
```
-promscrape.marathonSDCheckInterval duration
Interval for checking for changes in Marathon service discovery. This works only if marathon_sd_configs is configured in '-promscrape.config' file. See https://docs.victoriametrics.com/sd_configs.html#marathon_sd_configs for details (default 30s)
```
The service discovery could be config like:
```yaml
scrape_configs:
- job_name: marathon_job
marathon_sd_configs:
servers:
- "..."
- "..."
```
See:
[b555d94d1a/docs/sd_configs.md (marathon_sd_configs))
related issue:
https://github.com/VictoriaMetrics/VictoriaMetrics/issues/6642
---------
Co-authored-by: Zakhar Bessarab <z.bessarab@victoriametrics.com>
104 lines
3.9 KiB
Go
104 lines
3.9 KiB
Go
package vultr
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promauth"
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promscrape/discoveryutils"
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promutils"
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/proxy"
|
|
)
|
|
|
|
// SDCheckInterval defines interval for Vultr targets refresh.
|
|
var SDCheckInterval = flag.Duration("promscrape.vultrSDCheckInterval", 30*time.Second, "Interval for checking for changes in Vultr. "+
|
|
"This works only if vultr_sd_configs is configured in '-promscrape.config' file. "+
|
|
"See https://docs.victoriametrics.com/sd_configs.html#vultr_sd_configs for details")
|
|
|
|
// SDConfig represents service discovery config for Vultr.
|
|
//
|
|
// See: https://prometheus.io/docs/prometheus/latest/configuration/configuration/#vultr_sd_config
|
|
// Additional query params are supported, while Prometheus only supports `Port` and HTTP auth.
|
|
type SDConfig struct {
|
|
// API query params for filtering. All of them are optional.
|
|
// See: https://www.vultr.com/api/#tag/instances/operation/list-instances
|
|
Label string `yaml:"label,omitempty"`
|
|
MainIP string `yaml:"main_ip,omitempty"`
|
|
Region string `yaml:"region,omitempty"`
|
|
FirewallGroupID string `yaml:"firewall_group_id,omitempty"`
|
|
Hostname string `yaml:"hostname,omitempty"`
|
|
|
|
// The port to scrape metrics from. Default 80.
|
|
Port int `yaml:"port"`
|
|
|
|
// General HTTP / Auth configs.
|
|
HTTPClientConfig promauth.HTTPClientConfig `yaml:",inline"`
|
|
ProxyURL *proxy.URL `yaml:"proxy_url,omitempty"`
|
|
ProxyClientConfig promauth.ProxyClientConfig `yaml:",inline"`
|
|
|
|
// refresh_interval is obtained from `-promscrape.vultrSDCheckInterval` command-line option.
|
|
}
|
|
|
|
var configMap = discoveryutils.NewConfigMap()
|
|
|
|
// GetLabels returns Vultr instances' labels according to sdc.
|
|
func (sdc *SDConfig) GetLabels(baseDir string) ([]*promutils.Labels, error) {
|
|
ac, err := getAPIConfig(sdc, baseDir)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot get API config: %w", err)
|
|
}
|
|
instances, err := getInstances(ac)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return getInstanceLabels(instances, ac.port), nil
|
|
}
|
|
|
|
// MustStop stops further usage for sdc.
|
|
func (sdc *SDConfig) MustStop() {
|
|
_ = configMap.Delete(sdc)
|
|
}
|
|
|
|
// getInstanceLabels returns labels for Vultr instances obtained from the given cfg
|
|
func getInstanceLabels(instances []Instance, port int) []*promutils.Labels {
|
|
ms := make([]*promutils.Labels, 0, len(instances))
|
|
|
|
for _, instance := range instances {
|
|
m := promutils.NewLabels(18)
|
|
m.Add("__address__", discoveryutils.JoinHostPort(instance.MainIP, port))
|
|
m.Add("__meta_vultr_instance_allowed_bandwidth_gb", strconv.Itoa(instance.AllowedBandwidth))
|
|
m.Add("__meta_vultr_instance_disk_gb", strconv.Itoa(instance.Disk))
|
|
m.Add("__meta_vultr_instance_hostname", instance.Hostname)
|
|
m.Add("__meta_vultr_instance_id", instance.ID)
|
|
m.Add("__meta_vultr_instance_internal_ip", instance.InternalIP)
|
|
m.Add("__meta_vultr_instance_label", instance.Label)
|
|
m.Add("__meta_vultr_instance_main_ip", instance.MainIP)
|
|
m.Add("__meta_vultr_instance_main_ipv6", instance.V6MainIP)
|
|
m.Add("__meta_vultr_instance_os", instance.OS)
|
|
m.Add("__meta_vultr_instance_os_id", strconv.Itoa(instance.OSID))
|
|
m.Add("__meta_vultr_instance_plan", instance.Plan)
|
|
m.Add("__meta_vultr_instance_region", instance.Region)
|
|
m.Add("__meta_vultr_instance_ram_mb", strconv.Itoa(instance.RAM))
|
|
m.Add("__meta_vultr_instance_server_status", instance.ServerStatus)
|
|
m.Add("__meta_vultr_instance_vcpu_count", strconv.Itoa(instance.VCPUCount))
|
|
|
|
if len(instance.Features) > 0 {
|
|
m.Add("__meta_vultr_instance_features", joinStrings(instance.Features))
|
|
}
|
|
|
|
if len(instance.Tags) > 0 {
|
|
m.Add("__meta_vultr_instance_tags", joinStrings(instance.Tags))
|
|
}
|
|
|
|
ms = append(ms, m)
|
|
}
|
|
|
|
return ms
|
|
}
|
|
|
|
func joinStrings(a []string) string {
|
|
return "," + strings.Join(a, ",") + ","
|
|
}
|