Skip to content

Commit 4fd9b0a

Browse files
authored
enhance: Return segment id hint in QueryStream response (#36487)
Related to #36482 This PR reuses `SealedSegmentIDsRetrieved` field in `RetrieveResults` struct to store segment id hint. Signed-off-by: Congqi Xia <[email protected]>
1 parent 9e8cafc commit 4fd9b0a

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)