storj/cmd/statreceiver/sanitizer.go
JT Olio 362f447d9f
cmd/statreceiver: lua-scriptable stat receiver (#636)
* cmd/statreceiver: lua-scriptable stat receiver

Change-Id: I3ce0fe3f1ef4b1f4f27eed90bac0e91cfecf22d7

* some updates

Change-Id: I7c3485adcda1278fce01ae077b4761b3ddb9fb7a

* more comments

Change-Id: I0bb22993cd934c3d40fc1da80d07e49e686b80dd

* linter fixes

Change-Id: Ied014304ecb9aadcf00a6b66ad28f856a428d150

* catch errors

Change-Id: I6e1920f1fd941e66199b30bc427285c19769fc70

* review feedback

Change-Id: I9d4051851eab18970c5f5ddcf4ff265508e541d3

* errorgroup improvements

Change-Id: I4699dda3022f0485fbb50c9dafe692d3921734ff

* too tricky

the previous thing was better for memory with lots of errors at a time
but https://play.golang.org/p/RweTMRjoSCt is too much of a foot gun

Change-Id: I23f0b3d77dd4288fcc20b3756a7110359576bf44
2018-12-11 11:24:31 -07:00

49 lines
1.0 KiB
Go

// Copyright (C) 2018 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 {
m MetricDest
}
// NewSanitizer creates a Sanitizer that sends sanitized metrics to m.
func NewSanitizer(m MetricDest) *Sanitizer { return &Sanitizer{m: m} }
// Metric implements MetricDest
func (s *Sanitizer) Metric(application, instance string, key []byte,
val float64, ts time.Time) error {
return s.m.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
}
switch r {
case '/':
return '.'
case '.', '-':
return r
}
return '_'
}