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