nxshock/gwp
1
0
Fork 0
mirror of https://github.com/nxshock/gwp.git synced 2024-11-27 03:31:02 +05:00
Code Issues Projects Releases Wiki Activity

Do not update progress message without changes

Browse Source
This commit is contained in:
nxshock 2022-06-25 17:53:32 +05:00
parent 54dac42925
commit a2986a9485
1 changed files with 20 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
gwp.go
View File

@ -1,6 +1,7 @@
package gwp
import (
"bytes"
"fmt"
"os"
"runtime"
@ -22,6 +23,8 @@ type WorkerPool struct {
processedCount int // processed jobs count
errorCount int // processed jobs count that returned error
currentSpeed float64 // speed calculated for last minute
lastProgressMessage string
}
// New creates new pool of workers with specified goroutine count.
@ -93,31 +96,41 @@ func (workerPool *WorkerPool) printProgress() {
return
}
fmt.Fprintf(os.Stderr, newLine)
buf := new(bytes.Buffer)
fmt.Fprintf(buf, newLine)
if workerPool.EstimateCount == 0 {
fmt.Fprintf(os.Stderr, "Progress: %d", workerPool.processedCount)
fmt.Fprintf(buf, "Progress: %d", workerPool.processedCount)
} else {
fmt.Fprintf(os.Stderr, "Progress: %.1f%% (%d / %d)",
fmt.Fprintf(buf, "Progress: %.1f%% (%d / %d)",
float64(workerPool.processedCount*100)/float64(workerPool.EstimateCount), workerPool.processedCount, workerPool.EstimateCount)
}
if workerPool.errorCount > 0 {
fmt.Fprintf(os.Stderr, " Errors: %d (%.1f%%)",
fmt.Fprintf(buf, " Errors: %d (%.1f%%)",
workerPool.errorCount, float64(workerPool.errorCount*100)/float64(workerPool.EstimateCount))
}
if workerPool.currentSpeed > 0 {
if workerPool.EstimateCount > 0 {
fmt.Fprintf(os.Stderr, " ETA: %s", fmtDuration(time.Second*time.Duration(float64(workerPool.EstimateCount-workerPool.processedCount)/workerPool.currentSpeed)))
fmt.Fprintf(buf, " ETA: %s", fmtDuration(time.Second*time.Duration(float64(workerPool.EstimateCount-workerPool.processedCount)/workerPool.currentSpeed)))
}
if workerPool.ShowSpeed {
fmt.Fprintf(os.Stderr, " Speed: %.2f rps", workerPool.currentSpeed)
fmt.Fprintf(buf, " Speed: %.2f rps", workerPool.currentSpeed)
}
}
fmt.Fprint(os.Stderr, endLine)
fmt.Fprint(buf, endLine)
if buf.String() == workerPool.lastProgressMessage {
return
}
buf.WriteTo(os.Stderr)
workerPool.lastProgressMessage = buf.String()
}
// Add sends specified task for execution.