pass flushTs in getFlushState

wayblink · Dec 14, 2023 · 1a1079c · 1a1079c
1 parent 5ed5a34
commit 1a1079c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/client/insert.go b/client/insert.go
@@ -217,13 +217,15 @@ func (c *GrpcClient) FlushV2(ctx context.Context, collName string, async bool) (
 		return nil, nil, 0, nil, err
 	}
 	channelCPs := resp.GetChannelCps()
+	flushTs := resp.GetCollFlushTs()[collName]
 	if !async {
 		segmentIDs, has := resp.GetCollSegIDs()[collName]
 		ids := segmentIDs.GetData()
 		if has && len(ids) > 0 {
 			flushed := func() bool {
 				resp, err := c.Service.GetFlushState(ctx, &milvuspb.GetFlushStateRequest{
 					SegmentIDs: ids,
+					FlushTs:    flushTs,
 				})
 
 				if err != nil {