86 lines
2.0 KiB
Go
86 lines
2.0 KiB
Go
package main
|
|
|
|
import (
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/by-cx/lobby/server"
|
|
)
|
|
|
|
// [
|
|
// {
|
|
// "targets": [ "<host>", ... ],
|
|
// "labels": {
|
|
// "<labelname>": "<labelvalue>", ...
|
|
// }
|
|
// },
|
|
// ...
|
|
// ]
|
|
|
|
// PrometheusServices holds multiple PrometheusService structs
|
|
type PrometheusServices []PrometheusService
|
|
|
|
// PrometheusService represents a single set of targets and labels for Prometheus
|
|
type PrometheusService struct {
|
|
Targets []string
|
|
Labels map[string]string
|
|
}
|
|
|
|
// preparePrometheusOutput returns PrometheusServices which is struct compatible to what Prometheus expects
|
|
// labels starting "ne:" will be used as NodeExporter labels. Label "ne:port:9123" will be used as port
|
|
// used in the targets field. Same for "ne:host:1.2.3.4".
|
|
func preparePrometheusOutput(name string, discoveries []server.Discovery) PrometheusServices {
|
|
services := PrometheusServices{}
|
|
|
|
for _, discovery := range discoveries {
|
|
port := strconv.Itoa(int(config.NodeExporterPort))
|
|
host := discovery.Hostname
|
|
var add bool // add to the prometheus output when there is at least one prometheus related label
|
|
|
|
labels := map[string]string{}
|
|
|
|
for _, label := range discovery.FindLabelsByPrefix("prometheus:" + name + ":") {
|
|
trimmed := strings.TrimPrefix(label.String(), "prometheus:"+name+":")
|
|
parts := strings.SplitN(trimmed, ":", 2)
|
|
if len(parts) == 2 {
|
|
if parts[0] == "port" {
|
|
port = parts[1]
|
|
} else if parts[0] == "host" {
|
|
host = parts[1]
|
|
} else {
|
|
labels[parts[0]] = parts[1]
|
|
}
|
|
add = true
|
|
}
|
|
}
|
|
|
|
// This has to be checked here again because FindLabels adds : at the end of the label name.
|
|
if !add {
|
|
for _, label := range discovery.Labels {
|
|
if label.String() == "prometheus:"+name {
|
|
add = true
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
if add {
|
|
// Omit port part if "-" is set
|
|
target := host + ":" + port
|
|
if port == "-" {
|
|
target = host
|
|
}
|
|
|
|
service := PrometheusService{
|
|
Targets: []string{target},
|
|
Labels: labels,
|
|
}
|
|
|
|
services = append(services, service)
|
|
}
|
|
|
|
}
|
|
|
|
return services
|
|
}
|