summaryrefslogtreecommitdiff
path: root/vendor/github.com/dsoprea/go-utility/v2/filesystem/progress_wrapper.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/progress_wrapper.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/progress_wrapper.go')
-rw-r--r--vendor/github.com/dsoprea/go-utility/v2/filesystem/progress_wrapper.go93
1 files changed, 93 insertions, 0 deletions
diff --git a/vendor/github.com/dsoprea/go-utility/v2/filesystem/progress_wrapper.go b/vendor/github.com/dsoprea/go-utility/v2/filesystem/progress_wrapper.go
new file mode 100644
index 000000000..0a064c53d
--- /dev/null
+++ b/vendor/github.com/dsoprea/go-utility/v2/filesystem/progress_wrapper.go
@@ -0,0 +1,93 @@
+package rifs
+
+import (
+ "io"
+ "time"
+
+ "github.com/dsoprea/go-logging"
+)
+
+// ProgressFunc receives progress updates.
+type ProgressFunc func(n int, duration time.Duration, isEof bool) error
+
+// WriteProgressWrapper wraps a reader and calls a callback after each read with
+// count and duration info.
+type WriteProgressWrapper struct {
+ w io.Writer
+ progressCb ProgressFunc
+}
+
+// NewWriteProgressWrapper returns a new WPW instance.
+func NewWriteProgressWrapper(w io.Writer, progressCb ProgressFunc) io.Writer {
+ return &WriteProgressWrapper{
+ w: w,
+ progressCb: progressCb,
+ }
+}
+
+// Write does a write and calls the callback.
+func (wpw *WriteProgressWrapper) Write(buffer []byte) (n int, err error) {
+ defer func() {
+ if state := recover(); state != nil {
+ err = log.Wrap(state.(error))
+ }
+ }()
+
+ startAt := time.Now()
+
+ n, err = wpw.w.Write(buffer)
+ log.PanicIf(err)
+
+ duration := time.Since(startAt)
+
+ err = wpw.progressCb(n, duration, false)
+ log.PanicIf(err)
+
+ return n, nil
+}
+
+// ReadProgressWrapper wraps a reader and calls a callback after each read with
+// count and duration info.
+type ReadProgressWrapper struct {
+ r io.Reader
+ progressCb ProgressFunc
+}
+
+// NewReadProgressWrapper returns a new RPW instance.
+func NewReadProgressWrapper(r io.Reader, progressCb ProgressFunc) io.Reader {
+ return &ReadProgressWrapper{
+ r: r,
+ progressCb: progressCb,
+ }
+}
+
+// Read reads data and calls the callback.
+func (rpw *ReadProgressWrapper) Read(buffer []byte) (n int, err error) {
+ defer func() {
+ if state := recover(); state != nil {
+ err = log.Wrap(state.(error))
+ }
+ }()
+
+ startAt := time.Now()
+
+ n, err = rpw.r.Read(buffer)
+
+ duration := time.Since(startAt)
+
+ if err != nil {
+ if err == io.EOF {
+ errInner := rpw.progressCb(n, duration, true)
+ log.PanicIf(errInner)
+
+ return n, err
+ }
+
+ log.Panic(err)
+ }
+
+ err = rpw.progressCb(n, duration, false)
+ log.PanicIf(err)
+
+ return n, nil
+}