1 files changed, 36 insertions, 0 deletions
diff --git a/vendor/github.com/dsoprea/go-utility/v2/filesystem/read_counter.go b/vendor/github.com/dsoprea/go-utility/v2/filesystem/read_counter.go
new file mode 100644
index 000000000..d878ca4e6
--- /dev/null
+++ b/vendor/github.com/dsoprea/go-utility/v2/filesystem/read_counter.go
@@ -0,0 +1,36 @@
+package rifs
+
+import (
+	"io"
+)
+
+// ReadCounter proxies read requests and maintains a counter of bytes read.
+type ReadCounter struct {
+	r       io.Reader
+	counter int
+}
+
+// NewReadCounter returns a new `ReadCounter` struct wrapping a `Reader`.
+func NewReadCounter(r io.Reader) *ReadCounter {
+	return &ReadCounter{
+		r: r,
+	}
+}
+
+// Count returns the total number of bytes read.
+func (rc *ReadCounter) Count() int {
+	return rc.counter
+}
+
+// Reset resets the counter to zero.
+func (rc *ReadCounter) Reset() {
+	rc.counter = 0
+}
+
+// Read forwards a read to the underlying `Reader` while bumping the counter.
+func (rc *ReadCounter) Read(b []byte) (n int, err error) {
+	n, err = rc.r.Read(b)
+	rc.counter += n
+
+	return n, err
+}