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

Set load meta info in load balance task (#16060) 

Signed-off-by: dragondriver <jiquan.long@zilliz.com>
pull/16030/merge
Jiquan Long 2022-03-15 20:37:21 +08:00 committed by GitHub
parent b61e8201e7
commit c984c4ea73
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
internal/querycoord/task.go
View File

@ -1712,6 +1712,11 @@ func (lbt *loadBalanceTask) execute(ctx context.Context) error {
Infos: []*querypb.SegmentLoadInfo{segmentLoadInfo}, Infos: []*querypb.SegmentLoadInfo{segmentLoadInfo},
Schema: schema, Schema: schema,
CollectionID: collectionID, CollectionID: collectionID,
LoadMeta: &querypb.LoadMetaInfo{
LoadType: collectionInfo.LoadType,
CollectionID: collectionID,
PartitionIDs: toRecoverPartitionIDs,
},
} }
loadSegmentReqs = append(loadSegmentReqs, loadSegmentReq) loadSegmentReqs = append(loadSegmentReqs, loadSegmentReq)
@ -1749,6 +1754,11 @@ func (lbt *loadBalanceTask) execute(ctx context.Context) error {
CollectionID: collectionID, CollectionID: collectionID,
Infos: []*datapb.VchannelInfo{vChannelInfo}, Infos: []*datapb.VchannelInfo{vChannelInfo},
Schema: schema, Schema: schema,
LoadMeta: &querypb.LoadMetaInfo{
LoadType: collectionInfo.LoadType,
CollectionID: collectionID,
PartitionIDs: toRecoverPartitionIDs,
},
} }
if collectionInfo.LoadType == querypb.LoadType_LoadPartition { if collectionInfo.LoadType == querypb.LoadType_LoadPartition {