summaryrefslogtreecommitdiff
path: root/vendor/github.com/dsoprea/go-utility/v2/filesystem/write_counter.go
diff options
context:
space:
mode:
authorLibravatar kim <89579420+NyaaaWhatsUpDoc@users.noreply.github.com>2024-08-02 11:46:41 +0000
committerLibravatar GitHub <noreply@github.com>2024-08-02 12:46:41 +0100
commit94e87610c4ce9bbb1c614a61bab29c1422fed11b (patch)
tree2e06b8ce64212140e796f6077ba841b6cc678501 /vendor/github.com/dsoprea/go-utility/v2/filesystem/write_counter.go
parent[feature] Allow import of following and blocks via CSV (#3150) (diff)
downloadgotosocial-94e87610c4ce9bbb1c614a61bab29c1422fed11b.tar.xz
[chore] add back exif-terminator and use only for jpeg,png,webp (#3161)
* add back exif-terminator and use only for jpeg,png,webp * fix arguments passed to terminateExif() * pull in latest exif-terminator * fix test * update processed img --------- Co-authored-by: tobi <tobi.smethurst@protonmail.com>
Diffstat (limited to 'vendor/github.com/dsoprea/go-utility/v2/filesystem/write_counter.go')
-rw-r--r--vendor/github.com/dsoprea/go-utility/v2/filesystem/write_counter.go36
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
+}