VictoriaMetrics/lib/promscrape/discovery/consulagent/service_node.go

66 lines
2.3 KiB
Go
Raw Normal View History

package consulagent
import (
"fmt"
"strconv"
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promscrape/discovery/consul"
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promscrape/discoveryutils"
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promutils"
)
// getServiceNodesLabels returns labels for Consul service nodes with given cfg.
func getServiceNodesLabels(cfg *apiConfig) []*promutils.Labels {
sns := cfg.consulWatcher.getServiceNodesSnapshot()
var ms []*promutils.Labels
for svc, sn := range sns {
for i := range sn {
ms = appendTargetLabels(sn[i], ms, svc, cfg.tagSeparator, cfg.agent)
}
}
return ms
}
func appendTargetLabels(sn consul.ServiceNode, ms []*promutils.Labels, serviceName, tagSeparator string, agent *consul.Agent) []*promutils.Labels {
const metaPrefix = "__meta_consulagent_"
var addr string
// If the service address is not empty it should be used instead of the node address
// since the service may be registered remotely through a different node.
if sn.Service.Address != "" {
addr = discoveryutils.JoinHostPort(sn.Service.Address, sn.Service.Port)
} else {
addr = discoveryutils.JoinHostPort(agent.Member.Addr, sn.Service.Port)
}
m := promutils.NewLabels(16)
m.Add("__address__", addr)
m.Add(metaPrefix+"address", agent.Member.Addr)
m.Add(metaPrefix+"dc", agent.Config.Datacenter)
m.Add(metaPrefix+"health", consul.AggregatedStatus(sn.Checks))
m.Add(metaPrefix+"namespace", sn.Service.Namespace)
m.Add(metaPrefix+"node", agent.Config.NodeName)
m.Add(metaPrefix+"service", serviceName)
m.Add(metaPrefix+"service_address", sn.Service.Address)
m.Add(metaPrefix+"service_id", sn.Service.ID)
m.Add(metaPrefix+"service_port", strconv.Itoa(sn.Service.Port))
discoveryutils.AddTagsToLabels(m, sn.Service.Tags, metaPrefix, tagSeparator)
for k, v := range agent.Meta {
m.Add(discoveryutils.SanitizeLabelName(metaPrefix+"metadata_"+k), v)
}
for k, v := range sn.Service.Meta {
m.Add(discoveryutils.SanitizeLabelName(metaPrefix+"service_metadata_"+k), v)
}
for k, v := range sn.Node.TaggedAddresses {
m.Add(discoveryutils.SanitizeLabelName(metaPrefix+"tagged_address_"+k), v)
}
for k, v := range sn.Service.TaggedAddresses {
address := fmt.Sprintf("%s:%d", v.Address, v.Port)
m.Add(discoveryutils.SanitizeLabelName(metaPrefix+"tagged_address_"+k), address)
}
ms = append(ms, m)
return ms
}