milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Optimize proxy getChannels (#9964) 

Signed-off-by: yudong.cai <yudong.cai@zilliz.com>
pull/10000/head
Cai Yudong 2021-10-15 20:37:04 +08:00 committed by GitHub
parent ae5d2ee9ba
commit dc0269cf4c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
internal/proxy/task.go
View File

@ -1410,15 +1410,17 @@ func (st *searchTask) getChannels() ([]pChan, error) {
return nil, err
}
_, err = st.chMgr.getChannels(collID)
var channels []pChan
channels, err = st.chMgr.getChannels(collID)
if err != nil {
err := st.chMgr.createDMLMsgStream(collID)
if err != nil {
return nil, err
}
return st.chMgr.getChannels(collID)
}
return st.chMgr.getChannels(collID)
return channels, nil
}
func (st *searchTask) getVChannels() ([]vChan, error) {
@ -1427,15 +1429,17 @@ func (st *searchTask) getVChannels() ([]vChan, error) {
return nil, err
}
_, err = st.chMgr.getChannels(collID)
var channels []vChan
channels, err = st.chMgr.getVChannels(collID)
if err != nil {
err := st.chMgr.createDMLMsgStream(collID)
if err != nil {
return nil, err
}
return st.chMgr.getVChannels(collID)
}
return st.chMgr.getVChannels(collID)
return channels, nil
}
func (st *searchTask) PreExecute(ctx context.Context) error {
@ -2129,15 +2133,17 @@ func (qt *queryTask) getChannels() ([]pChan, error) {
return nil, err
}
_, err = qt.chMgr.getChannels(collID)
var channels []pChan
channels, err = qt.chMgr.getChannels(collID)
if err != nil {
err := qt.chMgr.createDMLMsgStream(collID)
if err != nil {
return nil, err
}
return qt.chMgr.getChannels(collID)
}
return qt.chMgr.getChannels(collID)
return channels, nil
}
func (qt *queryTask) getVChannels() ([]vChan, error) {
@ -2146,15 +2152,17 @@ func (qt *queryTask) getVChannels() ([]vChan, error) {
return nil, err
}
_, err = qt.chMgr.getChannels(collID)
var channels []vChan
channels, err = qt.chMgr.getVChannels(collID)
if err != nil {
err := qt.chMgr.createDMLMsgStream(collID)
if err != nil {
return nil, err
}
return qt.chMgr.getVChannels(collID)
}
return qt.chMgr.getVChannels(collID)
return channels, nil
}
func IDs2Expr(fieldName string, ids []int64) string {