Browse code

pkg/ioutils: add WriteCloserWrapper

Docker-DCO-1.1-Signed-off-by: Cristian Staretu <cristian.staretu@gmail.com> (github: unclejack)

unclejack authored on 2014/08/13 01:20:59
Showing 1 changed files
... ...
@@ -21,3 +21,19 @@ func NopWriteCloser(w io.Writer) io.WriteCloser {
21 21
 type NopFlusher struct{}
22 22
 
23 23
 func (f *NopFlusher) Flush() {}
24
+
25
+type writeCloserWrapper struct {
26
+	io.Writer
27
+	closer func() error
28
+}
29
+
30
+func (r *writeCloserWrapper) Close() error {
31
+	return r.closer()
32
+}
33
+
34
+func NewWriteCloserWrapper(r io.Writer, closer func() error) io.WriteCloser {
35
+	return &writeCloserWrapper{
36
+		Writer: r,
37
+		closer: closer,
38
+	}
39
+}