Files
podman/vendor/github.com/vbauerster/mpb/v4/proxyreader.go
T
2019-12-20 09:30:47 -05:00

46 lines
694 B
Go

package mpb
import (
"io"
"time"
)
type proxyReader struct {
io.ReadCloser
bar *Bar
iT time.Time
}
func (prox *proxyReader) Read(p []byte) (n int, err error) {
n, err = prox.ReadCloser.Read(p)
if n > 0 {
prox.bar.IncrBy(n, time.Since(prox.iT))
prox.iT = time.Now()
}
if err == io.EOF {
go func() {
prox.bar.SetTotal(0, true)
}()
}
return
}
type proxyWriterTo struct {
*proxyReader
wt io.WriterTo
}
func (prox *proxyWriterTo) WriteTo(w io.Writer) (n int64, err error) {
n, err = prox.wt.WriteTo(w)
if n > 0 {
prox.bar.IncrInt64(n, time.Since(prox.iT))
prox.iT = time.Now()
}
if err == io.EOF {
go func() {
prox.bar.SetTotal(0, true)
}()
}
return
}