mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-07 17:48:29 +08:00
b5f178f22b
Signed-off-by: lixinguo <xinguo.li@zilliz.com> Signed-off-by: lixinguo <xinguo.li@zilliz.com> Co-authored-by: lixinguo <xinguo.li@zilliz.com>
32 lines
680 B
Go
32 lines
680 B
Go
package kafka
|
|
|
|
import (
|
|
"github.com/confluentinc/confluent-kafka-go/kafka"
|
|
"github.com/milvus-io/milvus/internal/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
|
|
}
|