Fix query task panicking (#8060)

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
This commit is contained in:
congqixia 2021-09-16 20:03:50 +08:00 committed by GitHub
parent 41cb0f30cb
commit 4fe22bcaaa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -2464,13 +2464,14 @@ func (qt *queryTask) PostExecute(ctx context.Context) error {
}, },
FieldsData: make([]*schemapb.FieldData, 0), FieldsData: make([]*schemapb.FieldData, 0),
} }
for idx, partialRetrieveResult := range retrieveResult { for _, partialRetrieveResult := range retrieveResult {
availableQueryNodeNum++ availableQueryNodeNum++
if partialRetrieveResult.Ids == nil { if partialRetrieveResult.Ids == nil {
reason += "ids is nil\n" reason += "ids is nil\n"
continue continue
} else { } else {
if idx == 0 { // handles initialization, cannot use idx==0 since first result may be empty
if len(qt.result.FieldsData) == 0 {
qt.result.FieldsData = append(qt.result.FieldsData, partialRetrieveResult.FieldsData...) qt.result.FieldsData = append(qt.result.FieldsData, partialRetrieveResult.FieldsData...)
} else { } else {
for k, fieldData := range partialRetrieveResult.FieldsData { for k, fieldData := range partialRetrieveResult.FieldsData {