milvus/internal/mq/msgstream/wasted_mock_msgstream.go
yah01 081572d31c
Refactor QueryNode (#21625)
Signed-off-by: yah01 <yang.cen@zilliz.com>
Co-authored-by: Congqi Xia <congqi.xia@zilliz.com>
Co-authored-by: aoiasd <zhicheng.yue@zilliz.com>
2023-03-27 00:42:00 +08:00

26 lines
639 B
Go

package msgstream
type WastedMockMsgStream struct {
MsgStream
AsProducerFunc func(channels []string)
BroadcastMarkFunc func(*MsgPack) (map[string][]MessageID, error)
BroadcastFunc func(*MsgPack) error
ChanFunc func() <-chan *MsgPack
}
func NewWastedMockMsgStream() *WastedMockMsgStream {
return &WastedMockMsgStream{}
}
func (m WastedMockMsgStream) AsProducer(channels []string) {
m.AsProducerFunc(channels)
}
func (m WastedMockMsgStream) Broadcast(pack *MsgPack) (map[string][]MessageID, error) {
return m.BroadcastMarkFunc(pack)
}
func (m WastedMockMsgStream) Chan() <-chan *MsgPack {
return m.ChanFunc()
}