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

Fix fields check 

Signed-off-by: bigsheeper <yihao.dai@zilliz.com>
pull/4973/head^2
bigsheeper 2021-02-26 11:15:44 +08:00 committed by yefu.chen
parent d34e3e77d1
commit 9edbff0a6b
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internal/querynode/load_service_test.go
View File

@ -1143,7 +1143,7 @@ func TestSegmentLoad_Search_Vector(t *testing.T) {
assert.NoError(t, err)
fieldsMap, _ := node.loadService.segLoader.checkTargetFields(paths, srcFieldIDs, fieldIDs)
assert.Equal(t, len(fieldsMap), 2)
assert.Equal(t, len(fieldsMap), 4)
segment, err := node.replica.getSegmentByID(segmentID)
assert.NoError(t, err)

7
internal/querynode/segment_loader.go
View File

@ -109,11 +109,14 @@ func (loader *segmentLoader) checkTargetFields(paths []*internalpb2.StringList,
return false
}
for i, fieldID := range dstFieldIDs {
for _, fieldID := range dstFieldIDs {
if !containsFunc(srcFieldIDs, fieldID) {
return nil, errors.New("uncompleted fields")
}
targetFields[fieldID] = paths[i]
}
for i := range srcFieldIDs {
targetFields[srcFieldIDs[i]] = paths[i]
}
return targetFields, nil