summaryrefslogtreecommitdiff
path: root/vendor/github.com/dsoprea/go-utility/v2/filesystem/progress_wrapper.go
diff options
context:
space:
mode:
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, 0 insertions, 93 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
deleted file mode 100644
index 0a064c53d..000000000
--- a/vendor/github.com/dsoprea/go-utility/v2/filesystem/progress_wrapper.go
+++ /dev/null
@@ -1,93 +0,0 @@
-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
-}