mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-24 19:30:06 +01:00
c85a5b7fcb
* add docker discovery * add test * add labels test and add scrape work * remove TODO * refactor to merge apiConfig and sdConfig * apply suggestion
38 lines
1.1 KiB
Go
38 lines
1.1 KiB
Go
package docker
|
|
|
|
import (
|
|
"fmt"
|
|
)
|
|
|
|
// DockerSwarmSDConfig represents docker swarm service discovery configuration
|
|
//
|
|
// See https://prometheus.io/docs/prometheus/latest/configuration/configuration/#dockerswarm_sd_config
|
|
type DockerSwarmSDConfig struct {
|
|
Role string `yaml:"role"`
|
|
DockerSDConfig `yaml:",inline"`
|
|
// refresh_interval is obtained from `-promscrape.dockerswarmSDCheckInterval` command-line option
|
|
}
|
|
|
|
// GetLabels returns dockerswarm labels according to sdc.
|
|
func (sdc *DockerSwarmSDConfig) GetLabels(baseDir string) ([]map[string]string, error) {
|
|
cfg, err := getAPIConfigFromDockerSwarmSDConfig(sdc, baseDir)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot get API config: %w", err)
|
|
}
|
|
switch sdc.Role {
|
|
case "tasks":
|
|
return getTasksLabels(cfg)
|
|
case "services":
|
|
return getServicesLabels(cfg)
|
|
case "nodes":
|
|
return getNodesLabels(cfg)
|
|
default:
|
|
return nil, fmt.Errorf("unexpected `role`: %q; must be one of `tasks`, `services` or `nodes`; skipping it", sdc.Role)
|
|
}
|
|
}
|
|
|
|
// MustStop stops further usage for sdc.
|
|
func (sdc *DockerSwarmSDConfig) MustStop() {
|
|
configMap.Delete(sdc)
|
|
}
|