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

View File

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