mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-07-08 22:51:47 +08:00
385 lines
12 KiB
Go
385 lines
12 KiB
Go
package kafka
|
|
|
|
import (
|
|
"fmt"
|
|
"go.signoz.io/signoz/pkg/query-service/common"
|
|
"go.signoz.io/signoz/pkg/query-service/constants"
|
|
v3 "go.signoz.io/signoz/pkg/query-service/model/v3"
|
|
)
|
|
|
|
var defaultStepInterval int64 = 60
|
|
|
|
func BuildQueryRangeParams(messagingQueue *MessagingQueue, queryContext string) (*v3.QueryRangeParamsV3, error) {
|
|
|
|
if constants.KafkaSpanEval == "false" && queryContext == "producer-consumer-eval" {
|
|
return nil, fmt.Errorf("span evaluation feature is disabled and is experimental")
|
|
}
|
|
// ToDo: propagate this through APIs when there are different handlers
|
|
queueType := KafkaQueue
|
|
|
|
chq, err := BuildClickHouseQuery(messagingQueue, queueType, queryContext)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var cq *v3.CompositeQuery
|
|
|
|
cq, err = buildCompositeQuery(chq, queryContext)
|
|
|
|
queryRangeParams := &v3.QueryRangeParamsV3{
|
|
Start: messagingQueue.Start,
|
|
End: messagingQueue.End,
|
|
Step: defaultStepInterval,
|
|
CompositeQuery: cq,
|
|
Version: "v4",
|
|
FormatForWeb: true,
|
|
}
|
|
|
|
return queryRangeParams, nil
|
|
}
|
|
|
|
func buildClickHouseQueryNetwork(messagingQueue *MessagingQueue, queueType string) (*v3.ClickHouseQuery, error) {
|
|
start := messagingQueue.Start
|
|
end := messagingQueue.End
|
|
consumerGroup, ok := messagingQueue.Variables["consumer_group"]
|
|
if !ok {
|
|
return nil, fmt.Errorf("consumer_group not found in the request")
|
|
}
|
|
|
|
partitionID, ok := messagingQueue.Variables["partition"]
|
|
if !ok {
|
|
return nil, fmt.Errorf("partition not found in the request")
|
|
}
|
|
|
|
query := generateNetworkLatencyThroughputSQL(start, end, consumerGroup, partitionID, queueType)
|
|
|
|
return &v3.ClickHouseQuery{
|
|
Query: query,
|
|
}, nil
|
|
}
|
|
|
|
func buildBuilderQueriesProducerBytes(unixMilliStart, unixMilliEnd int64, attributeCache *Clients) (map[string]*v3.BuilderQuery, error) {
|
|
bq := make(map[string]*v3.BuilderQuery)
|
|
queryName := fmt.Sprintf("byte_rate")
|
|
|
|
chq := &v3.BuilderQuery{
|
|
QueryName: queryName,
|
|
StepInterval: common.MinAllowedStepInterval(unixMilliStart, unixMilliEnd),
|
|
DataSource: v3.DataSourceMetrics,
|
|
AggregateAttribute: v3.AttributeKey{
|
|
Key: "kafka_producer_byte_rate",
|
|
DataType: v3.AttributeKeyDataTypeFloat64,
|
|
Type: v3.AttributeKeyType("Gauge"),
|
|
IsColumn: true,
|
|
},
|
|
AggregateOperator: v3.AggregateOperatorAvg,
|
|
Temporality: v3.Unspecified,
|
|
TimeAggregation: v3.TimeAggregationAvg,
|
|
SpaceAggregation: v3.SpaceAggregationAvg,
|
|
Filters: &v3.FilterSet{
|
|
Operator: "AND",
|
|
Items: []v3.FilterItem{
|
|
{
|
|
Key: v3.AttributeKey{
|
|
Key: "service_name",
|
|
Type: v3.AttributeKeyTypeTag,
|
|
DataType: v3.AttributeKeyDataTypeString,
|
|
},
|
|
Operator: v3.FilterOperatorIn,
|
|
Value: attributeCache.ServiceName,
|
|
},
|
|
{
|
|
Key: v3.AttributeKey{
|
|
Key: "topic",
|
|
Type: v3.AttributeKeyTypeTag,
|
|
DataType: v3.AttributeKeyDataTypeString,
|
|
},
|
|
Operator: v3.FilterOperatorIn,
|
|
Value: attributeCache.TopicName,
|
|
},
|
|
},
|
|
},
|
|
Expression: queryName,
|
|
ReduceTo: v3.ReduceToOperatorAvg,
|
|
GroupBy: []v3.AttributeKey{{
|
|
Key: "service_name",
|
|
DataType: v3.AttributeKeyDataTypeString,
|
|
Type: v3.AttributeKeyTypeTag,
|
|
},
|
|
{
|
|
Key: "topic",
|
|
DataType: v3.AttributeKeyDataTypeString,
|
|
Type: v3.AttributeKeyTypeTag,
|
|
},
|
|
},
|
|
}
|
|
bq[queryName] = chq
|
|
return bq, nil
|
|
}
|
|
|
|
func buildBuilderQueriesNetwork(unixMilliStart, unixMilliEnd int64, attributeCache *Clients) (map[string]*v3.BuilderQuery, error) {
|
|
bq := make(map[string]*v3.BuilderQuery)
|
|
queryName := fmt.Sprintf("latency")
|
|
|
|
chq := &v3.BuilderQuery{
|
|
QueryName: queryName,
|
|
StepInterval: common.MinAllowedStepInterval(unixMilliStart, unixMilliEnd),
|
|
DataSource: v3.DataSourceMetrics,
|
|
AggregateAttribute: v3.AttributeKey{
|
|
Key: "kafka_consumer_fetch_latency_avg",
|
|
},
|
|
AggregateOperator: v3.AggregateOperatorAvg,
|
|
Temporality: v3.Unspecified,
|
|
TimeAggregation: v3.TimeAggregationAvg,
|
|
SpaceAggregation: v3.SpaceAggregationAvg,
|
|
Filters: &v3.FilterSet{
|
|
Operator: "AND",
|
|
Items: []v3.FilterItem{
|
|
{
|
|
Key: v3.AttributeKey{
|
|
Key: "service_name",
|
|
Type: v3.AttributeKeyTypeTag,
|
|
DataType: v3.AttributeKeyDataTypeString,
|
|
},
|
|
Operator: v3.FilterOperatorIn,
|
|
Value: attributeCache.ServiceName,
|
|
},
|
|
{
|
|
Key: v3.AttributeKey{
|
|
Key: "client_id",
|
|
Type: v3.AttributeKeyTypeTag,
|
|
DataType: v3.AttributeKeyDataTypeString,
|
|
},
|
|
Operator: v3.FilterOperatorIn,
|
|
Value: attributeCache.ClientID,
|
|
},
|
|
{
|
|
Key: v3.AttributeKey{
|
|
Key: "service_instance_id",
|
|
Type: v3.AttributeKeyTypeTag,
|
|
DataType: v3.AttributeKeyDataTypeString,
|
|
},
|
|
Operator: v3.FilterOperatorIn,
|
|
Value: attributeCache.ServiceInstanceID,
|
|
},
|
|
},
|
|
},
|
|
Expression: queryName,
|
|
ReduceTo: v3.ReduceToOperatorAvg,
|
|
GroupBy: []v3.AttributeKey{{
|
|
Key: "service_name",
|
|
DataType: v3.AttributeKeyDataTypeString,
|
|
Type: v3.AttributeKeyTypeTag,
|
|
},
|
|
{
|
|
Key: "client_id",
|
|
DataType: v3.AttributeKeyDataTypeString,
|
|
Type: v3.AttributeKeyTypeTag,
|
|
},
|
|
{
|
|
Key: "service_instance_id",
|
|
DataType: v3.AttributeKeyDataTypeString,
|
|
Type: v3.AttributeKeyTypeTag,
|
|
},
|
|
},
|
|
}
|
|
bq[queryName] = chq
|
|
return bq, nil
|
|
}
|
|
|
|
func BuildBuilderQueriesKafkaOnboarding(messagingQueue *MessagingQueue) (*v3.QueryRangeParamsV3, error) {
|
|
bq := make(map[string]*v3.BuilderQuery)
|
|
|
|
unixMilliStart := messagingQueue.Start / 1000000
|
|
unixMilliEnd := messagingQueue.End / 1000000
|
|
|
|
buiderQuery := &v3.BuilderQuery{
|
|
QueryName: "fetch_latency",
|
|
StepInterval: common.MinAllowedStepInterval(unixMilliStart, unixMilliEnd),
|
|
DataSource: v3.DataSourceMetrics,
|
|
AggregateAttribute: v3.AttributeKey{
|
|
Key: "kafka_consumer_fetch_latency_avg",
|
|
},
|
|
AggregateOperator: v3.AggregateOperatorCount,
|
|
Temporality: v3.Unspecified,
|
|
TimeAggregation: v3.TimeAggregationCount,
|
|
SpaceAggregation: v3.SpaceAggregationSum,
|
|
Expression: "fetch_latency",
|
|
}
|
|
bq["fetch_latency"] = buiderQuery
|
|
|
|
buiderQuery = &v3.BuilderQuery{
|
|
QueryName: "consumer_lag",
|
|
StepInterval: common.MinAllowedStepInterval(unixMilliStart, unixMilliEnd),
|
|
DataSource: v3.DataSourceMetrics,
|
|
AggregateAttribute: v3.AttributeKey{
|
|
Key: "kafka_consumer_group_lag",
|
|
},
|
|
AggregateOperator: v3.AggregateOperatorCount,
|
|
Temporality: v3.Unspecified,
|
|
TimeAggregation: v3.TimeAggregationCount,
|
|
SpaceAggregation: v3.SpaceAggregationSum,
|
|
Expression: "consumer_lag",
|
|
}
|
|
bq["consumer_lag"] = buiderQuery
|
|
|
|
cq := &v3.CompositeQuery{
|
|
QueryType: v3.QueryTypeBuilder,
|
|
BuilderQueries: bq,
|
|
PanelType: v3.PanelTypeTable,
|
|
}
|
|
|
|
queryRangeParams := &v3.QueryRangeParamsV3{
|
|
Start: unixMilliStart,
|
|
End: unixMilliEnd,
|
|
Step: defaultStepInterval,
|
|
CompositeQuery: cq,
|
|
Version: "v4",
|
|
FormatForWeb: true,
|
|
}
|
|
|
|
return queryRangeParams, nil
|
|
}
|
|
|
|
func BuildQRParamsWithCache(messagingQueue *MessagingQueue, queryContext string, attributeCache *Clients) (*v3.QueryRangeParamsV3, error) {
|
|
|
|
queueType := KafkaQueue
|
|
|
|
unixMilliStart := messagingQueue.Start / 1000000
|
|
unixMilliEnd := messagingQueue.End / 1000000
|
|
|
|
var cq *v3.CompositeQuery
|
|
var err error
|
|
|
|
if queryContext == "throughput" {
|
|
chq, err := buildClickHouseQueryNetwork(messagingQueue, queueType)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
cq, err = buildCompositeQuery(chq, queryContext)
|
|
|
|
} else if queryContext == "fetch-latency" {
|
|
bhq, err := buildBuilderQueriesNetwork(unixMilliStart, unixMilliEnd, attributeCache)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
cq = &v3.CompositeQuery{
|
|
QueryType: v3.QueryTypeBuilder,
|
|
BuilderQueries: bhq,
|
|
PanelType: v3.PanelTypeTable,
|
|
}
|
|
} else if queryContext == "producer-throughput-overview" {
|
|
start := messagingQueue.Start
|
|
end := messagingQueue.End
|
|
query := generateProducerPartitionThroughputSQL(start, end, queueType)
|
|
|
|
cq, err = buildCompositeQuery(&v3.ClickHouseQuery{
|
|
Query: query,
|
|
}, queryContext)
|
|
} else if queryContext == "producer-throughput-overview-byte-rate" {
|
|
bhq, err := buildBuilderQueriesProducerBytes(unixMilliStart, unixMilliEnd, attributeCache)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
cq = &v3.CompositeQuery{
|
|
QueryType: v3.QueryTypeBuilder,
|
|
BuilderQueries: bhq,
|
|
PanelType: v3.PanelTypeTable,
|
|
FillGaps: false,
|
|
}
|
|
}
|
|
|
|
queryRangeParams := &v3.QueryRangeParamsV3{
|
|
Start: unixMilliStart,
|
|
End: unixMilliEnd,
|
|
Step: defaultStepInterval,
|
|
CompositeQuery: cq,
|
|
Version: "v4",
|
|
FormatForWeb: true,
|
|
}
|
|
|
|
return queryRangeParams, err
|
|
}
|
|
|
|
func BuildClickHouseQuery(messagingQueue *MessagingQueue, queueType string, queryContext string) (*v3.ClickHouseQuery, error) {
|
|
start := messagingQueue.Start
|
|
end := messagingQueue.End
|
|
|
|
var topic, partition string
|
|
if queryContext == "producer" ||
|
|
queryContext == "consumer" ||
|
|
queryContext == "consumer_partition_latency" ||
|
|
queryContext == "producer-throughput-details" ||
|
|
queryContext == "consumer-throughput-details" {
|
|
var ok bool
|
|
topic, ok = messagingQueue.Variables["topic"]
|
|
if !ok {
|
|
return nil, fmt.Errorf("invalid type for Topic")
|
|
}
|
|
if !(queryContext == "consumer-throughput-details" || queryContext == "producer-throughput-details") {
|
|
partition, ok = messagingQueue.Variables["partition"]
|
|
if !ok {
|
|
return nil, fmt.Errorf("invalid type for Partition")
|
|
}
|
|
}
|
|
}
|
|
|
|
var query string
|
|
if queryContext == "producer" {
|
|
query = generateProducerSQL(start, end, topic, partition, queueType)
|
|
} else if queryContext == "consumer" {
|
|
consumerGroup, ok := messagingQueue.Variables["consumer_group"]
|
|
if !ok {
|
|
return nil, fmt.Errorf("invalid type for consumer group")
|
|
}
|
|
query = generateConsumerSQL(start, end, topic, partition, consumerGroup, queueType)
|
|
} else if queryContext == "producer-topic-throughput" {
|
|
query = generatePartitionLatencySQL(start, end, queueType)
|
|
} else if queryContext == "consumer_partition_latency" {
|
|
query = generateConsumerPartitionLatencySQL(start, end, topic, partition, queueType)
|
|
} else if queryContext == "producer-throughput-details" {
|
|
svcName, ok := messagingQueue.Variables["service_name"]
|
|
if !ok {
|
|
return nil, fmt.Errorf("invalid type for service")
|
|
}
|
|
query = generateProducerTopicLatencySQL(start, end, topic, svcName, queueType)
|
|
} else if queryContext == "consumer-throughput-overview" {
|
|
query = generateConsumerLatencySQL(start, end, queueType)
|
|
} else if queryContext == "consumer-throughput-details" {
|
|
svcName, ok := messagingQueue.Variables["service_name"]
|
|
if !ok {
|
|
return nil, fmt.Errorf("invalid type for service")
|
|
}
|
|
query = generateConsumerServiceLatencySQL(start, end, topic, svcName, queueType)
|
|
} else if queryContext == "producer-consumer-eval" {
|
|
query = generateProducerConsumerEvalSQL(start, end, queueType, messagingQueue.EvalTime)
|
|
} else if queryContext == "onboard_producers" {
|
|
query = onboardProducersSQL(start, end, queueType)
|
|
} else if queryContext == "onboard_consumers" {
|
|
query = onboardConsumerSQL(start, end, queueType)
|
|
}
|
|
return &v3.ClickHouseQuery{
|
|
Query: query,
|
|
}, nil
|
|
}
|
|
|
|
func buildCompositeQuery(chq *v3.ClickHouseQuery, queryContext string) (*v3.CompositeQuery, error) {
|
|
|
|
if queryContext == "producer-consumer-eval" {
|
|
return &v3.CompositeQuery{
|
|
QueryType: v3.QueryTypeClickHouseSQL,
|
|
ClickHouseQueries: map[string]*v3.ClickHouseQuery{queryContext: chq},
|
|
PanelType: v3.PanelTypeList,
|
|
}, nil
|
|
}
|
|
|
|
return &v3.CompositeQuery{
|
|
QueryType: v3.QueryTypeClickHouseSQL,
|
|
ClickHouseQueries: map[string]*v3.ClickHouseQuery{queryContext: chq},
|
|
PanelType: v3.PanelTypeTable,
|
|
}, nil
|
|
}
|