mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-29 15:20:04 +01:00
d5c180e680
It is better developing vmctl tool in VictoriaMetrics repository, so it could be released together with the rest of vmutils tools such as vmalert, vmagent, vmbackup, vmrestore and vmauth.
50 lines
1.0 KiB
Go
50 lines
1.0 KiB
Go
package pb
|
|
|
|
import (
|
|
"io"
|
|
)
|
|
|
|
// Reader it's a wrapper for given reader, but with progress handle
|
|
type Reader struct {
|
|
io.Reader
|
|
bar *ProgressBar
|
|
}
|
|
|
|
// Read reads bytes from wrapped reader and add amount of bytes to progress bar
|
|
func (r *Reader) Read(p []byte) (n int, err error) {
|
|
n, err = r.Reader.Read(p)
|
|
r.bar.Add(n)
|
|
return
|
|
}
|
|
|
|
// Close the wrapped reader when it implements io.Closer
|
|
func (r *Reader) Close() (err error) {
|
|
r.bar.Finish()
|
|
if closer, ok := r.Reader.(io.Closer); ok {
|
|
return closer.Close()
|
|
}
|
|
return
|
|
}
|
|
|
|
// Writer it's a wrapper for given writer, but with progress handle
|
|
type Writer struct {
|
|
io.Writer
|
|
bar *ProgressBar
|
|
}
|
|
|
|
// Write writes bytes to wrapped writer and add amount of bytes to progress bar
|
|
func (r *Writer) Write(p []byte) (n int, err error) {
|
|
n, err = r.Writer.Write(p)
|
|
r.bar.Add(n)
|
|
return
|
|
}
|
|
|
|
// Close the wrapped reader when it implements io.Closer
|
|
func (r *Writer) Close() (err error) {
|
|
r.bar.Finish()
|
|
if closer, ok := r.Writer.(io.Closer); ok {
|
|
return closer.Close()
|
|
}
|
|
return
|
|
}
|