milvus/pkg/mq/msgstream/mqwrapper/kafka/kafka_message.go
congqixia cb7f2fa6fd
enhance: Use v2 package name for pkg module (#39990)
Related to #39095

https://go.dev/doc/modules/version-numbers

Update pkg version according to golang dep version convention

---------

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
2025-02-22 23:15:58 +08:00

33 lines
663 B
Go

package kafka
import (
"github.com/confluentinc/confluent-kafka-go/kafka"
"github.com/milvus-io/milvus/pkg/v2/mq/common"
)
type kafkaMessage struct {
msg *kafka.Message
}
func (km *kafkaMessage) Topic() string {
return *km.msg.TopicPartition.Topic
}
func (km *kafkaMessage) Properties() map[string]string {
properties := make(map[string]string)
for _, header := range km.msg.Headers {
properties[header.Key] = string(header.Value)
}
return properties
}
func (km *kafkaMessage) Payload() []byte {
return km.msg.Value
}
func (km *kafkaMessage) ID() common.MessageID {
kid := &KafkaID{MessageID: int64(km.msg.TopicPartition.Offset)}
return kid
}