diff --git a/pkg/task/tranfer.go b/pkg/task/tranfer.go index 596ae9e..5f9aa58 100644 --- a/pkg/task/tranfer.go +++ b/pkg/task/tranfer.go @@ -96,9 +96,8 @@ func (job *TransferTask) Do() { return } - for index, file := range job.TaskProps.Src { - job.TaskModel.SetProgress(index) - + successCount := 0 + for _, file := range job.TaskProps.Src { dst := path.Join(job.TaskProps.Dst, filepath.Base(file)) if job.TaskProps.TrimPath { // 保留原始目录 @@ -132,6 +131,9 @@ func (job *TransferTask) Do() { if err != nil { job.SetErrorMsg("文件转存失败", err) + } else { + successCount++ + job.TaskModel.SetProgress(successCount) } } diff --git a/pkg/task/transfer_test.go b/pkg/task/transfer_test.go index 94301f6..612a453 100644 --- a/pkg/task/transfer_test.go +++ b/pkg/task/transfer_test.go @@ -88,11 +88,6 @@ func TestTransferTask_Do(t *testing.T) { } task.TaskProps.Src = []string{"test/not_exist"} task.TaskProps.Parent = "test/not_exist" - // 更新进度 - mock.ExpectBegin() - mock.ExpectExec("UPDATE(.+)").WillReturnResult(sqlmock.NewResult(1, - 1)) - mock.ExpectCommit() // 更新错误 mock.ExpectBegin() mock.ExpectExec("UPDATE(.+)").WillReturnResult(sqlmock.NewResult(1, @@ -113,11 +108,6 @@ func TestTransferTask_Do(t *testing.T) { task.TaskProps.Src = []string{"test/not_exist"} task.TaskProps.Parent = "test/not_exist" task.TaskProps.TrimPath = true - // 更新进度 - mock.ExpectBegin() - mock.ExpectExec("UPDATE(.+)").WillReturnResult(sqlmock.NewResult(1, - 1)) - mock.ExpectCommit() // 更新错误 mock.ExpectBegin() mock.ExpectExec("UPDATE(.+)").WillReturnResult(sqlmock.NewResult(1,