feat: dfdaemon update content length (#895)

* feat: dfdaemon update content length Signed-off-by: Gaius <[email protected]> * feat: error check Signed-off-by: Gaius <[email protected]>
dragonflyoss · Dec 8, 2021 · 332c976 · 332c976
1 parent ee23dec
commit 332c976
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/client/daemon/peer/peertask_base.go b/client/daemon/peer/peertask_base.go
@@ -472,6 +472,13 @@ loop:
 			pt.Debugf("update digest: %s", pt.md5)
 		}
 
+		// update content length
+		if piecePacket.ContentLength > 0 {
+			_ = pt.SetContentLength(piecePacket.ContentLength)
+			_ = pt.callback.Update(pt)
+			pt.Debugf("update content length: %d", pt.GetContentLength())
+		}
+
 		// 3. dispatch piece request to all workers
 		pt.dispatchPieceRequest(pieceRequestCh, piecePacket)