signoz/pkg/modules/tracefunnel/query.go

package tracefunnel

import (
	v3 "github.com/SigNoz/signoz/pkg/query-service/model/v3"
	tracefunnel "github.com/SigNoz/signoz/pkg/types/tracefunnel"
)

func ValidateTraces(funnel *tracefunnel.Funnel, timeRange tracefunnel.TimeRange) (*v3.ClickHouseQuery, error) {
	var query string

	// (todo) sort by order
	funnelSteps := funnel.Steps
	containsErrorT1 := 0
	containsErrorT2 := 0
	containsErrorT3 := 0

	// (todo) prepare funnel step wise whereclause

	if len(funnel.Steps) == 2 && funnelSteps[0].HasErrors {
		containsErrorT1 = 1
	}
	if len(funnel.Steps) == 2 && funnelSteps[1].HasErrors {
		containsErrorT2 = 1
	}
	if len(funnel.Steps) > 2 && funnelSteps[2].HasErrors {
		containsErrorT3 = 1
	}

	if len(funnel.Steps) > 2 {
		query = BuildThreeStepFunnelValidationQuery(
			containsErrorT1,            // containsErrorT1
			containsErrorT2,            // containsErrorT2
			containsErrorT3,            // containsErrorT3
			timeRange.StartTime,        // startTs
			timeRange.EndTime,          // endTs
			funnelSteps[0].ServiceName, // serviceNameT1
			funnelSteps[0].SpanName,    // spanNameT1
			funnelSteps[1].ServiceName, // serviceNameT1
			funnelSteps[1].SpanName,    // spanNameT2
			funnelSteps[2].ServiceName, // serviceNameT1
			funnelSteps[2].SpanName,    // spanNameT3
			"",
			"",
			"",
		)
	} else {
		query = BuildTwoStepFunnelValidationQuery(
			containsErrorT1,            // containsErrorT1
			containsErrorT2,            // containsErrorT2
			timeRange.StartTime,        // startTs
			timeRange.EndTime,          // endTs
			funnelSteps[0].ServiceName, // serviceNameT1
			funnelSteps[0].SpanName,    // spanNameT1
			funnelSteps[1].ServiceName, // serviceNameT1
			funnelSteps[1].SpanName,    // spanNameT2
			"",
			"",
		)
	}

	return &v3.ClickHouseQuery{
		Query: query,
	}, nil
}

func GetFunnelAnalytics(funnel *tracefunnel.Funnel, timeRange tracefunnel.TimeRange, stepStart, stepEnd int64) (*v3.ClickHouseQuery, error) {
	var query string

	funnelSteps := funnel.Steps
	containsErrorT1 := 0
	containsErrorT2 := 0
	containsErrorT3 := 0
	latencyPointerT1 := "start"
	latencyPointerT2 := "start"
	latencyPointerT3 := "start"
	stepStartOrder := 0
	stepEndOrder := 1

	if funnelSteps[0].HasErrors {
		containsErrorT1 = 1
	}
	if funnelSteps[1].HasErrors {
		containsErrorT2 = 1
	}
	if len(funnel.Steps) > 2 && funnelSteps[2].HasErrors {
		containsErrorT3 = 1
	}
	if funnelSteps[0].LatencyPointer != "" {
		latencyPointerT1 = "end"
	}
	if funnelSteps[1].LatencyPointer != "" {
		latencyPointerT1 = "end"
	}
	if len(funnel.Steps) > 2 && funnelSteps[2].LatencyPointer != "" {
		latencyPointerT1 = "end"
	}

	if len(funnel.Steps) > 2 {
		query = BuildThreeStepFunnelOverviewQuery(
			containsErrorT1, // containsErrorT1
			containsErrorT2, // containsErrorT2
			containsErrorT3, // containsErrorT3
			latencyPointerT1,
			latencyPointerT2,
			latencyPointerT3,
			timeRange.StartTime,        // startTs
			timeRange.EndTime,          // endTs
			funnelSteps[0].ServiceName, // serviceNameT1
			funnelSteps[0].SpanName,    // spanNameT1
			funnelSteps[1].ServiceName, // serviceNameT1
			funnelSteps[1].SpanName,    // spanNameT2
			funnelSteps[2].ServiceName, // serviceNameT1
			funnelSteps[2].SpanName,    // spanNameT3
			"",
			"",
			"",
			//"http_method = 'POST'",         // clauseStep1
			//"response_status_code = '500'", // clauseStep2
			//"db_operation = 'SELECT'",      // clauseStep3
		)
	} else {
		if stepStart != stepEnd {
			stepStartOrder = int(stepStart) - 1
			stepEndOrder = int(stepEnd) - 1
			if funnelSteps[stepStartOrder].HasErrors {
				containsErrorT1 = 1
			}
			if funnelSteps[stepEndOrder].HasErrors {
				containsErrorT2 = 1
			}
		}
		query = BuildTwoStepFunnelOverviewQuery(
			containsErrorT1, // containsErrorT1
			containsErrorT2, // containsErrorT2
			latencyPointerT1,
			latencyPointerT2,
			timeRange.StartTime,                     // startTs
			timeRange.EndTime,                       // endTs
			funnelSteps[stepStartOrder].ServiceName, // serviceNameT1
			funnelSteps[stepStartOrder].SpanName,    // spanNameT1
			funnelSteps[stepEndOrder].ServiceName,   // serviceNameT1
			funnelSteps[stepEndOrder].SpanName,      // spanNameT2
			"",
			"",
		)
	}
	return &v3.ClickHouseQuery{Query: query}, nil
}

