update flush

zilliztech · Oct 9, 2022 · d8cbe1b · d8cbe1b
1 parent 1acf6c7
commit d8cbe1b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/core/backup_context.go b/core/backup_context.go
@@ -322,8 +322,9 @@ func (b BackupContext) CreateBackup(ctx context.Context, request *backuppb.Creat
 	collSegmentsMap := make(map[string][]int64)
 	collSealTimeMap := make(map[string]int64)
 	for _, coll := range toBackupCollections {
-		segmentIDs, timeOfSeal, err := b.milvusClient.Flush(ctx, coll.Name, false)
-		collSegmentsMap[coll.Name] = segmentIDs
+		newSealedSegmentIDs, flushedSegmentIDs, timeOfSeal, err := b.milvusClient.Flush(ctx, coll.Name, false)
+		log.Info("flush segments", zap.Int64s("newSealedSegmentIDs", newSealedSegmentIDs), zap.Int64s("flushedSegmentIDs", flushedSegmentIDs))
+		collSegmentsMap[coll.Name] = append(newSealedSegmentIDs, flushedSegmentIDs...)
 		collSealTimeMap[coll.Name] = timeOfSeal
 		if err != nil {
 			log.Error(fmt.Sprintf("fail to flush the collection: %s", coll.Name))
@@ -538,7 +539,7 @@ func (b BackupContext) GetBackup(ctx context.Context, request *backuppb.GetBacku
 
 	backup, err := b.readBackup(ctx, request.GetBackupName())
 	if err != nil {
-		log.Error("Fail to read backup", zap.String("backupName", request.GetBackupName()), zap.Error(err))
+		log.Warn("Fail to read backup", zap.String("backupName", request.GetBackupName()), zap.Error(err))
 		resp.Status.Reason = err.Error()
 		return resp, nil
 	}

diff --git a/go.mod b/go.mod
@@ -31,6 +31,6 @@ require (
 
 replace (
 	github.com/apache/pulsar-client-go => github.com/milvus-io/pulsar-client-go v0.6.8
-	github.com/milvus-io/milvus-sdk-go/v2 => github.com/wayblink/milvus-sdk-go/v2 v2.2.1
+	github.com/milvus-io/milvus-sdk-go/v2 => github.com/wayblink/milvus-sdk-go/v2 v2.2.2
 	github.com/streamnative/pulsarctl => github.com/xiaofan-luan/pulsarctl v0.5.1
 )