mirror of
https://github.com/open-telemetry/opentelemetry-go.git
synced 2024-12-14 10:13:10 +02:00
ae9033e01f
* Use an array key to label encoding in the SDK * Comment * Precommit * Comment * Comment * Feedback from krnowak * Do not overwrite the Key * Add the value test requested * Add a comment
290 lines
7.5 KiB
Go
290 lines
7.5 KiB
Go
// Copyright 2019, OpenTelemetry 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 metric_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"math"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"go.opentelemetry.io/otel/api/core"
|
|
"go.opentelemetry.io/otel/api/key"
|
|
"go.opentelemetry.io/otel/api/metric"
|
|
export "go.opentelemetry.io/otel/sdk/export/metric"
|
|
"go.opentelemetry.io/otel/sdk/export/metric/aggregator"
|
|
sdk "go.opentelemetry.io/otel/sdk/metric"
|
|
"go.opentelemetry.io/otel/sdk/metric/aggregator/array"
|
|
"go.opentelemetry.io/otel/sdk/metric/aggregator/counter"
|
|
)
|
|
|
|
type correctnessBatcher struct {
|
|
t *testing.T
|
|
|
|
records []export.Record
|
|
}
|
|
|
|
type testLabelEncoder struct{}
|
|
|
|
func (cb *correctnessBatcher) AggregatorFor(descriptor *export.Descriptor) export.Aggregator {
|
|
name := descriptor.Name()
|
|
switch {
|
|
case strings.HasSuffix(name, ".counter"):
|
|
return counter.New()
|
|
case strings.HasSuffix(name, ".disabled"):
|
|
return nil
|
|
default:
|
|
return array.New()
|
|
}
|
|
}
|
|
|
|
func (cb *correctnessBatcher) CheckpointSet() export.CheckpointSet {
|
|
cb.t.Fatal("Should not be called")
|
|
return nil
|
|
}
|
|
|
|
func (*correctnessBatcher) FinishedCollection() {
|
|
}
|
|
|
|
func (cb *correctnessBatcher) Process(_ context.Context, record export.Record) error {
|
|
cb.records = append(cb.records, record)
|
|
return nil
|
|
}
|
|
|
|
func (testLabelEncoder) Encode(labels []core.KeyValue) string {
|
|
return fmt.Sprint(labels)
|
|
}
|
|
|
|
func TestInputRangeTestCounter(t *testing.T) {
|
|
ctx := context.Background()
|
|
batcher := &correctnessBatcher{
|
|
t: t,
|
|
}
|
|
sdk := sdk.New(batcher, sdk.NewDefaultLabelEncoder())
|
|
|
|
var sdkErr error
|
|
sdk.SetErrorHandler(func(handleErr error) {
|
|
sdkErr = handleErr
|
|
})
|
|
|
|
counter := sdk.NewInt64Counter("name.counter", metric.WithMonotonic(true))
|
|
|
|
counter.Add(ctx, -1, sdk.Labels())
|
|
require.Equal(t, aggregator.ErrNegativeInput, sdkErr)
|
|
sdkErr = nil
|
|
|
|
checkpointed := sdk.Collect(ctx)
|
|
sum, err := batcher.records[0].Aggregator().(aggregator.Sum).Sum()
|
|
require.Equal(t, int64(0), sum.AsInt64())
|
|
require.Equal(t, 1, checkpointed)
|
|
require.Nil(t, err)
|
|
|
|
batcher.records = nil
|
|
counter.Add(ctx, 1, sdk.Labels())
|
|
checkpointed = sdk.Collect(ctx)
|
|
sum, err = batcher.records[0].Aggregator().(aggregator.Sum).Sum()
|
|
require.Equal(t, int64(1), sum.AsInt64())
|
|
require.Equal(t, 1, checkpointed)
|
|
require.Nil(t, err)
|
|
require.Nil(t, sdkErr)
|
|
}
|
|
|
|
func TestInputRangeTestMeasure(t *testing.T) {
|
|
ctx := context.Background()
|
|
batcher := &correctnessBatcher{
|
|
t: t,
|
|
}
|
|
sdk := sdk.New(batcher, sdk.NewDefaultLabelEncoder())
|
|
|
|
var sdkErr error
|
|
sdk.SetErrorHandler(func(handleErr error) {
|
|
sdkErr = handleErr
|
|
})
|
|
|
|
measure := sdk.NewFloat64Measure("name.measure", metric.WithAbsolute(true))
|
|
|
|
measure.Record(ctx, -1, sdk.Labels())
|
|
require.Equal(t, aggregator.ErrNegativeInput, sdkErr)
|
|
sdkErr = nil
|
|
|
|
checkpointed := sdk.Collect(ctx)
|
|
count, err := batcher.records[0].Aggregator().(aggregator.Distribution).Count()
|
|
require.Equal(t, int64(0), count)
|
|
require.Equal(t, 1, checkpointed)
|
|
require.Nil(t, err)
|
|
|
|
measure.Record(ctx, 1, sdk.Labels())
|
|
measure.Record(ctx, 2, sdk.Labels())
|
|
|
|
batcher.records = nil
|
|
checkpointed = sdk.Collect(ctx)
|
|
|
|
count, err = batcher.records[0].Aggregator().(aggregator.Distribution).Count()
|
|
require.Equal(t, int64(2), count)
|
|
require.Equal(t, 1, checkpointed)
|
|
require.Nil(t, sdkErr)
|
|
require.Nil(t, err)
|
|
}
|
|
|
|
func TestDisabledInstrument(t *testing.T) {
|
|
ctx := context.Background()
|
|
batcher := &correctnessBatcher{
|
|
t: t,
|
|
}
|
|
sdk := sdk.New(batcher, sdk.NewDefaultLabelEncoder())
|
|
measure := sdk.NewFloat64Measure("name.disabled", metric.WithAbsolute(true))
|
|
|
|
measure.Record(ctx, -1, sdk.Labels())
|
|
checkpointed := sdk.Collect(ctx)
|
|
|
|
require.Equal(t, 0, checkpointed)
|
|
require.Equal(t, 0, len(batcher.records))
|
|
}
|
|
|
|
func TestRecordNaN(t *testing.T) {
|
|
ctx := context.Background()
|
|
batcher := &correctnessBatcher{
|
|
t: t,
|
|
}
|
|
sdk := sdk.New(batcher, sdk.NewDefaultLabelEncoder())
|
|
|
|
var sdkErr error
|
|
sdk.SetErrorHandler(func(handleErr error) {
|
|
sdkErr = handleErr
|
|
})
|
|
c := sdk.NewFloat64Counter("counter.name")
|
|
|
|
require.Nil(t, sdkErr)
|
|
c.Add(ctx, math.NaN(), sdk.Labels())
|
|
require.Error(t, sdkErr)
|
|
}
|
|
|
|
func TestSDKAltLabelEncoder(t *testing.T) {
|
|
ctx := context.Background()
|
|
batcher := &correctnessBatcher{
|
|
t: t,
|
|
}
|
|
sdk := sdk.New(batcher, testLabelEncoder{})
|
|
|
|
measure := sdk.NewFloat64Measure("measure")
|
|
measure.Record(ctx, 1, sdk.Labels(key.String("A", "B"), key.String("C", "D")))
|
|
|
|
sdk.Collect(ctx)
|
|
|
|
require.Equal(t, 1, len(batcher.records))
|
|
|
|
labels := batcher.records[0].Labels()
|
|
require.Equal(t, `[{A {8 0 B}} {C {8 0 D}}]`, labels.Encoded())
|
|
}
|
|
|
|
func TestSDKLabelsDeduplication(t *testing.T) {
|
|
ctx := context.Background()
|
|
batcher := &correctnessBatcher{
|
|
t: t,
|
|
}
|
|
sdk := sdk.New(batcher, sdk.NewDefaultLabelEncoder())
|
|
|
|
counter := sdk.NewInt64Counter("counter")
|
|
|
|
const (
|
|
maxKeys = 21
|
|
keySets = 2
|
|
repeats = 3
|
|
)
|
|
var keysA []core.Key
|
|
var keysB []core.Key
|
|
|
|
for i := 0; i < maxKeys; i++ {
|
|
keysA = append(keysA, core.Key(fmt.Sprintf("A%03d", i)))
|
|
keysB = append(keysB, core.Key(fmt.Sprintf("B%03d", i)))
|
|
}
|
|
|
|
var allExpect [][]core.KeyValue
|
|
for numKeys := 0; numKeys < maxKeys; numKeys++ {
|
|
|
|
var kvsA []core.KeyValue
|
|
var kvsB []core.KeyValue
|
|
for r := 0; r < repeats; r++ {
|
|
for i := 0; i < numKeys; i++ {
|
|
kvsA = append(kvsA, keysA[i].Int(r))
|
|
kvsB = append(kvsB, keysB[i].Int(r))
|
|
}
|
|
}
|
|
|
|
var expectA []core.KeyValue
|
|
var expectB []core.KeyValue
|
|
for i := 0; i < numKeys; i++ {
|
|
expectA = append(expectA, keysA[i].Int(repeats-1))
|
|
expectB = append(expectB, keysB[i].Int(repeats-1))
|
|
}
|
|
|
|
counter.Add(ctx, 1, sdk.Labels(kvsA...))
|
|
counter.Add(ctx, 1, sdk.Labels(kvsA...))
|
|
allExpect = append(allExpect, expectA)
|
|
|
|
if numKeys != 0 {
|
|
// In this case A and B sets are the same.
|
|
counter.Add(ctx, 1, sdk.Labels(kvsB...))
|
|
counter.Add(ctx, 1, sdk.Labels(kvsB...))
|
|
allExpect = append(allExpect, expectB)
|
|
}
|
|
|
|
}
|
|
|
|
sdk.Collect(ctx)
|
|
|
|
var actual [][]core.KeyValue
|
|
for _, rec := range batcher.records {
|
|
sum, _ := rec.Aggregator().(aggregator.Sum).Sum()
|
|
require.Equal(t, sum, core.NewInt64Number(2))
|
|
|
|
actual = append(actual, rec.Labels().Ordered())
|
|
}
|
|
|
|
require.ElementsMatch(t, allExpect, actual)
|
|
}
|
|
|
|
func TestDefaultLabelEncoder(t *testing.T) {
|
|
encoder := sdk.NewDefaultLabelEncoder()
|
|
|
|
encoded := encoder.Encode([]core.KeyValue{key.String("A", "B"), key.String("C", "D")})
|
|
require.Equal(t, `A=B,C=D`, encoded)
|
|
|
|
encoded = encoder.Encode([]core.KeyValue{key.String("A", "B,c=d"), key.String(`C\`, "D")})
|
|
require.Equal(t, `A=B\,c\=d,C\\=D`, encoded)
|
|
|
|
encoded = encoder.Encode([]core.KeyValue{key.String(`\`, `=`), key.String(`,`, `\`)})
|
|
require.Equal(t, `\\=\=,\,=\\`, encoded)
|
|
|
|
// Note: the label encoder does not sort or de-dup values,
|
|
// that is done in Labels(...).
|
|
encoded = encoder.Encode([]core.KeyValue{
|
|
key.Int("I", 1),
|
|
key.Uint("U", 1),
|
|
key.Int32("I32", 1),
|
|
key.Uint32("U32", 1),
|
|
key.Int64("I64", 1),
|
|
key.Uint64("U64", 1),
|
|
key.Float64("F64", 1),
|
|
key.Float64("F64", 1),
|
|
key.String("S", "1"),
|
|
key.Bool("B", true),
|
|
})
|
|
require.Equal(t, "I=1,U=1,I32=1,U32=1,I64=1,U64=1,F64=1,F64=1,S=1,B=true", encoded)
|
|
}
|