mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-22 08:26:19 +01:00
9edeecabc8
This makes easier to read and debug these tests. This also reduces test lines count by 15% from 3K to 2.5K . See https://itnext.io/f-tests-as-a-replacement-for-table-driven-tests-in-go-8814a8b19e9e . While at it, consistently use t.Fatal* instead of t.Error*, since t.Error* usually leads to more complicated and fragile tests, while it doesn't bring any practical benefits over t.Fatal*.
79 lines
2.4 KiB
Go
79 lines
2.4 KiB
Go
package nomad
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promscrape/discoveryutils"
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promutils"
|
|
)
|
|
|
|
// getServiceLabels returns labels for Nomad services with given cfg.
|
|
func getServiceLabels(cfg *apiConfig) []*promutils.Labels {
|
|
svcs := cfg.nomadWatcher.getServiceSnapshot()
|
|
var ms []*promutils.Labels
|
|
for _, s := range svcs {
|
|
for i := range s {
|
|
ms = s[i].appendTargetLabels(ms, cfg.tagSeparator)
|
|
}
|
|
}
|
|
return ms
|
|
}
|
|
|
|
// ServiceList is a list of Nomad services.
|
|
// See https://developer.hashicorp.com/nomad/api-docs/services#list-services
|
|
type ServiceList struct {
|
|
Namespace string `json:"Namespace"`
|
|
Services []service `json:"Services"`
|
|
}
|
|
|
|
type service struct {
|
|
ServiceName string `json:"ServiceName"`
|
|
Tags []string `json:"Tags"`
|
|
}
|
|
|
|
// Service is Nomad service.
|
|
// See https://developer.hashicorp.com/nomad/api-docs/services#list-services
|
|
type Service struct {
|
|
ID string `json:"ID"`
|
|
ServiceName string `json:"ServiceName"`
|
|
Namespace string `json:"Namespace"`
|
|
NodeID string `json:"NodeID"`
|
|
Datacenter string `json:"Datacenter"`
|
|
JobID string `json:"JobID"`
|
|
AllocID string `json:"AllocID"`
|
|
Tags []string `json:"Tags"`
|
|
Address string `json:"Address"`
|
|
Port int `json:"Port"`
|
|
}
|
|
|
|
func parseServices(data []byte) ([]Service, error) {
|
|
var sns []Service
|
|
if err := json.Unmarshal(data, &sns); err != nil {
|
|
return nil, fmt.Errorf("cannot unmarshal Services from %q: %w", data, err)
|
|
}
|
|
return sns, nil
|
|
}
|
|
|
|
func (svc *Service) appendTargetLabels(ms []*promutils.Labels, tagSeparator string) []*promutils.Labels {
|
|
addr := discoveryutils.JoinHostPort(svc.Address, svc.Port)
|
|
m := promutils.NewLabels(16)
|
|
m.Add("__address__", addr)
|
|
m.Add("__meta_nomad_address", svc.Address)
|
|
m.Add("__meta_nomad_dc", svc.Datacenter)
|
|
m.Add("__meta_nomad_namespace", svc.Namespace)
|
|
m.Add("__meta_nomad_node_id", svc.NodeID)
|
|
m.Add("__meta_nomad_service", svc.ServiceName)
|
|
m.Add("__meta_nomad_service_address", svc.Address)
|
|
m.Add("__meta_nomad_service_alloc_id", svc.AllocID)
|
|
m.Add("__meta_nomad_service_id", svc.ID)
|
|
m.Add("__meta_nomad_service_job_id", svc.JobID)
|
|
m.Add("__meta_nomad_service_port", strconv.Itoa(svc.Port))
|
|
|
|
discoveryutils.AddTagsToLabels(m, svc.Tags, "__meta_nomad_", tagSeparator)
|
|
|
|
ms = append(ms, m)
|
|
return ms
|
|
}
|