mirror of
https://github.com/k3s-io/k3s.git
synced 2024-06-07 19:41:36 +00:00
26 lines
496 B
Go
26 lines
496 B
Go
package ioprogress
|
|
|
|
import (
|
|
"io"
|
|
)
|
|
|
|
// ProgressReader is a wrapper around ReadCloser which allows for progress tracking
|
|
type ProgressReader struct {
|
|
io.ReadCloser
|
|
Tracker *ProgressTracker
|
|
}
|
|
|
|
// Read in ProgressReader is the same as io.Read
|
|
func (pt *ProgressReader) Read(p []byte) (int, error) {
|
|
// Do normal reader tasks
|
|
n, err := pt.ReadCloser.Read(p)
|
|
|
|
// Do the actual progress tracking
|
|
if pt.Tracker != nil {
|
|
pt.Tracker.total += int64(n)
|
|
pt.Tracker.update(n)
|
|
}
|
|
|
|
return n, err
|
|
}
|