mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-02 20:09:57 +08:00
29 lines
686 B
Go
29 lines
686 B
Go
|
package proxy
|
||
|
|
|
||
|
|
type getVChannelsFuncType = func(collectionID UniqueID) ([]vChan, error)
|
||
|
|
type removeDMLStreamFuncType = func(collectionID UniqueID) error
|
||
|
|
|
||
|
|
type mockChannelsMgr struct {
|
||
|
|
channelsMgr
|
||
|
|
getVChannelsFuncType
|
||
|
|
removeDMLStreamFuncType
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *mockChannelsMgr) getVChannels(collectionID UniqueID) ([]vChan, error) {
|
||
|
|
if m.getVChannelsFuncType != nil {
|
||
|
|
return m.getVChannelsFuncType(collectionID)
|
||
|
|
}
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *mockChannelsMgr) removeDMLStream(collectionID UniqueID) error {
|
||
|
|
if m.removeDMLStreamFuncType != nil {
|
||
|
|
return m.removeDMLStreamFuncType(collectionID)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func newMockChannelsMgr() *mockChannelsMgr {
|
||
|
|
return &mockChannelsMgr{}
|
||
|
|
}
|