mirror of https://github.com/milvus-io/milvus.git
21 lines
436 B
Go
21 lines
436 B
Go
package msgstream
|
|
|
|
import "context"
|
|
|
|
type MockMqFactory struct {
|
|
Factory
|
|
NewMsgStreamFunc func(ctx context.Context) (MsgStream, error)
|
|
}
|
|
|
|
func NewMockMqFactory() *MockMqFactory {
|
|
return &MockMqFactory{}
|
|
}
|
|
|
|
func (m MockMqFactory) NewMsgStream(ctx context.Context) (MsgStream, error) {
|
|
return m.NewMsgStreamFunc(ctx)
|
|
}
|
|
|
|
func (m MockMqFactory) NewTtMsgStream(ctx context.Context) (MsgStream, error) {
|
|
return m.NewMsgStreamFunc(ctx)
|
|
}
|