storj/cmd/statreceiver/sanitizer.go

48 lines
1.0 KiB
Go
Raw Normal View History

2019-01-24 20:15:10 +00:00
// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.
package main
import (
"bytes"
"strings"
"time"
"unicode"
)
// Sanitizer is a MetricDest that replaces nonalphanumeric characters with
// underscores.
type Sanitizer struct {
2019-01-01 09:41:27 +00:00
dest MetricDest
}
2019-01-01 09:41:27 +00:00
// NewSanitizer creates a Sanitizer that sends sanitized metrics to dest.
func NewSanitizer(dest MetricDest) *Sanitizer { return &Sanitizer{dest: dest} }
// Metric implements MetricDest
2019-01-01 09:41:27 +00:00
func (s *Sanitizer) Metric(application, instance string, key []byte, val float64, ts time.Time) error {
return s.dest.Metric(sanitize(application), sanitize(instance), sanitizeb(key), val, ts)
}
func sanitize(val string) string {
return strings.Replace(strings.Map(safechar, val), "..", ".", -1)
}
func sanitizeb(val []byte) []byte {
return bytes.Replace(bytes.Map(safechar, val), []byte(".."), []byte("."), -1)
}
func safechar(r rune) rune {
if unicode.IsLetter(r) || unicode.IsNumber(r) {
return r
}
2019-01-01 09:41:27 +00:00
switch r {
case '/':
return '.'
case '.', '-':
return r
}
return '_'
}