35 lines
564 B
Go
35 lines
564 B
Go
|
package utils
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"io"
|
||
|
|
)
|
||
|
|
|
||
|
|
// NewWriteCloser creates a new io.WriteCloser from an io.Writer.
|
||
|
|
func NewWriteCloser(wr io.Writer) io.WriteCloser {
|
||
|
|
return &WriteCloser{wr: wr}
|
||
|
|
}
|
||
|
|
|
||
|
|
// WriteCloser is a io.Writer with an io.Closer.
|
||
|
|
type WriteCloser struct {
|
||
|
|
wr io.Writer
|
||
|
|
|
||
|
|
closed bool
|
||
|
|
}
|
||
|
|
|
||
|
|
// Write to the io.Writer.
|
||
|
|
func (w *WriteCloser) Write(p []byte) (n int, err error) {
|
||
|
|
if w.closed {
|
||
|
|
return -1, errors.New("already closed")
|
||
|
|
}
|
||
|
|
|
||
|
|
return w.wr.Write(p)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Close the io.Closer.
|
||
|
|
func (w *WriteCloser) Close() error {
|
||
|
|
w.closed = true
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|