Merge pull request #361 from superseriousbusiness/media_refactorv0.2.0

Refactor media handler to allow async media resolution

1 files changed, 54 insertions, 0 deletions

diff --git a/vendor/github.com/dsoprea/go-utility/v2/filesystem/graceful_copy.go b/vendor/github.com/dsoprea/go-utility/v2/filesystem/graceful_copy.go
new file mode 100644
index 000000000..8705e5fe0
--- /dev/null
+++ b/vendor/github.com/dsoprea/go-utility/v2/filesystem/graceful_copy.go
@@ -0,0 +1,54 @@
+package rifs
+
+import (
+	"fmt"
+	"io"
+)
+
+const (
+	defaultCopyBufferSize = 1024 * 1024
+)
+
+// GracefulCopy willcopy while enduring lesser normal issues.
+//
+// - We'll ignore EOF if the read byte-count is more than zero. Only an EOF when
+//   zero bytes were read will terminate the loop.
+//
+// - Ignore short-writes. If less bytes were written than the bytes that were
+//   given, we'll keep trying until done.
+func GracefulCopy(w io.Writer, r io.Reader, buffer []byte) (copyCount int, err error) {
+	if buffer == nil {
+		buffer = make([]byte, defaultCopyBufferSize)
+	}
+
+	for {
+		readCount, err := r.Read(buffer)
+		if err != nil {
+			if err != io.EOF {
+				err = fmt.Errorf("read error: %s", err.Error())
+				return 0, err
+			}
+
+			// Only break on EOF if no bytes were actually read.
+			if readCount == 0 {
+				break
+			}
+		}
+
+		writeBuffer := buffer[:readCount]
+
+		for len(writeBuffer) > 0 {
+			writtenCount, err := w.Write(writeBuffer)
+			if err != nil {
+				err = fmt.Errorf("write error: %s", err.Error())
+				return 0, err
+			}
+
+			writeBuffer = writeBuffer[writtenCount:]
+		}
+
+		copyCount += readCount
+	}
+
+	return copyCount, nil
+}