mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-08 10:08:42 +08:00
33 lines
676 B
Go
33 lines
676 B
Go
package kafka
|
|
|
|
import (
|
|
"github.com/confluentinc/confluent-kafka-go/kafka"
|
|
|
|
"github.com/milvus-io/milvus/pkg/mq/msgstream/mqwrapper"
|
|
)
|
|
|
|
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() mqwrapper.MessageID {
|
|
kid := &kafkaID{messageID: int64(km.msg.TopicPartition.Offset)}
|
|
return kid
|
|
}
|