Merge pull request #820 from yeya24/fix/update-dfgettask-success

bugfix: update dfgettask status to success
dragonflyoss · Aug 15, 2019 · bbeb1b1 · bbeb1b1
2 parents b2fd8ee + a1a4d20
commit bbeb1b1
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/supernode/daemon/mgr/task/manager_util.go b/supernode/daemon/mgr/task/manager_util.go
@@ -324,6 +324,11 @@ func (tm *Manager) parseAvailablePeers(ctx context.Context, clientID string, tas
 	pieceSuccess, _ := tm.progressMgr.GetPieceProgressByCID(ctx, task.ID, clientID, "success")
 	logrus.Debugf("taskID: %s, get successful pieces: %v", task.ID, pieceSuccess)
 	if cdnSuccess && (int32(len(pieceSuccess)) == task.PieceTotal) {
+		// update dfget task status to success
+		if err := tm.dfgetTaskMgr.UpdateStatus(ctx, clientID, task.ID, types.DfGetTaskStatusSUCCESS); err != nil {
+			logrus.Errorf("failed to update dfget task status with "+
+				"taskID(%s) clientID(%s) status(%s): %v", task.ID, clientID, types.DfGetTaskStatusSUCCESS, err)
+		}
 		finishInfo := make(map[string]interface{})
 		finishInfo["md5"] = task.Md5
 		finishInfo["fileLength"] = task.FileLength