milvus-io--milvus/pkg/mq/msgstream/wasted_mock_msgstream.go

package msgstream

import "context"

type WastedMockMsgStream struct {
	MsgStream
	AsProducerFunc    func(channels []string)
	BroadcastMarkFunc func(*MsgPack) (map[string][]MessageID, error)
	BroadcastFunc     func(*MsgPack) error
	ChanFunc          func() <-chan *ConsumeMsgPack
}

func NewWastedMockMsgStream() *WastedMockMsgStream {
	return &WastedMockMsgStream{}
}

func (m WastedMockMsgStream) AsProducer(ctx context.Context, channels []string) {
	m.AsProducerFunc(channels)
}

func (m WastedMockMsgStream) Broadcast(ctx context.Context, pack *MsgPack) (map[string][]MessageID, error) {
	return m.BroadcastMarkFunc(pack)
}

func (m WastedMockMsgStream) Chan() <-chan *ConsumeMsgPack {
	return m.ChanFunc()
}