72 lines
1.1 KiB
Go
72 lines
1.1 KiB
Go
package ioext
|
|
|
|
import "io"
|
|
|
|
type CounterReader struct {
|
|
Reader io.Reader
|
|
|
|
BytesRead int
|
|
}
|
|
|
|
func (r *CounterReader) Read(p []byte) (n int, err error) {
|
|
n, err = r.Reader.Read(p)
|
|
|
|
r.BytesRead += n
|
|
|
|
return n, err
|
|
}
|
|
|
|
type CounterReadCloser struct {
|
|
Reader io.ReadCloser
|
|
|
|
BytesRead int
|
|
}
|
|
|
|
func (r *CounterReadCloser) Read(p []byte) (n int, err error) {
|
|
n, err = r.Reader.Read(p)
|
|
|
|
r.BytesRead += n
|
|
|
|
return n, err
|
|
}
|
|
|
|
func (r *CounterReadCloser) Close() error {
|
|
return r.Reader.Close()
|
|
}
|
|
|
|
type CounterReadSeekCloser struct {
|
|
Reader io.ReadSeekCloser
|
|
|
|
BytesRead int
|
|
}
|
|
|
|
func (r *CounterReadSeekCloser) Read(p []byte) (n int, err error) {
|
|
n, err = r.Reader.Read(p)
|
|
|
|
r.BytesRead += n
|
|
|
|
return n, err
|
|
}
|
|
|
|
func (r *CounterReadSeekCloser) Close() error {
|
|
return r.Reader.Close()
|
|
}
|
|
|
|
func (r *CounterReadSeekCloser) Seek(offset int64, whence int) (int64, error) {
|
|
return r.Reader.Seek(offset, whence)
|
|
}
|
|
|
|
type CounterWriter struct {
|
|
Writer io.Writer
|
|
|
|
BytesRead int
|
|
}
|
|
|
|
func (w *CounterWriter) Write(p []byte) (n int, err error) {
|
|
n, err = w.Writer.Write(p)
|
|
|
|
w.BytesRead += n
|
|
|
|
return n, err
|
|
}
|