Skip to content

Instantly share code, notes, and snippets.

@film42

film42/redirect.go

Created Aug 24, 2017
Embed
What would you like to do?
If you need to go from io.Reader to io.Reader but need to do a streaming io.Writer -> io.Reader before hand, this redirects the output
type NoOpWriteCloser struct {
writer io.Writer
}
func (c *NoOpWriteCloser) Write(p []byte) (int, error) {
return c.writer.Write(p)
}
func (c *NoOpWriteCloser) Close() error {
return nil
}
type Redirect struct {
src io.Reader
srcEOF bool
redirect io.WriteCloser
dest io.Reader
}
func NewRedirect(src io.Reader, redirect io.WriteCloser, dest io.Reader) *Redirect {
var writeCloser io.WriteCloser
if wcloser, ok := redirect.(io.WriteCloser); ok {
writeCloser = wcloser
} else {
writeCloser = &NoOpWriteCloser{redirect}
}
return &Redirect{
src: io.TeeReader(src, redirect),
srcEOF: false,
dest: dest,
redirect: writeCloser,
}
}
func (r *Redirect) Read(p []byte) (int, error) {
if !r.srcEOF {
_, err := r.src.Read(p)
if err == io.EOF {
r.srcEOF = true
r.redirect.Close()
}
}
return r.dest.Read(p)
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.