func GetStepAnalytics(funnel *tracefunnel.Funnel, timeRange tracefunnel.TimeRange) (*v3.ClickHouseQuery, error) {

	var query string

	funnelSteps := funnel.Steps
	containsErrorT1 := 0
	containsErrorT2 := 0
	containsErrorT3 := 0

	if len(funnel.Steps) == 2 && funnelSteps[0].HasErrors {
		containsErrorT1 = 1
	}
	if len(funnel.Steps) == 2 && funnelSteps[1].HasErrors {
		containsErrorT2 = 1
	}
	if len(funnel.Steps) > 2 && funnelSteps[2].HasErrors {
		containsErrorT3 = 1
	}

	if len(funnel.Steps) > 2 {
		query = BuildThreeStepFunnelCountQuery(
			containsErrorT1,            // containsErrorT1
			containsErrorT2,            // containsErrorT2
			containsErrorT3,            // containsErrorT3
			timeRange.StartTime,        // startTs
			timeRange.EndTime,          // endTs
			funnelSteps[0].ServiceName, // serviceNameT1
			funnelSteps[0].SpanName,    // spanNameT1
			funnelSteps[1].ServiceName, // serviceNameT1
			funnelSteps[1].SpanName,    // spanNameT2
			funnelSteps[2].ServiceName, // serviceNameT1
			funnelSteps[2].SpanName,    // spanNameT3
			"",
			"",
			"",
		)
	} else {
		query = BuildTwoStepFunnelCountQuery(
			containsErrorT1,            // containsErrorT1
			containsErrorT2,            // containsErrorT2
			timeRange.StartTime,        // startTs
			timeRange.EndTime,          // endTs
			funnelSteps[0].ServiceName, // serviceNameT1
			funnelSteps[0].SpanName,    // spanNameT1
			funnelSteps[1].ServiceName, // serviceNameT1
			funnelSteps[1].SpanName,    // spanNameT2
			"",
			"",
		)
	}

	return &v3.ClickHouseQuery{
		Query: query,
	}, nil
}

func GetSlowestTraces(funnel *tracefunnel.Funnel, timeRange tracefunnel.TimeRange, stepStart, stepEnd int64) (*v3.ClickHouseQuery, error) {

	funnelSteps := funnel.Steps
	containsErrorT1 := 0
	containsErrorT2 := 0
	stepStartOrder := 0
	stepEndOrder := 1

	if stepStart != stepEnd {
		stepStartOrder = int(stepStart) - 1
		stepEndOrder = int(stepEnd) - 1
		if funnelSteps[stepStartOrder].HasErrors {
			containsErrorT1 = 1
		}
		if funnelSteps[stepEndOrder].HasErrors {
			containsErrorT2 = 1
		}
	}

	query := BuildTwoStepFunnelTopSlowTracesQuery(
		containsErrorT1,                         // containsErrorT1
		containsErrorT2,                         // containsErrorT2
		timeRange.StartTime,                     // startTs
		timeRange.EndTime,                       // endTs
		funnelSteps[stepStartOrder].ServiceName, // serviceNameT1
		funnelSteps[stepStartOrder].SpanName,    // spanNameT1
		funnelSteps[stepEndOrder].ServiceName,   // serviceNameT1
		funnelSteps[stepEndOrder].SpanName,      // spanNameT2
		"",
		"",
	)
	return &v3.ClickHouseQuery{Query: query}, nil
}

func GetErroredTraces(funnel *tracefunnel.Funnel, timeRange tracefunnel.TimeRange, stepStart, stepEnd int64) (*v3.ClickHouseQuery, error) {

	funnelSteps := funnel.Steps
	containsErrorT1 := 0
	containsErrorT2 := 0
	stepStartOrder := 0
	stepEndOrder := 1

	// (todo): check the logic
	if stepStart != stepEnd {
		stepStartOrder = int(stepStart) - 1
		stepEndOrder = int(stepEnd) - 1
		if funnelSteps[stepStartOrder].HasErrors {
			containsErrorT1 = 1
		}
		if funnelSteps[stepEndOrder].HasErrors {
			containsErrorT2 = 1
		}
	}

	query := BuildTwoStepFunnelTopSlowErrorTracesQuery(
		containsErrorT1,                         // containsErrorT1
		containsErrorT2,                         // containsErrorT2
		timeRange.StartTime,                     // startTs
		timeRange.EndTime,                       // endTs
		funnelSteps[stepStartOrder].ServiceName, // serviceNameT1
		funnelSteps[stepStartOrder].SpanName,    // spanNameT1
		funnelSteps[stepEndOrder].ServiceName,   // serviceNameT1
		funnelSteps[stepEndOrder].SpanName,      // spanNameT2
		"",
		"",
	)
	return &v3.ClickHouseQuery{Query: query}, nil
}