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 info in reschedule tasks (#16065) 

Signed-off-by: dragondriver <jiquan.long@zilliz.com>
pull/16068/head
Jiquan Long 2022-03-16 11:43:21 +08:00 committed by GitHub
parent cf4c014b2e
commit ffdad82b9c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
internal/querycoord/task.go
View File

@ -1090,6 +1090,11 @@ func (lst *loadSegmentTask) reschedule(ctx context.Context) ([]task, error) {
Schema: lst.Schema,
SourceNodeID: lst.SourceNodeID,
CollectionID: lst.CollectionID,
LoadMeta: &querypb.LoadMetaInfo{
LoadType: lst.GetLoadMeta().GetLoadType(),
CollectionID: lst.GetCollectionID(),
PartitionIDs: lst.GetLoadMeta().GetPartitionIDs(),
},
}
loadSegmentReqs = append(loadSegmentReqs, req)
}
@ -1264,6 +1269,11 @@ func (wdt *watchDmChannelTask) reschedule(ctx context.Context) ([]task, error) {
Infos: []*datapb.VchannelInfo{info},
Schema: wdt.Schema,
ExcludeInfos: wdt.ExcludeInfos,
LoadMeta: &querypb.LoadMetaInfo{
LoadType: wdt.GetLoadMeta().GetLoadType(),
CollectionID: collectionID,
PartitionIDs: wdt.GetLoadMeta().GetPartitionIDs(),
},
}
watchDmChannelReqs = append(watchDmChannelReqs, req)
}

2
internal/querynode/task.go
View File

@ -644,7 +644,7 @@ func (l *loadSegmentsTask) Execute(ctx context.Context) error {
var err error
// init meta
collectionID := l.req.GetLoadMeta().GetCollectionID()
collectionID := l.req.GetCollectionID()
l.node.historical.replica.addCollection(collectionID, l.req.GetSchema())
l.node.streaming.replica.addCollection(collectionID, l.req.GetSchema())
for _, partitionID := range l.req.GetLoadMeta().GetPartitionIDs() {