You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
2.3 KiB
Go
100 lines
2.3 KiB
Go
2 months ago
|
package mget
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"github.com/vbauerster/mpb/v8"
|
||
|
|
"github.com/vbauerster/mpb/v8/decor"
|
||
|
|
"io"
|
||
|
|
"strings"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (m *Mget) processMiddleware(base mpb.BarFiller) mpb.BarFiller {
|
||
|
|
fn := func(w io.Writer, st decor.Statistics) error {
|
||
|
|
var res string
|
||
|
|
count := 0
|
||
|
|
_, err := fmt.Fprintf(w, "\nFinished:%s Total Write:%d Speed:%v\n\n", m.Redo.FormatPercent(), m.Redo.Total(), m.Redo.FormatSpeed("MB"))
|
||
|
|
for k := range m.threads {
|
||
|
|
v := m.threads[len(m.threads)-1-k]
|
||
|
|
if v != nil {
|
||
|
|
count++
|
||
|
|
res = fmt.Sprintf("Thread %v: %s %s\t", len(m.threads)-k, v.FormatSpeed("MB"), v.FormatPercent()) + res
|
||
|
|
if count%3 == 0 {
|
||
|
|
res = strings.TrimRight(res, "\t")
|
||
|
|
fmt.Fprintf(w, "%s\n", res)
|
||
|
|
res = ""
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if res != "" {
|
||
|
|
res = strings.TrimRight(res, "\t")
|
||
|
|
fmt.Fprintf(w, "%s\n", res)
|
||
|
|
}
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
if base == nil {
|
||
|
|
return mpb.BarFillerFunc(fn)
|
||
|
|
}
|
||
|
|
return mpb.BarFillerFunc(func(w io.Writer, st decor.Statistics) error {
|
||
|
|
err := fn(w, st)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return base.Fill(w, st)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (w *Mget) Process() {
|
||
|
|
w.processEnable = true
|
||
|
|
defer func() {
|
||
|
|
w.processEnable = false
|
||
|
|
}()
|
||
|
|
fmt.Println()
|
||
|
|
p := mpb.New()
|
||
|
|
var filler mpb.BarFiller
|
||
|
|
filler = w.processMiddleware(filler)
|
||
|
|
bar := p.New(int64(w.ContentLength),
|
||
|
|
mpb.BarStyle().Rbound("|"),
|
||
|
|
mpb.BarExtender(filler, true), // all bars share same extender filler
|
||
|
|
mpb.PrependDecorators(
|
||
|
|
decor.Counters(decor.SizeB1024(0), "% .2f / % .2f"),
|
||
|
|
),
|
||
|
|
mpb.AppendDecorators(
|
||
|
|
decor.EwmaETA(decor.ET_STYLE_GO, 30),
|
||
|
|
decor.Name(" ] "),
|
||
|
|
decor.EwmaSpeed(decor.SizeB1024(0), "% .2f ", 30),
|
||
|
|
),
|
||
|
|
)
|
||
|
|
defer p.Wait()
|
||
|
|
for {
|
||
|
|
last := w.Redo.Total()
|
||
|
|
lastTime := time.Now()
|
||
|
|
bar.SetCurrent(int64(w.Redo.Total()))
|
||
|
|
select {
|
||
|
|
case <-w.ctx.Done():
|
||
|
|
bar.SetCurrent(int64(w.Redo.Total()))
|
||
|
|
if w.dynLength {
|
||
|
|
bar.SetTotal(int64(w.Redo.ContentLength), true)
|
||
|
|
}
|
||
|
|
bar.Abort(false)
|
||
|
|
return
|
||
|
|
case <-time.After(time.Second):
|
||
|
|
if !w.writeEnable {
|
||
|
|
bar.SetCurrent(int64(w.Redo.Total()))
|
||
|
|
if w.dynLength {
|
||
|
|
bar.SetTotal(int64(w.Redo.ContentLength), true)
|
||
|
|
}
|
||
|
|
bar.Abort(true)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
now := w.Redo.Total()
|
||
|
|
bar.EwmaIncrInt64(int64(now-last), time.Since(lastTime))
|
||
|
|
lastTime = time.Now()
|
||
|
|
last = now
|
||
|
|
if w.dynLength {
|
||
|
|
bar.SetTotal(int64(w.Redo.ContentLength), false)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|