mirror of
https://github.com/aykhans/dodo.git
synced 2025-04-16 09:53:12 +00:00
Aykhan Shahsuvarov
00f0bcb2de
- Moved readers to the config package - Added an option to read remote config files - Moved the validation package to the config package and removed the validator dependency - Moved the customerrors package to the config package - Replaced fatih/color with jedib0t/go-pretty/v6/text - Removed proxy check functionality - Added param, header, cookie, body, and proxy flags to the CLI - Allowed multiple values for the same key in params, headers, and cookies
52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package requests
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/jedib0t/go-pretty/v6/progress"
|
|
)
|
|
|
|
// streamProgress streams the progress of a task to the console using a progress bar.
|
|
// It listens for increments on the provided channel and updates the progress bar accordingly.
|
|
//
|
|
// The function will stop and mark the progress as errored if the context is cancelled.
|
|
// It will also stop and mark the progress as done when the total number of increments is reached.
|
|
func streamProgress(
|
|
ctx context.Context,
|
|
wg *sync.WaitGroup,
|
|
total int64,
|
|
message string,
|
|
increase <-chan int64,
|
|
) {
|
|
defer wg.Done()
|
|
pw := progress.NewWriter()
|
|
pw.SetTrackerPosition(progress.PositionRight)
|
|
pw.SetStyle(progress.StyleBlocks)
|
|
pw.SetTrackerLength(40)
|
|
pw.SetUpdateFrequency(time.Millisecond * 250)
|
|
go pw.Render()
|
|
dodosTracker := progress.Tracker{
|
|
Message: message,
|
|
Total: total,
|
|
}
|
|
pw.AppendTracker(&dodosTracker)
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
if ctx.Err() != context.Canceled {
|
|
dodosTracker.MarkAsErrored()
|
|
}
|
|
fmt.Printf("\r")
|
|
time.Sleep(time.Millisecond * 500)
|
|
pw.Stop()
|
|
return
|
|
|
|
case value := <-increase:
|
|
dodosTracker.Increment(value)
|
|
}
|
|
}
|
|
}
|