fix getBackup panic if pass a invalid backup ID (#424)

Signed-off-by: wayblink <anyang.wang@zilliz.com>
zilliztech · Sep 12, 2024 · ab69696 · ab69696
1 parent 7aad2e6
commit ab69696
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/core/backup_context.go b/core/backup_context.go
@@ -647,13 +647,13 @@ func (b *BackupContext) GetRestore(ctx context.Context, request *backuppb.GetRes
 	}
 
 	task := b.meta.GetRestoreTask(request.GetId())
-	progress := int32(float32(task.GetRestoredSize()) * 100 / float32(task.GetToRestoreSize()))
-	// don't return zero
-	if progress == 0 {
-		progress = 1
-	}
-	task.Progress = progress
 	if task != nil {
+		progress := int32(float32(task.GetRestoredSize()) * 100 / float32(task.GetToRestoreSize()))
+		// don't return zero
+		if progress == 0 {
+			progress = 1
+		}
+		task.Progress = progress
 		resp.Code = backuppb.ResponseCode_Success
 		resp.Msg = "success"
 		resp.Data = task