mirror of https://github.com/milvus-io/milvus.git
17 lines
321 B
Go
17 lines
321 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)
|
||
|
|
}
|