Skip to content

Commit 7fc5a41

Browse files
authored
Expose the underlying ReadCloser during restore (#501)
1 parent 1b44a7b commit 7fc5a41

File tree

1 file changed

+18
-2
lines changed

1 file changed

+18
-2
lines changed

Diff for: progress.go

+18-2
Original file line numberDiff line numberDiff line change
@@ -119,12 +119,28 @@ func newCountingReader(r io.Reader) *countingReader {
119119

120120
type countingReadCloser struct {
121121
*countingReader
122-
io.Closer
122+
readCloser io.ReadCloser
123123
}
124124

125125
func newCountingReadCloser(rc io.ReadCloser) *countingReadCloser {
126126
return &countingReadCloser{
127127
countingReader: newCountingReader(rc),
128-
Closer: rc,
128+
readCloser: rc,
129129
}
130130
}
131+
132+
func (c countingReadCloser) Close() error {
133+
return c.readCloser.Close()
134+
}
135+
136+
func (c countingReadCloser) WrappedReadCloser() io.ReadCloser {
137+
return c.readCloser
138+
}
139+
140+
// ReadCloserWrapper allows access to an underlying ReadCloser from a wrapper.
141+
type ReadCloserWrapper interface {
142+
io.ReadCloser
143+
WrappedReadCloser() io.ReadCloser
144+
}
145+
146+
var _ ReadCloserWrapper = &countingReadCloser{}

0 commit comments

Comments
 (0)