diff --git a/internal/querycoord/task.go b/internal/querycoord/task.go index a2bf775ab8..2a1aad44d0 100644 --- a/internal/querycoord/task.go +++ b/internal/querycoord/task.go @@ -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) } diff --git a/internal/querynode/task.go b/internal/querynode/task.go index e168992638..6d3d0c9679 100644 --- a/internal/querynode/task.go +++ b/internal/querynode/task.go @@ -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() {