Codebase list golang-github-vbauerster-mpb / 43a3d29 proxyreader.go
43a3d29

Tree @43a3d29 (Download .tar.gz)

proxyreader.go @43a3d29raw · history · blame

package mpb

import (
	"io"
	"time"
)

// proxyReader is io.Reader wrapper, for proxy read bytes
type proxyReader struct {
	io.ReadCloser
	bar *Bar
	iT  time.Time
}

func (pr *proxyReader) Read(p []byte) (n int, err error) {
	n, err = pr.ReadCloser.Read(p)
	if n > 0 {
		pr.bar.IncrBy(n, time.Since(pr.iT))
		pr.iT = time.Now()
	}
	return
}