mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-16 00:41:24 +01:00
103 lines
2.5 KiB
Go
103 lines
2.5 KiB
Go
package discoveryutils
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net"
|
|
"regexp"
|
|
"sort"
|
|
"strconv"
|
|
"sync"
|
|
"sync/atomic"
|
|
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/prompbmarshal"
|
|
)
|
|
|
|
// SanitizeLabelName replaces anything that doesn't match
|
|
// client_label.LabelNameRE with an underscore.
|
|
//
|
|
// This has been copied from Prometheus sources at util/strutil/strconv.go
|
|
func SanitizeLabelName(name string) string {
|
|
m := sanitizedLabelNames.Load().(*sync.Map)
|
|
v, ok := m.Load(name)
|
|
if ok {
|
|
// Fast path - the sanitized label name is found in the cache.
|
|
sp := v.(*string)
|
|
return *sp
|
|
}
|
|
// Slow path - sanitize name and store it in the cache.
|
|
sanitizedName := invalidLabelCharRE.ReplaceAllString(name, "_")
|
|
// Make a copy of name in order to limit memory usage to the name length,
|
|
// since the name may point to bigger string.
|
|
s := string(append([]byte{}, name...))
|
|
if sanitizedName == name {
|
|
// point sanitizedName to just allocated s, since it may point to name,
|
|
// which, in turn, can point to bigger string.
|
|
sanitizedName = s
|
|
}
|
|
sp := &sanitizedName
|
|
m.Store(s, sp)
|
|
n := atomic.AddUint64(&sanitizedLabelNamesLen, 1)
|
|
if n > 100e3 {
|
|
atomic.StoreUint64(&sanitizedLabelNamesLen, 0)
|
|
sanitizedLabelNames.Store(&sync.Map{})
|
|
}
|
|
return sanitizedName
|
|
}
|
|
|
|
var (
|
|
sanitizedLabelNames atomic.Value
|
|
sanitizedLabelNamesLen uint64
|
|
|
|
invalidLabelCharRE = regexp.MustCompile(`[^a-zA-Z0-9_]`)
|
|
)
|
|
|
|
func init() {
|
|
sanitizedLabelNames.Store(&sync.Map{})
|
|
}
|
|
|
|
// JoinHostPort returns host:port.
|
|
//
|
|
// Host may be dns name, ipv4 or ipv6 address.
|
|
func JoinHostPort(host string, port int) string {
|
|
portStr := strconv.Itoa(port)
|
|
return net.JoinHostPort(host, portStr)
|
|
}
|
|
|
|
// SortedLabels represents sorted labels.
|
|
type SortedLabels []prompbmarshal.Label
|
|
|
|
// GetByName returns the label with the given name from sls.
|
|
func (sls *SortedLabels) GetByName(name string) string {
|
|
for _, lb := range *sls {
|
|
if lb.Name == name {
|
|
return lb.Value
|
|
}
|
|
}
|
|
return ""
|
|
}
|
|
|
|
// UnmarshalJSON unmarshals JSON from data.
|
|
func (sls *SortedLabels) UnmarshalJSON(data []byte) error {
|
|
var m map[string]string
|
|
if err := json.Unmarshal(data, &m); err != nil {
|
|
return err
|
|
}
|
|
*sls = GetSortedLabels(m)
|
|
return nil
|
|
}
|
|
|
|
// GetSortedLabels returns SortedLabels built from m.
|
|
func GetSortedLabels(m map[string]string) SortedLabels {
|
|
a := make([]prompbmarshal.Label, 0, len(m))
|
|
for k, v := range m {
|
|
a = append(a, prompbmarshal.Label{
|
|
Name: k,
|
|
Value: v,
|
|
})
|
|
}
|
|
sort.Slice(a, func(i, j int) bool {
|
|
return a[i].Name < a[j].Name
|
|
})
|
|
return a
|
|
}
|