diff --git a/internal/dao/taskDao.go b/internal/dao/taskDao.go index 5cee10d..4c00f8a 100644 --- a/internal/dao/taskDao.go +++ b/internal/dao/taskDao.go @@ -506,6 +506,7 @@ func UpdateTaskProgress(req *CompleteTaskRequest) error { ) } updateData["complete_video_count"] = newCompleteCount + updateData["pending_video_count"] = assignRecord.PendingVideoCount - req.CompleteCount case "post": newCompleteCount := assignRecord.CompletePostCount + req.CompleteCount if newCompleteCount > assignRecord.AssignPostCount { @@ -521,6 +522,7 @@ func UpdateTaskProgress(req *CompleteTaskRequest) error { ) } updateData["complete_post_count"] = newCompleteCount + updateData["pending_post_count"] = assignRecord.PendingPostCount - req.CompleteCount case "data": newCompleteCount := assignRecord.CompleteDataCount + req.CompleteCount if newCompleteCount > assignRecord.AssignDataCount { @@ -536,6 +538,7 @@ func UpdateTaskProgress(req *CompleteTaskRequest) error { ) } updateData["complete_data_count"] = newCompleteCount + updateData["pending_data_count"] = assignRecord.PendingDataCount - req.CompleteCount default: tx.Rollback() return commonErr.ReturnError(nil, "无效的任务类型", "任务类型必须是video、post或data") @@ -976,9 +979,9 @@ func UpdatePendingCount(req *UpdatePendingCountRequest) error { PendingVideoCount: req.PendingVideoCount, PendingPostCount: req.PendingPostCount, PendingDataCount: req.PendingDataCount, - AssignVideoCount: 0, - AssignPostCount: 0, - AssignDataCount: 0, + AssignVideoCount: req.PendingVideoCount, + AssignPostCount: req.PendingPostCount, + AssignDataCount: req.PendingDataCount, CompleteVideoCount: 0, CompletePostCount: 0, CompleteDataCount: 0,