yukun 432b53149b Refactor msgstream
Signed-off-by: yukun <kun.yu@zilliz.com>
2021-03-26 20:10:11 +08:00

36 lines
736 B
Go

package rocksmq
import (
"strconv"
"github.com/zilliztech/milvus-distributed/internal/msgstream/client"
"github.com/zilliztech/milvus-distributed/internal/util/rocksmq/client/rocksmq"
)
type rmqConsumer struct {
c rocksmq.Consumer
msgChannel chan client.ConsumerMessage
}
func (rc *rmqConsumer) Subscription() string {
return rc.c.Subscription()
}
func (rc *rmqConsumer) Chan() <-chan client.ConsumerMessage {
return rc.msgChannel
}
func (rc *rmqConsumer) Seek(id client.MessageID) error {
msgID, err := strconv.ParseInt(string(id.Serialize()), 10, 64)
if err != nil {
return err
}
return rc.c.Seek(msgID)
}
func (rc *rmqConsumer) Ack(message client.ConsumerMessage) {
}
func (rc *rmqConsumer) Close() {
}