pkg/ioutils: add ReaderErrWrapper to readers

unclejack · unclejack · commit bd130e72a06d · 2014-09-03T11:36:21.000+03:00
Docker-DCO-1.1-Signed-off-by: Cristian Staretu &lt;cristian.staretu@gmail.com&gt; (github: unclejack)
diff --git a/pkg/ioutils/readers.go b/pkg/ioutils/readers.go
@@ -22,6 +22,26 @@ func NewReadCloserWrapper(r io.Reader, closer func() error) io.ReadCloser {
 	}
 }
 
+type readerErrWrapper struct {
+	reader io.Reader
+	closer func()
+}
+
+func (r *readerErrWrapper) Read(p []byte) (int, error) {
+	n, err := r.reader.Read(p)
+	if err != nil {
+		r.closer()
+	}
+	return n, err
+}
+
+func NewReaderErrWrapper(r io.Reader, closer func()) io.Reader {
+	return &readerErrWrapper{
+		reader: r,
+		closer: closer,
+	}
+}
+
 type bufReader struct {
 	sync.Mutex
 	buf      *bytes.Buffer
