Skip to content

Commit a1d7522

Browse files
committed
enhance: Return segment id hint in QueryStream response
Related to milvus-io#36482 This PR reuses `SealedSegmentIDsRetrieved` field in `RetrieveResults` struct to store segment id hint. Signed-off-by: Congqi Xia <[email protected]>
1 parent da2d3ce commit a1d7522

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

internal/querynodev2/segments/retrieve.go

+2-1
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,8 @@ func retrieveOnSegmentsWithStream(ctx context.Context, mgr *Manager, segments []
124124
CostAggregation: &internalpb.CostAggregation{
125125
TotalRelatedDataSize: GetSegmentRelatedDataSize(segment),
126126
},
127-
AllRetrieveCount: result.GetAllRetrieveCount(),
127+
SealedSegmentIDsRetrieved: []int64{segment.ID()},
128+
AllRetrieveCount: result.GetAllRetrieveCount(),
128129
}); err != nil {
129130
errs[i] = err
130131
}

0 commit comments

Comments
 (0)