123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400 |
- /*
- Copyright 2019 The Vitess Authors.
- Licensed under the Apache License, Version 2.0 (the "License");
- you may not use this file except in compliance with the License.
- You may obtain a copy of the License at
- http://www.apache.org/licenses/LICENSE-2.0
- Unless required by applicable law or agreed to in writing, software
- distributed under the License is distributed on an "AS IS" BASIS,
- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- See the License for the specific language governing permissions and
- limitations under the License.
- */
- package opentsdb
- import (
- "encoding/json"
- "expvar"
- "reflect"
- "sort"
- "testing"
- "time"
- "git.nspix.com/golang/micro/stats"
- )
- func TestOpenTsdbCounter(t *testing.T) {
- name := "counter_name"
- c := stats.NewCounter(name, "counter description")
- c.Add(1)
- checkOutput(t, name, `
- [
- {
- "metric": "vtgate.counter_name",
- "timestamp": 1234,
- "value": 1,
- "tags": {
- "host": "localhost"
- }
- }
- ]`)
- }
- func TestOpenTsdbCounterFunc(t *testing.T) {
- name := "counter_fn_name"
- stats.NewCounterFunc(name, "help", func() int64 {
- return 2
- })
- checkOutput(t, name, `
- [
- {
- "metric": "vtgate.counter_fn_name",
- "timestamp": 1234,
- "value": 2,
- "tags": {
- "host": "localhost"
- }
- }
- ]`)
- }
- func TestGaugesWithMultiLabels(t *testing.T) {
- name := "gauges_with_multi_labels_name"
- gauges := stats.NewGaugesWithMultiLabels(name, "help", []string{"flavor", "texture"})
- gauges.Add([]string{"sour", "brittle"}, 3)
- checkOutput(t, name, `
- [
- {
- "metric": "vtgate.gauges_with_multi_labels_name",
- "timestamp": 1234,
- "value": 3,
- "tags": {
- "flavor": "sour",
- "host": "localhost",
- "texture": "brittle"
- }
- }
- ]`)
- }
- type myVar bool
- func (mv *myVar) String() string {
- return `{"myKey": 1.2}`
- }
- func TestExpvar(t *testing.T) {
- name := "blah_expvar"
- expvar.Publish(name, new(myVar))
- checkOutput(t, name, `
- [
- {
- "metric": "vtgate.expvar.blah_expvar.mykey",
- "timestamp": 1234,
- "value": 1.2,
- "tags": {
- "host": "localhost"
- }
- }
- ]`)
- }
- func TestOpenTsdbTimings(t *testing.T) {
- name := "blah_timings"
- cats := []string{"cat1", "cat2"}
- timing := stats.NewTimings(name, "help", "category", cats...)
- timing.Add("cat1", time.Duration(1000000000))
- timing.Add("cat1", time.Duration(1))
- checkOutput(t, name, `
- [
- {
- "metric": "vtgate.blah_timings.1000000",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat1",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.1000000",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat2",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.10000000",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat1",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.10000000",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat2",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.100000000",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat1",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.100000000",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat2",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.1000000000",
- "timestamp": 1234,
- "value": 1,
- "tags": {
- "category": "cat1",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.1000000000",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat2",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.10000000000",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat1",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.10000000000",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat2",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.500000",
- "timestamp": 1234,
- "value": 1,
- "tags": {
- "category": "cat1",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.500000",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat2",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.5000000",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat1",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.5000000",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat2",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.50000000",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat1",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.50000000",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat2",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.500000000",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat1",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.500000000",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat2",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.5000000000",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat1",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.5000000000",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat2",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.count",
- "timestamp": 1234,
- "value": 2,
- "tags": {
- "category": "cat1",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.count",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat2",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.inf",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat1",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.inf",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat2",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.time",
- "timestamp": 1234,
- "value": 1000000001,
- "tags": {
- "category": "cat1",
- "host": "localhost"
- }
- },
- {
- "metric": "vtgate.blah_timings.time",
- "timestamp": 1234,
- "value": 0,
- "tags": {
- "category": "cat2",
- "host": "localhost"
- }
- }
- ]`)
- }
- func checkOutput(t *testing.T, statName string, wantJSON string) {
- backend := &openTSDBBackend{
- prefix: "vtgate",
- commonTags: map[string]string{"host": "localhost"},
- }
- timestamp := int64(1234)
- dc := &dataCollector{
- settings: backend,
- timestamp: timestamp,
- }
- found := false
- expvar.Do(func(kv expvar.KeyValue) {
- if kv.Key == statName {
- found = true
- dc.addExpVar(kv)
- sort.Sort(byMetric(dc.dataPoints))
- gotBytes, err := json.MarshalIndent(dc.dataPoints, "", " ")
- if err != nil {
- t.Errorf("Failed to marshal json: %v", err)
- return
- }
- var got interface{}
- err = json.Unmarshal(gotBytes, &got)
- if err != nil {
- t.Errorf("Failed to marshal json: %v", err)
- return
- }
- var want interface{}
- err = json.Unmarshal([]byte(wantJSON), &want)
- if err != nil {
- t.Errorf("Failed to marshal json: %v", err)
- return
- }
- if !reflect.DeepEqual(got, want) {
- t.Errorf("addExpVar(%#v) = %s, want %s", kv, string(gotBytes), wantJSON)
- }
- }
- })
- if !found {
- t.Errorf("Stat %s not found?...", statName)
- }
- }
|