mirror of
https://github.com/SigNoz/signoz.git
synced 2025-12-17 15:36:48 +00:00
552 lines
33 KiB
Go
552 lines
33 KiB
Go
|
package telemetrytraces
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/SigNoz/signoz/pkg/instrumentation/instrumentationtest"
|
||
|
|
"github.com/SigNoz/signoz/pkg/querybuilder"
|
||
|
|
qbtypes "github.com/SigNoz/signoz/pkg/types/querybuildertypes/querybuildertypesv5"
|
||
|
|
"github.com/SigNoz/signoz/pkg/types/telemetrytypes"
|
||
|
|
"github.com/SigNoz/signoz/pkg/types/telemetrytypes/telemetrytypestest"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestTraceOperatorStatementBuilder(t *testing.T) {
|
||
|
|
cases := []struct {
|
||
|
|
name string
|
||
|
|
requestType qbtypes.RequestType
|
||
|
|
operator qbtypes.QueryBuilderTraceOperator
|
||
|
|
compositeQuery *qbtypes.CompositeQuery
|
||
|
|
expected qbtypes.Statement
|
||
|
|
expectedErr error
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "simple direct descendant operator",
|
||
|
|
requestType: qbtypes.RequestTypeRaw,
|
||
|
|
operator: qbtypes.QueryBuilderTraceOperator{
|
||
|
|
Expression: "A => B",
|
||
|
|
SelectFields: []telemetrytypes.TelemetryFieldKey{
|
||
|
|
{
|
||
|
|
Name: "service.name",
|
||
|
|
FieldContext: telemetrytypes.FieldContextResource,
|
||
|
|
FieldDataType: telemetrytypes.FieldDataTypeString,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Name: "name",
|
||
|
|
FieldContext: telemetrytypes.FieldContextSpan,
|
||
|
|
FieldDataType: telemetrytypes.FieldDataTypeString,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
Limit: 10,
|
||
|
|
},
|
||
|
|
compositeQuery: &qbtypes.CompositeQuery{
|
||
|
|
Queries: []qbtypes.QueryEnvelope{
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "A",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
Filter: &qbtypes.Filter{
|
||
|
|
Expression: "service.name = 'frontend'",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "B",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
Filter: &qbtypes.Filter{
|
||
|
|
Expression: "service.name = 'backend'",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
expected: qbtypes.Statement{
|
||
|
|
Query: "WITH toDateTime64(1747947419000000000, 9) AS t_from, toDateTime64(1747983448000000000, 9) AS t_to, 1747945619 AS bucket_from, 1747983448 AS bucket_to, all_spans AS (SELECT *, resource_string_service$$name AS `service.name` FROM signoz_traces.distributed_signoz_index_v3 WHERE timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ?), __resource_filter_A AS (SELECT fingerprint FROM signoz_traces.distributed_traces_v3_resource WHERE (simpleJSONExtractString(labels, 'service.name') = ? AND labels LIKE ? AND labels LIKE ?) AND seen_at_ts_bucket_start >= ? AND seen_at_ts_bucket_start <= ?), A AS (SELECT * FROM signoz_traces.distributed_signoz_index_v3 WHERE resource_fingerprint GLOBAL IN (SELECT fingerprint FROM __resource_filter_A) AND timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ? AND true), __resource_filter_B AS (SELECT fingerprint FROM signoz_traces.distributed_traces_v3_resource WHERE (simpleJSONExtractString(labels, 'service.name') = ? AND labels LIKE ? AND labels LIKE ?) AND seen_at_ts_bucket_start >= ? AND seen_at_ts_bucket_start <= ?), B AS (SELECT * FROM signoz_traces.distributed_signoz_index_v3 WHERE resource_fingerprint GLOBAL IN (SELECT fingerprint FROM __resource_filter_B) AND timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ? AND true), A_DIR_DESC_B AS (SELECT p.* FROM A AS p INNER JOIN B AS c ON p.trace_id = c.trace_id AND p.span_id = c.parent_span_id) SELECT timestamp, trace_id, span_id, name, duration_nano, parent_span_id, resources_string['service.name'] AS `service.name` FROM A_DIR_DESC_B ORDER BY timestamp DESC LIMIT ? SETTINGS distributed_product_mode='allow', max_memory_usage=10000000000",
|
||
|
|
Args: []any{"1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), "frontend", "%service.name%", "%service.name\":\"frontend%", uint64(1747945619), uint64(1747983448), "1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), "backend", "%service.name%", "%service.name\":\"backend%", uint64(1747945619), uint64(1747983448), "1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), 10},
|
||
|
|
},
|
||
|
|
expectedErr: nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "indirect descendant operator",
|
||
|
|
requestType: qbtypes.RequestTypeRaw,
|
||
|
|
operator: qbtypes.QueryBuilderTraceOperator{
|
||
|
|
Expression: "A -> B",
|
||
|
|
Limit: 5,
|
||
|
|
},
|
||
|
|
compositeQuery: &qbtypes.CompositeQuery{
|
||
|
|
Queries: []qbtypes.QueryEnvelope{
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "A",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
Filter: &qbtypes.Filter{
|
||
|
|
Expression: "service.name = 'gateway'",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "B",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
Filter: &qbtypes.Filter{
|
||
|
|
Expression: "service.name = 'database'",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
expected: qbtypes.Statement{
|
||
|
|
Query: "WITH toDateTime64(1747947419000000000, 9) AS t_from, toDateTime64(1747983448000000000, 9) AS t_to, 1747945619 AS bucket_from, 1747983448 AS bucket_to, all_spans AS (SELECT *, resource_string_service$$name AS `service.name` FROM signoz_traces.distributed_signoz_index_v3 WHERE timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ?), __resource_filter_A AS (SELECT fingerprint FROM signoz_traces.distributed_traces_v3_resource WHERE (simpleJSONExtractString(labels, 'service.name') = ? AND labels LIKE ? AND labels LIKE ?) AND seen_at_ts_bucket_start >= ? AND seen_at_ts_bucket_start <= ?), A AS (SELECT * FROM signoz_traces.distributed_signoz_index_v3 WHERE resource_fingerprint GLOBAL IN (SELECT fingerprint FROM __resource_filter_A) AND timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ? AND true), __resource_filter_B AS (SELECT fingerprint FROM signoz_traces.distributed_traces_v3_resource WHERE (simpleJSONExtractString(labels, 'service.name') = ? AND labels LIKE ? AND labels LIKE ?) AND seen_at_ts_bucket_start >= ? AND seen_at_ts_bucket_start <= ?), B AS (SELECT * FROM signoz_traces.distributed_signoz_index_v3 WHERE resource_fingerprint GLOBAL IN (SELECT fingerprint FROM __resource_filter_B) AND timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ? AND true), A_INDIR_DESC_B AS (WITH RECURSIVE up AS (SELECT d.trace_id, d.span_id, d.parent_span_id, 0 AS depth FROM B AS d UNION ALL SELECT p.trace_id, p.span_id, p.parent_span_id, up.depth + 1 FROM all_spans AS p JOIN up ON p.trace_id = up.trace_id AND p.span_id = up.parent_span_id WHERE up.depth < 100) SELECT DISTINCT a.* FROM A AS a GLOBAL INNER JOIN (SELECT DISTINCT trace_id, span_id FROM up WHERE depth > 0 ) AS ancestors ON ancestors.trace_id = a.trace_id AND ancestors.span_id = a.span_id) SELECT timestamp, trace_id, span_id, name, duration_nano, parent_span_id FROM A_INDIR_DESC_B ORDER BY timestamp DESC LIMIT ? SETTINGS distributed_product_mode='allow', max_memory_usage=10000000000",
|
||
|
|
Args: []any{"1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), "gateway", "%service.name%", "%service.name\":\"gateway%", uint64(1747945619), uint64(1747983448), "1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), "database", "%service.name%", "%service.name\":\"database%", uint64(1747945619), uint64(1747983448), "1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), 5},
|
||
|
|
},
|
||
|
|
expectedErr: nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "AND operator",
|
||
|
|
requestType: qbtypes.RequestTypeRaw,
|
||
|
|
operator: qbtypes.QueryBuilderTraceOperator{
|
||
|
|
Expression: "A && B",
|
||
|
|
Limit: 15,
|
||
|
|
},
|
||
|
|
compositeQuery: &qbtypes.CompositeQuery{
|
||
|
|
Queries: []qbtypes.QueryEnvelope{
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "A",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
Filter: &qbtypes.Filter{
|
||
|
|
Expression: "service.name = 'frontend'",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "B",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
Filter: &qbtypes.Filter{
|
||
|
|
Expression: "service.name = 'backend'",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
expected: qbtypes.Statement{
|
||
|
|
Query: "WITH toDateTime64(1747947419000000000, 9) AS t_from, toDateTime64(1747983448000000000, 9) AS t_to, 1747945619 AS bucket_from, 1747983448 AS bucket_to, all_spans AS (SELECT *, resource_string_service$$name AS `service.name` FROM signoz_traces.distributed_signoz_index_v3 WHERE timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ?), __resource_filter_A AS (SELECT fingerprint FROM signoz_traces.distributed_traces_v3_resource WHERE (simpleJSONExtractString(labels, 'service.name') = ? AND labels LIKE ? AND labels LIKE ?) AND seen_at_ts_bucket_start >= ? AND seen_at_ts_bucket_start <= ?), A AS (SELECT * FROM signoz_traces.distributed_signoz_index_v3 WHERE resource_fingerprint GLOBAL IN (SELECT fingerprint FROM __resource_filter_A) AND timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ? AND true), __resource_filter_B AS (SELECT fingerprint FROM signoz_traces.distributed_traces_v3_resource WHERE (simpleJSONExtractString(labels, 'service.name') = ? AND labels LIKE ? AND labels LIKE ?) AND seen_at_ts_bucket_start >= ? AND seen_at_ts_bucket_start <= ?), B AS (SELECT * FROM signoz_traces.distributed_signoz_index_v3 WHERE resource_fingerprint GLOBAL IN (SELECT fingerprint FROM __resource_filter_B) AND timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ? AND true), A_AND_B AS (SELECT l.* FROM A AS l INNER JOIN B AS r ON l.trace_id = r.trace_id) SELECT timestamp, trace_id, span_id, name, duration_nano, parent_span_id FROM A_AND_B ORDER BY timestamp DESC LIMIT ? SETTINGS distributed_product_mode='allow', max_memory_usage=10000000000",
|
||
|
|
Args: []any{"1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), "frontend", "%service.name%", "%service.name\":\"frontend%", uint64(1747945619), uint64(1747983448), "1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), "backend", "%service.name%", "%service.name\":\"backend%", uint64(1747945619), uint64(1747983448), "1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), 15},
|
||
|
|
},
|
||
|
|
expectedErr: nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "OR operator",
|
||
|
|
requestType: qbtypes.RequestTypeRaw,
|
||
|
|
operator: qbtypes.QueryBuilderTraceOperator{
|
||
|
|
Expression: "A || B",
|
||
|
|
Limit: 20,
|
||
|
|
},
|
||
|
|
compositeQuery: &qbtypes.CompositeQuery{
|
||
|
|
Queries: []qbtypes.QueryEnvelope{
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "A",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
Filter: &qbtypes.Filter{
|
||
|
|
Expression: "service.name = 'frontend'",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "B",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
Filter: &qbtypes.Filter{
|
||
|
|
Expression: "service.name = 'backend'",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
expected: qbtypes.Statement{
|
||
|
|
Query: "WITH toDateTime64(1747947419000000000, 9) AS t_from, toDateTime64(1747983448000000000, 9) AS t_to, 1747945619 AS bucket_from, 1747983448 AS bucket_to, all_spans AS (SELECT *, resource_string_service$$name AS `service.name` FROM signoz_traces.distributed_signoz_index_v3 WHERE timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ?), __resource_filter_A AS (SELECT fingerprint FROM signoz_traces.distributed_traces_v3_resource WHERE (simpleJSONExtractString(labels, 'service.name') = ? AND labels LIKE ? AND labels LIKE ?) AND seen_at_ts_bucket_start >= ? AND seen_at_ts_bucket_start <= ?), A AS (SELECT * FROM signoz_traces.distributed_signoz_index_v3 WHERE resource_fingerprint GLOBAL IN (SELECT fingerprint FROM __resource_filter_A) AND timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ? AND true), __resource_filter_B AS (SELECT fingerprint FROM signoz_traces.distributed_traces_v3_resource WHERE (simpleJSONExtractString(labels, 'service.name') = ? AND labels LIKE ? AND labels LIKE ?) AND seen_at_ts_bucket_start >= ? AND seen_at_ts_bucket_start <= ?), B AS (SELECT * FROM signoz_traces.distributed_signoz_index_v3 WHERE resource_fingerprint GLOBAL IN (SELECT fingerprint FROM __resource_filter_B) AND timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ? AND true), A_OR_B AS (SELECT * FROM A UNION DISTINCT SELECT * FROM B) SELECT timestamp, trace_id, span_id, name, duration_nano, parent_span_id FROM A_OR_B ORDER BY timestamp DESC LIMIT ? SETTINGS distributed_product_mode='allow', max_memory_usage=10000000000",
|
||
|
|
Args: []any{"1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), "frontend", "%service.name%", "%service.name\":\"frontend%", uint64(1747945619), uint64(1747983448), "1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), "backend", "%service.name%", "%service.name\":\"backend%", uint64(1747945619), uint64(1747983448), "1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), 20},
|
||
|
|
},
|
||
|
|
expectedErr: nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "NOT operator",
|
||
|
|
requestType: qbtypes.RequestTypeRaw,
|
||
|
|
operator: qbtypes.QueryBuilderTraceOperator{
|
||
|
|
Expression: "A NOT B",
|
||
|
|
Limit: 10,
|
||
|
|
},
|
||
|
|
compositeQuery: &qbtypes.CompositeQuery{
|
||
|
|
Queries: []qbtypes.QueryEnvelope{
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "A",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
Filter: &qbtypes.Filter{
|
||
|
|
Expression: "service.name = 'frontend'",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "B",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
Filter: &qbtypes.Filter{
|
||
|
|
Expression: "service.name = 'backend'",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
expected: qbtypes.Statement{
|
||
|
|
Query: "WITH toDateTime64(1747947419000000000, 9) AS t_from, toDateTime64(1747983448000000000, 9) AS t_to, 1747945619 AS bucket_from, 1747983448 AS bucket_to, all_spans AS (SELECT *, resource_string_service$$name AS `service.name` FROM signoz_traces.distributed_signoz_index_v3 WHERE timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ?), __resource_filter_A AS (SELECT fingerprint FROM signoz_traces.distributed_traces_v3_resource WHERE (simpleJSONExtractString(labels, 'service.name') = ? AND labels LIKE ? AND labels LIKE ?) AND seen_at_ts_bucket_start >= ? AND seen_at_ts_bucket_start <= ?), A AS (SELECT * FROM signoz_traces.distributed_signoz_index_v3 WHERE resource_fingerprint GLOBAL IN (SELECT fingerprint FROM __resource_filter_A) AND timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ? AND true), __resource_filter_B AS (SELECT fingerprint FROM signoz_traces.distributed_traces_v3_resource WHERE (simpleJSONExtractString(labels, 'service.name') = ? AND labels LIKE ? AND labels LIKE ?) AND seen_at_ts_bucket_start >= ? AND seen_at_ts_bucket_start <= ?), B AS (SELECT * FROM signoz_traces.distributed_signoz_index_v3 WHERE resource_fingerprint GLOBAL IN (SELECT fingerprint FROM __resource_filter_B) AND timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ? AND true), A_not_B AS (SELECT l.* FROM A AS l WHERE l.trace_id GLOBAL NOT IN (SELECT DISTINCT trace_id FROM B)) SELECT timestamp, trace_id, span_id, name, duration_nano, parent_span_id FROM A_not_B ORDER BY timestamp DESC LIMIT ? SETTINGS distributed_product_mode='allow', max_memory_usage=10000000000",
|
||
|
|
Args: []any{"1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), "frontend", "%service.name%", "%service.name\":\"frontend%", uint64(1747945619), uint64(1747983448), "1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), "backend", "%service.name%", "%service.name\":\"backend%", uint64(1747945619), uint64(1747983448), "1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), 10},
|
||
|
|
},
|
||
|
|
expectedErr: nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "time series query with aggregations",
|
||
|
|
requestType: qbtypes.RequestTypeTimeSeries,
|
||
|
|
operator: qbtypes.QueryBuilderTraceOperator{
|
||
|
|
Expression: "A => B",
|
||
|
|
StepInterval: qbtypes.Step{Duration: 60 * time.Second},
|
||
|
|
Aggregations: []qbtypes.TraceAggregation{
|
||
|
|
{
|
||
|
|
Expression: "count()",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
GroupBy: []qbtypes.GroupByKey{
|
||
|
|
{
|
||
|
|
TelemetryFieldKey: telemetrytypes.TelemetryFieldKey{
|
||
|
|
Name: "service.name",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
compositeQuery: &qbtypes.CompositeQuery{
|
||
|
|
Queries: []qbtypes.QueryEnvelope{
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "A",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
Filter: &qbtypes.Filter{
|
||
|
|
Expression: "service.name = 'frontend'",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "B",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
Filter: &qbtypes.Filter{
|
||
|
|
Expression: "service.name = 'backend'",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
expected: qbtypes.Statement{
|
||
|
|
Query: "WITH toDateTime64(1747947419000000000, 9) AS t_from, toDateTime64(1747983448000000000, 9) AS t_to, 1747945619 AS bucket_from, 1747983448 AS bucket_to, all_spans AS (SELECT *, resource_string_service$$name AS `service.name` FROM signoz_traces.distributed_signoz_index_v3 WHERE timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ?), __resource_filter_A AS (SELECT fingerprint FROM signoz_traces.distributed_traces_v3_resource WHERE (simpleJSONExtractString(labels, 'service.name') = ? AND labels LIKE ? AND labels LIKE ?) AND seen_at_ts_bucket_start >= ? AND seen_at_ts_bucket_start <= ?), A AS (SELECT * FROM signoz_traces.distributed_signoz_index_v3 WHERE resource_fingerprint GLOBAL IN (SELECT fingerprint FROM __resource_filter_A) AND timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ? AND true), __resource_filter_B AS (SELECT fingerprint FROM signoz_traces.distributed_traces_v3_resource WHERE (simpleJSONExtractString(labels, 'service.name') = ? AND labels LIKE ? AND labels LIKE ?) AND seen_at_ts_bucket_start >= ? AND seen_at_ts_bucket_start <= ?), B AS (SELECT * FROM signoz_traces.distributed_signoz_index_v3 WHERE resource_fingerprint GLOBAL IN (SELECT fingerprint FROM __resource_filter_B) AND timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ? AND true), A_DIR_DESC_B AS (SELECT p.* FROM A AS p INNER JOIN B AS c ON p.trace_id = c.trace_id AND p.span_id = c.parent_span_id) SELECT toStartOfInterval(timestamp, INTERVAL 60 SECOND) AS ts, toString(multiIf(mapContains(resources_string, 'service.name') = ?, resources_string['service.name'], NULL)) AS `service.name`, count() AS __result_0 FROM A_DIR_DESC_B GROUP BY ts, `service.name` ORDER BY ts desc SETTINGS distributed_product_mode='allow', max_memory_usage=10000000000",
|
||
|
|
Args: []any{"1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), "frontend", "%service.name%", "%service.name\":\"frontend%", uint64(1747945619), uint64(1747983448), "1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), "backend", "%service.name%", "%service.name\":\"backend%", uint64(1747945619), uint64(1747983448), "1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), true},
|
||
|
|
},
|
||
|
|
expectedErr: nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "scalar query with aggregation and group by",
|
||
|
|
requestType: qbtypes.RequestTypeScalar,
|
||
|
|
operator: qbtypes.QueryBuilderTraceOperator{
|
||
|
|
Expression: "A && B",
|
||
|
|
Aggregations: []qbtypes.TraceAggregation{
|
||
|
|
{
|
||
|
|
Expression: "avg(duration_nano)",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
GroupBy: []qbtypes.GroupByKey{
|
||
|
|
{
|
||
|
|
TelemetryFieldKey: telemetrytypes.TelemetryFieldKey{
|
||
|
|
Name: "service.name",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
Order: []qbtypes.OrderBy{
|
||
|
|
{
|
||
|
|
Key: qbtypes.OrderByKey{
|
||
|
|
TelemetryFieldKey: telemetrytypes.TelemetryFieldKey{
|
||
|
|
Name: "__result_0",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
Direction: qbtypes.OrderDirectionDesc,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
Limit: 10,
|
||
|
|
},
|
||
|
|
compositeQuery: &qbtypes.CompositeQuery{
|
||
|
|
Queries: []qbtypes.QueryEnvelope{
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "A",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
Filter: &qbtypes.Filter{
|
||
|
|
Expression: "service.name = 'frontend'",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "B",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
Filter: &qbtypes.Filter{
|
||
|
|
Expression: "response_status_code < 400",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
expected: qbtypes.Statement{
|
||
|
|
Query: "WITH toDateTime64(1747947419000000000, 9) AS t_from, toDateTime64(1747983448000000000, 9) AS t_to, 1747945619 AS bucket_from, 1747983448 AS bucket_to, all_spans AS (SELECT *, resource_string_service$$name AS `service.name` FROM signoz_traces.distributed_signoz_index_v3 WHERE timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ?), __resource_filter_A AS (SELECT fingerprint FROM signoz_traces.distributed_traces_v3_resource WHERE (simpleJSONExtractString(labels, 'service.name') = ? AND labels LIKE ? AND labels LIKE ?) AND seen_at_ts_bucket_start >= ? AND seen_at_ts_bucket_start <= ?), A AS (SELECT * FROM signoz_traces.distributed_signoz_index_v3 WHERE resource_fingerprint GLOBAL IN (SELECT fingerprint FROM __resource_filter_A) AND timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ? AND true), __resource_filter_B AS (SELECT fingerprint FROM signoz_traces.distributed_traces_v3_resource WHERE true AND seen_at_ts_bucket_start >= ? AND seen_at_ts_bucket_start <= ?), B AS (SELECT * FROM signoz_traces.distributed_signoz_index_v3 WHERE resource_fingerprint GLOBAL IN (SELECT fingerprint FROM __resource_filter_B) AND timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ? AND toFloat64(response_status_code) < ?), A_AND_B AS (SELECT l.* FROM A AS l INNER JOIN B AS r ON l.trace_id = r.trace_id) SELECT toString(multiIf(mapContains(resources_string, 'service.name') = ?, resources_string['service.name'], NULL)) AS `service.name`, avg(multiIf(duration_nano <> ?, duration_nano, NULL)) AS __result_0 FROM A_AND_B GROUP BY `service.name` ORDER BY __result_0 desc SETTINGS distributed_product_mode='allow', max_memory_usage=10000000000",
|
||
|
|
Args: []any{"1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), "frontend", "%service.name%", "%service.name\":\"frontend%", uint64(1747945619), uint64(1747983448), "1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), uint64(1747945619), uint64(1747983448), "1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), float64(400), true, 0},
|
||
|
|
},
|
||
|
|
expectedErr: nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "complex nested expression",
|
||
|
|
requestType: qbtypes.RequestTypeRaw,
|
||
|
|
operator: qbtypes.QueryBuilderTraceOperator{
|
||
|
|
Expression: "(A => B) && (C => D)",
|
||
|
|
Limit: 5,
|
||
|
|
},
|
||
|
|
compositeQuery: &qbtypes.CompositeQuery{
|
||
|
|
Queries: []qbtypes.QueryEnvelope{
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "A",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
Filter: &qbtypes.Filter{
|
||
|
|
Expression: "service.name = 'frontend'",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "B",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
Filter: &qbtypes.Filter{
|
||
|
|
Expression: "service.name = 'backend'",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "C",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
Filter: &qbtypes.Filter{
|
||
|
|
Expression: "service.name = 'auth'",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "D",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
Filter: &qbtypes.Filter{
|
||
|
|
Expression: "service.name = 'database'",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
expected: qbtypes.Statement{
|
||
|
|
Query: "WITH toDateTime64(1747947419000000000, 9) AS t_from, toDateTime64(1747983448000000000, 9) AS t_to, 1747945619 AS bucket_from, 1747983448 AS bucket_to, all_spans AS (SELECT *, resource_string_service$$name AS `service.name` FROM signoz_traces.distributed_signoz_index_v3 WHERE timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ?), __resource_filter_A AS (SELECT fingerprint FROM signoz_traces.distributed_traces_v3_resource WHERE (simpleJSONExtractString(labels, 'service.name') = ? AND labels LIKE ? AND labels LIKE ?) AND seen_at_ts_bucket_start >= ? AND seen_at_ts_bucket_start <= ?), A AS (SELECT * FROM signoz_traces.distributed_signoz_index_v3 WHERE resource_fingerprint GLOBAL IN (SELECT fingerprint FROM __resource_filter_A) AND timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ? AND true), __resource_filter_B AS (SELECT fingerprint FROM signoz_traces.distributed_traces_v3_resource WHERE (simpleJSONExtractString(labels, 'service.name') = ? AND labels LIKE ? AND labels LIKE ?) AND seen_at_ts_bucket_start >= ? AND seen_at_ts_bucket_start <= ?), B AS (SELECT * FROM signoz_traces.distributed_signoz_index_v3 WHERE resource_fingerprint GLOBAL IN (SELECT fingerprint FROM __resource_filter_B) AND timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ? AND true), A_DIR_DESC_B AS (SELECT p.* FROM A AS p INNER JOIN B AS c ON p.trace_id = c.trace_id AND p.span_id = c.parent_span_id), __resource_filter_C AS (SELECT fingerprint FROM signoz_traces.distributed_traces_v3_resource WHERE (simpleJSONExtractString(labels, 'service.name') = ? AND labels LIKE ? AND labels LIKE ?) AND seen_at_ts_bucket_start >= ? AND seen_at_ts_bucket_start <= ?), C AS (SELECT * FROM signoz_traces.distributed_signoz_index_v3 WHERE resource_fingerprint GLOBAL IN (SELECT fingerprint FROM __resource_filter_C) AND timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ? AND true), __resource_filter_D AS (SELECT fingerprint FROM signoz_traces.distributed_traces_v3_resource WHERE (simpleJSONExtractString(labels, 'service.name') = ? AND labels LIKE ? AND labels LIKE ?) AND seen_at_ts_bucket_start >= ? AND seen_at_ts_bucket_start <= ?), D AS (SELECT * FROM signoz_traces.distributed_signoz_index_v3 WHERE resource_fingerprint GLOBAL IN (SELECT fingerprint FROM __resource_filter_D) AND timestamp >= ? AND timestamp < ? AND ts_bucket_start >= ? AND ts_bucket_start <= ? AND true), C_DIR_DESC_D AS (SELECT p.* FROM C AS p INNER JOIN D AS c ON p.trace_id = c.trace_id AND p.span_id = c.parent_span_id), A_DIR_DESC_B_AND_C_DIR_DESC_D AS (SELECT l.* FROM A_DIR_DESC_B AS l INNER JOIN C_DIR_DESC_D AS r ON l.trace_id = r.trace_id) SELECT timestamp, trace_id, span_id, name, duration_nano, parent_span_id FROM A_DIR_DESC_B_AND_C_DIR_DESC_D ORDER BY timestamp DESC LIMIT ? SETTINGS distributed_product_mode='allow', max_memory_usage=10000000000",
|
||
|
|
Args: []any{"1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), "frontend", "%service.name%", "%service.name\":\"frontend%", uint64(1747945619), uint64(1747983448), "1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), "backend", "%service.name%", "%service.name\":\"backend%", uint64(1747945619), uint64(1747983448), "1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), "auth", "%service.name%", "%service.name\":\"auth%", uint64(1747945619), uint64(1747983448), "1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), "database", "%service.name%", "%service.name\":\"database%", uint64(1747945619), uint64(1747983448), "1747947419000000000", "1747983448000000000", uint64(1747945619), uint64(1747983448), 5},
|
||
|
|
},
|
||
|
|
expectedErr: nil,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
fm := NewFieldMapper()
|
||
|
|
cb := NewConditionBuilder(fm)
|
||
|
|
mockMetadataStore := telemetrytypestest.NewMockMetadataStore()
|
||
|
|
mockMetadataStore.KeysMap = buildCompleteFieldKeyMap()
|
||
|
|
aggExprRewriter := querybuilder.NewAggExprRewriter(instrumentationtest.New().ToProviderSettings(), nil, fm, cb, "", nil)
|
||
|
|
|
||
|
|
resourceFilterStmtBuilder := resourceFilterStmtBuilder()
|
||
|
|
traceStmtBuilder := NewTraceQueryStatementBuilder(
|
||
|
|
instrumentationtest.New().ToProviderSettings(),
|
||
|
|
mockMetadataStore,
|
||
|
|
fm,
|
||
|
|
cb,
|
||
|
|
resourceFilterStmtBuilder,
|
||
|
|
aggExprRewriter,
|
||
|
|
nil,
|
||
|
|
)
|
||
|
|
|
||
|
|
statementBuilder := NewTraceOperatorStatementBuilder(
|
||
|
|
instrumentationtest.New().ToProviderSettings(),
|
||
|
|
mockMetadataStore,
|
||
|
|
fm,
|
||
|
|
cb,
|
||
|
|
traceStmtBuilder,
|
||
|
|
resourceFilterStmtBuilder,
|
||
|
|
aggExprRewriter,
|
||
|
|
)
|
||
|
|
|
||
|
|
for _, c := range cases {
|
||
|
|
t.Run(c.name, func(t *testing.T) {
|
||
|
|
// Parse the operator expression
|
||
|
|
err := c.operator.ParseExpression()
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
q, err := statementBuilder.Build(
|
||
|
|
context.Background(),
|
||
|
|
1747947419000,
|
||
|
|
1747983448000,
|
||
|
|
c.requestType,
|
||
|
|
c.operator,
|
||
|
|
c.compositeQuery,
|
||
|
|
)
|
||
|
|
|
||
|
|
if c.expectedErr != nil {
|
||
|
|
require.Error(t, err)
|
||
|
|
require.Contains(t, err.Error(), c.expectedErr.Error())
|
||
|
|
} else {
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, c.expected.Query, q.Query)
|
||
|
|
require.Equal(t, c.expected.Args, q.Args)
|
||
|
|
require.Equal(t, c.expected.Warnings, q.Warnings)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestTraceOperatorStatementBuilderErrors(t *testing.T) {
|
||
|
|
cases := []struct {
|
||
|
|
name string
|
||
|
|
operator qbtypes.QueryBuilderTraceOperator
|
||
|
|
compositeQuery *qbtypes.CompositeQuery
|
||
|
|
expectedErr string
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "missing referenced query",
|
||
|
|
operator: qbtypes.QueryBuilderTraceOperator{
|
||
|
|
Expression: "A => B",
|
||
|
|
},
|
||
|
|
compositeQuery: &qbtypes.CompositeQuery{
|
||
|
|
Queries: []qbtypes.QueryEnvelope{
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "A",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
// Missing query B
|
||
|
|
},
|
||
|
|
},
|
||
|
|
expectedErr: "referenced query 'B' not found",
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "nil composite query",
|
||
|
|
operator: qbtypes.QueryBuilderTraceOperator{
|
||
|
|
Expression: "A => B",
|
||
|
|
},
|
||
|
|
compositeQuery: nil,
|
||
|
|
expectedErr: "compositeQuery cannot be nil",
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "unsupported operator",
|
||
|
|
operator: qbtypes.QueryBuilderTraceOperator{
|
||
|
|
Expression: "A XOR B", // Assuming XOR is not supported
|
||
|
|
},
|
||
|
|
compositeQuery: &qbtypes.CompositeQuery{
|
||
|
|
Queries: []qbtypes.QueryEnvelope{
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "A",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Type: qbtypes.QueryTypeBuilder,
|
||
|
|
Spec: qbtypes.QueryBuilderQuery[qbtypes.TraceAggregation]{
|
||
|
|
Name: "B",
|
||
|
|
Signal: telemetrytypes.SignalTraces,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
expectedErr: "invalid query reference 'A XOR B'",
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
fm := NewFieldMapper()
|
||
|
|
cb := NewConditionBuilder(fm)
|
||
|
|
mockMetadataStore := telemetrytypestest.NewMockMetadataStore()
|
||
|
|
mockMetadataStore.KeysMap = buildCompleteFieldKeyMap()
|
||
|
|
aggExprRewriter := querybuilder.NewAggExprRewriter(instrumentationtest.New().ToProviderSettings(), nil, fm, cb, "", nil)
|
||
|
|
|
||
|
|
resourceFilterStmtBuilder := resourceFilterStmtBuilder()
|
||
|
|
traceStmtBuilder := NewTraceQueryStatementBuilder(
|
||
|
|
instrumentationtest.New().ToProviderSettings(),
|
||
|
|
mockMetadataStore,
|
||
|
|
fm,
|
||
|
|
cb,
|
||
|
|
resourceFilterStmtBuilder,
|
||
|
|
aggExprRewriter,
|
||
|
|
nil,
|
||
|
|
)
|
||
|
|
|
||
|
|
statementBuilder := NewTraceOperatorStatementBuilder(
|
||
|
|
instrumentationtest.New().ToProviderSettings(),
|
||
|
|
mockMetadataStore,
|
||
|
|
fm,
|
||
|
|
cb,
|
||
|
|
traceStmtBuilder,
|
||
|
|
resourceFilterStmtBuilder,
|
||
|
|
aggExprRewriter,
|
||
|
|
)
|
||
|
|
|
||
|
|
for _, c := range cases {
|
||
|
|
t.Run(c.name, func(t *testing.T) {
|
||
|
|
// Parse the operator expression
|
||
|
|
err := c.operator.ParseExpression()
|
||
|
|
if err == nil { // Only proceed if parsing succeeded
|
||
|
|
_, err = statementBuilder.Build(
|
||
|
|
context.Background(),
|
||
|
|
1747947419000,
|
||
|
|
1747983448000,
|
||
|
|
qbtypes.RequestTypeRaw,
|
||
|
|
c.operator,
|
||
|
|
c.compositeQuery,
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
require.Error(t, err)
|
||
|
|
require.Contains(t, err.Error(), c.expectedErr)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|