Fix querynode panic (#23181)

Signed-off-by: bigsheeper <yihao.dai@zilliz.com>
2023-04-03 14:18:24 +08:00 · 2023-04-03 14:18:24 +08:00 · 9cecdc4f4e
parent afd874b736
commit 9cecdc4f4e
1 changed files with 1 additions and 1 deletions
--- a/internal/querynodev2/segments/retrieve.go
+++ b/internal/querynodev2/segments/retrieve.go
@ -30,7 +30,7 @@ func retrieveOnSegments(ctx context.Context, manager *Manager, segType SegmentTy
 	var retrieveResults []*segcorepb.RetrieveResults

 	for _, segID := range segIDs {
-		segment := manager.Segment.Get(segID).(*LocalSegment)
+		segment, _ := manager.Segment.Get(segID).(*LocalSegment)
 		if segment == nil {
 			continue
 		}