You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
853 B
38 lines
853 B
1 year ago
|
package utils
|
||
|
|
|
||
|
|
import (
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/rs/zerolog/log"
|
||
|
|
)
|
||
|
|
|
||
|
|
var lastProgress time.Time = time.Now()
|
||
|
|
var startTime time.Time = time.Now()
|
||
|
|
|
||
|
|
func ResetDownloadTimers() {
|
||
|
|
lastProgress = time.Now()
|
||
|
|
startTime = time.Now()
|
||
|
|
}
|
||
|
|
|
||
|
|
func DisplayDownloadFunction(fileName string, current string, total string, percentage float64) {
|
||
|
|
currentTime := time.Now()
|
||
|
|
|
||
|
|
if currentTime.Sub(lastProgress) >= 5*time.Second {
|
||
|
|
|
||
|
|
lastProgress = currentTime
|
||
|
|
|
||
|
|
// calculate ETA based on percentage and elapsed time
|
||
|
|
var eta time.Duration
|
||
|
|
if percentage > 0 {
|
||
|
|
elapsed := currentTime.Sub(startTime)
|
||
|
|
eta = time.Duration(float64(elapsed)*(100/percentage) - float64(elapsed))
|
||
|
|
}
|
||
|
|
|
||
|
|
if total != "" {
|
||
|
|
log.Debug().Msgf("Downloading %s: %s/%s (%.2f%%) ETA: %s", fileName, current, total, percentage, eta)
|
||
|
|
} else {
|
||
|
|
log.Debug().Msgf("Downloading: %s", current)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|