mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-05 05:18:52 +08:00
36 lines
898 B
Go
36 lines
898 B
Go
package proxy
|
|
|
|
type getVChannelsFuncType = func(collectionID UniqueID) ([]vChan, error)
|
|
type removeDMLStreamFuncType = func(collectionID UniqueID) error
|
|
|
|
type mockChannelsMgr struct {
|
|
channelsMgr
|
|
getChannelsFunc func(collectionID UniqueID) ([]pChan, error)
|
|
getVChannelsFuncType
|
|
removeDMLStreamFuncType
|
|
}
|
|
|
|
func (m *mockChannelsMgr) getChannels(collectionID UniqueID) ([]pChan, error) {
|
|
if m.getChannelsFunc != nil {
|
|
return m.getChannelsFunc(collectionID)
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
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) {
|
|
if m.removeDMLStreamFuncType != nil {
|
|
m.removeDMLStreamFuncType(collectionID)
|
|
}
|
|
}
|
|
|
|
func newMockChannelsMgr() *mockChannelsMgr {
|
|
return &mockChannelsMgr{}
|
|
}
|