mirror of
https://github.com/hanzoai/traces.git
synced 2026-10-30 18:20:41 +00:00
Joe ElliottandGitHub
e3d03a1e03
* pesky structure laid out Signed-off-by: Joe Elliott <[email protected]> * sibling works Signed-off-by: Joe Elliott <[email protected]> * extended interface Signed-off-by: Joe Elliott <[email protected]> * everything but child Signed-off-by: Joe Elliott <[email protected]> * finished child/parent Signed-off-by: Joe Elliott <[email protected]> * restore to main Signed-off-by: Joe Elliott <[email protected]> * tests passing Signed-off-by: Joe Elliott <[email protected]> * cleanup Signed-off-by: Joe Elliott <[email protected]> * lint Signed-off-by: Joe Elliott <[email protected]> * additional lintery Signed-off-by: Joe Elliott <[email protected]> * i will end you, linter Signed-off-by: Joe Elliott <[email protected]> * some cleanup Signed-off-by: Joe Elliott <[email protected]> * fix not sibling on root Signed-off-by: Joe Elliott <[email protected]> * fix sort on vp2 Signed-off-by: Joe Elliott <[email protected]> * fix tests to watch for -1 Signed-off-by: Joe Elliott <[email protected]> * comments Signed-off-by: Joe Elliott <[email protected]> * Fix sort level in bridge iterator Signed-off-by: Joe Elliott <[email protected]> * nestedSetParent comments Signed-off-by: Joe Elliott <[email protected]> --------- Signed-off-by: Joe Elliott <[email protected]>
280 lines
6.3 KiB
Go
280 lines
6.3 KiB
Go
package vparquet3
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/dustin/go-humanize"
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/grafana/tempo/pkg/util/test"
|
|
"github.com/grafana/tempo/tempodb/backend"
|
|
)
|
|
|
|
func TestCombiner(t *testing.T) {
|
|
methods := []func(a, b *Trace) (*Trace, int, bool){
|
|
func(a, b *Trace) (*Trace, int, bool) {
|
|
c := NewCombiner()
|
|
c.Consume(a)
|
|
c.Consume(b)
|
|
return c.Result()
|
|
},
|
|
}
|
|
|
|
tests := []struct {
|
|
traceA *Trace
|
|
traceB *Trace
|
|
expectedTotal int
|
|
expectedTrace *Trace
|
|
}{
|
|
{
|
|
traceA: nil,
|
|
traceB: &Trace{},
|
|
expectedTotal: -1,
|
|
},
|
|
{
|
|
traceA: &Trace{},
|
|
traceB: nil,
|
|
expectedTotal: -1,
|
|
},
|
|
{
|
|
traceA: &Trace{},
|
|
traceB: &Trace{},
|
|
expectedTotal: 0,
|
|
},
|
|
// root meta from second overrides empty first
|
|
{
|
|
traceA: &Trace{
|
|
TraceID: []byte{0x00, 0x01},
|
|
},
|
|
traceB: &Trace{
|
|
TraceID: []byte{0x00, 0x01},
|
|
RootServiceName: "serviceNameB",
|
|
RootSpanName: "spanNameB",
|
|
StartTimeUnixNano: 10,
|
|
EndTimeUnixNano: 20,
|
|
DurationNano: 10,
|
|
},
|
|
expectedTrace: &Trace{
|
|
TraceID: []byte{0x00, 0x01},
|
|
RootServiceName: "serviceNameB",
|
|
RootSpanName: "spanNameB",
|
|
StartTimeUnixNano: 10,
|
|
EndTimeUnixNano: 20,
|
|
DurationNano: 10,
|
|
},
|
|
},
|
|
// if both set first root name wins
|
|
{
|
|
traceA: &Trace{
|
|
TraceID: []byte{0x00, 0x01},
|
|
RootServiceName: "serviceNameA",
|
|
RootSpanName: "spanNameA",
|
|
},
|
|
traceB: &Trace{
|
|
TraceID: []byte{0x00, 0x01},
|
|
RootServiceName: "serviceNameB",
|
|
RootSpanName: "spanNameB",
|
|
},
|
|
expectedTrace: &Trace{
|
|
TraceID: []byte{0x00, 0x01},
|
|
RootServiceName: "serviceNameA",
|
|
RootSpanName: "spanNameA",
|
|
},
|
|
},
|
|
// second trace start/end override
|
|
{
|
|
traceA: &Trace{
|
|
TraceID: []byte{0x00, 0x01},
|
|
StartTimeUnixNano: 10,
|
|
EndTimeUnixNano: 20,
|
|
DurationNano: 10,
|
|
},
|
|
traceB: &Trace{
|
|
TraceID: []byte{0x00, 0x01},
|
|
StartTimeUnixNano: 5,
|
|
EndTimeUnixNano: 25,
|
|
DurationNano: 20,
|
|
},
|
|
expectedTrace: &Trace{
|
|
TraceID: []byte{0x00, 0x01},
|
|
StartTimeUnixNano: 5,
|
|
EndTimeUnixNano: 25,
|
|
DurationNano: 20,
|
|
},
|
|
},
|
|
// second trace start/end ignored
|
|
{
|
|
traceA: &Trace{
|
|
TraceID: []byte{0x00, 0x01},
|
|
StartTimeUnixNano: 10,
|
|
EndTimeUnixNano: 20,
|
|
DurationNano: 10,
|
|
},
|
|
traceB: &Trace{
|
|
TraceID: []byte{0x00, 0x01},
|
|
StartTimeUnixNano: 12,
|
|
EndTimeUnixNano: 18,
|
|
DurationNano: 6,
|
|
},
|
|
expectedTrace: &Trace{
|
|
TraceID: []byte{0x00, 0x01},
|
|
StartTimeUnixNano: 10,
|
|
EndTimeUnixNano: 20,
|
|
DurationNano: 10,
|
|
},
|
|
},
|
|
{
|
|
traceA: &Trace{
|
|
TraceID: []byte{0x00, 0x01},
|
|
RootServiceName: "serviceNameA",
|
|
ResourceSpans: []ResourceSpans{
|
|
{
|
|
Resource: Resource{
|
|
ServiceName: "serviceNameA",
|
|
},
|
|
ScopeSpans: []ScopeSpans{
|
|
{
|
|
Spans: []Span{
|
|
{
|
|
SpanID: []byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01},
|
|
StatusCode: 0,
|
|
NestedSetLeft: 1,
|
|
NestedSetRight: 2,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
traceB: &Trace{
|
|
TraceID: []byte{0x00, 0x01},
|
|
RootServiceName: "serviceNameB",
|
|
ResourceSpans: []ResourceSpans{
|
|
{
|
|
Resource: Resource{
|
|
ServiceName: "serviceNameB",
|
|
},
|
|
ScopeSpans: []ScopeSpans{
|
|
{
|
|
Spans: []Span{
|
|
{
|
|
SpanID: []byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02},
|
|
ParentSpanID: []byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01},
|
|
StatusCode: 0,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expectedTotal: 2,
|
|
expectedTrace: &Trace{
|
|
TraceID: []byte{0x00, 0x01},
|
|
RootServiceName: "serviceNameA",
|
|
ResourceSpans: []ResourceSpans{
|
|
{
|
|
Resource: Resource{
|
|
ServiceName: "serviceNameA",
|
|
},
|
|
ScopeSpans: []ScopeSpans{
|
|
{
|
|
Spans: []Span{
|
|
{
|
|
SpanID: []byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01},
|
|
StatusCode: 0,
|
|
NestedSetLeft: 1,
|
|
NestedSetRight: 4,
|
|
ParentID: -1,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Resource: Resource{
|
|
ServiceName: "serviceNameB",
|
|
},
|
|
ScopeSpans: []ScopeSpans{
|
|
{
|
|
Spans: []Span{
|
|
{
|
|
SpanID: []byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02},
|
|
ParentSpanID: []byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01},
|
|
StatusCode: 0,
|
|
ParentID: 1,
|
|
NestedSetLeft: 2,
|
|
NestedSetRight: 3,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
/*{
|
|
traceA: sameTrace,
|
|
traceB: sameTrace,
|
|
expectedTotal: 100,
|
|
},*/
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
for _, m := range methods {
|
|
actualTrace, actualTotal, _ := m(tt.traceA, tt.traceB)
|
|
assert.Equal(t, tt.expectedTotal, actualTotal)
|
|
if tt.expectedTrace != nil {
|
|
assert.Equal(t, tt.expectedTrace, actualTrace)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkCombine(b *testing.B) {
|
|
batchCount := 100
|
|
spanCounts := []int{
|
|
100, 1000, 10000,
|
|
}
|
|
|
|
for _, spanCount := range spanCounts {
|
|
b.Run("SpanCount:"+humanize.SI(float64(batchCount*spanCount), ""), func(b *testing.B) {
|
|
id1 := test.ValidTraceID(nil)
|
|
tr1, _ := traceToParquet(&backend.BlockMeta{}, id1, test.MakeTraceWithSpanCount(batchCount, spanCount, id1), nil)
|
|
|
|
id2 := test.ValidTraceID(nil)
|
|
tr2, _ := traceToParquet(&backend.BlockMeta{}, id2, test.MakeTraceWithSpanCount(batchCount, spanCount, id2), nil)
|
|
|
|
b.ResetTimer()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
c := NewCombiner()
|
|
c.ConsumeWithFinal(tr1, false)
|
|
c.ConsumeWithFinal(tr2, true)
|
|
c.Result()
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func BenchmarkSortTrace(b *testing.B) {
|
|
batchCount := 100
|
|
spanCounts := []int{
|
|
100, 1000, 10000,
|
|
}
|
|
|
|
for _, spanCount := range spanCounts {
|
|
b.Run("SpanCount:"+humanize.SI(float64(batchCount*spanCount), ""), func(b *testing.B) {
|
|
id := test.ValidTraceID(nil)
|
|
tr, _ := traceToParquet(&backend.BlockMeta{}, id, test.MakeTraceWithSpanCount(batchCount, spanCount, id), nil)
|
|
|
|
b.ResetTimer()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
SortTrace(tr)
|
|
}
|
|
})
|
|
}
|
|
}
|