fix race condition

pkg/service/estimates.go

@@ -58,8 +58,7 @@ func RestoreJobStatus(job *jobInfo) *model.RestoreJobStatus {
 	}
 
 	if job.status == model.JobStatusRunning {
-		status.CurrentRestore = NewRunningJob(job.startTime, status.ReadRecords,
-			job.totalRecords)
+		status.CurrentRestore = NewRunningJob(job.startTime, status.ReadRecords, job.totalRecords)
 	}
 
 	if job.err != nil {

pkg/service/jobs_holder.go

@@ -74,6 +74,18 @@ func (h *RestoreJobsHolder) finishJob(id model.RestoreJobID, err error) {
 	})
 }
 
+func (h *RestoreJobsHolder) getStatus(id model.RestoreJobID) (*model.RestoreJobStatus, error) {
+	var result *model.RestoreJobStatus
+	h.Apply(id, func(value *jobInfo) {
+		result = RestoreJobStatus(value)
+	})
+
+	if result != nil {
+		return result, nil
+	}
+	return nil, NewErrJobNotFound(id)
+}
+
 func (h *RestoreJobsHolder) getJob(id model.RestoreJobID) (*jobInfo, error) {
 	if job, exists := h.Load(id); exists {
 		return job, nil

pkg/service/restore_data.go

@@ -247,12 +247,7 @@ func (r *dataRestorer) restoreFromPath(
 
 // JobStatus returns the status of the job with the given id.
 func (r *dataRestorer) JobStatus(jobID model.RestoreJobID) (*model.RestoreJobStatus, error) {
-	job, err := r.restoreJobs.getJob(jobID)
-	if err != nil {
-		return nil, err
-	}
-
-	return RestoreJobStatus(job), nil
+	return r.restoreJobs.getStatus(jobID)
 }
 
 func recordsInBackup(ctx context.Context, request *model.RestoreRequest) (uint64, error) {