fix: file peer task back source digest not match (#849)

Signed-off-by: Jim Ma <[email protected]>
dragonflyoss · Dec 1, 2021 · 7f28389 · 7f28389
1 parent 8d0ed54
commit 7f28389
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/client/daemon/peer/piece_manager.go b/client/daemon/peer/piece_manager.go
@@ -397,10 +397,6 @@ func (pm *pieceManager) DownloadSource(ctx context.Context, pt Task, request *sc
 			return storage.ErrShortRead
 		}
 	}
-	pt.SetTotalPieces(maxPieceNum)
-	if err := pt.SetContentLength(contentLength); err != nil {
-		log.Errorf("set content length failed %s", err)
-	}
 
 	if err := pm.storageManager.UpdateTask(ctx,
 		&storage.UpdateTaskRequest{
@@ -414,6 +410,10 @@ func (pm *pieceManager) DownloadSource(ctx context.Context, pt Task, request *sc
 		}); err != nil {
 		log.Errorf("update task failed %s", err)
 	}
+	if err := pt.SetContentLength(contentLength); err != nil {
+		log.Errorf("set content length failed %s", err)
+	}
+
 	log.Infof("download from source ok")
 	return nil
 }