johno/decypharr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add Speed to callbacks

Browse Source
This commit is contained in:
Mukhtar Akere
2025-02-09 19:17:20 +01:00
parent 186a24cc4a
commit 1614e29f8f
16 changed files with 76 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pkg/downloaders/grab.go
View File

@@ -20,7 +20,7 @@ func GetGrabClient() *grab.Client {
}
}
func NormalGrab(client *grab.Client, url, filename string, progressCallback func(int64)) error {
func NormalGrab(client *grab.Client, url, filename string, progressCallback func(int64, int64)) error {
req, err := grab.NewRequest(filename, url)
if err != nil {
return err
@@ -36,9 +36,10 @@ Loop:
select {
case <-t.C:
current := resp.BytesComplete()
speed := int64(resp.BytesPerSecond())
if current != lastReported {
if progressCallback != nil {
progressCallback(current - lastReported)
progressCallback(current-lastReported, speed)
}
lastReported = current
}
@@ -49,7 +50,7 @@ Loop:
// Report final bytes
if progressCallback != nil {
progressCallback(resp.BytesComplete() - lastReported)
progressCallback(resp.BytesComplete()-lastReported, 0)
}
return resp.Err()