werr onceError
}
-func (p *pipe) Read(b []byte) (n int, err error) {
+func (p *pipe) read(b []byte) (n int, err error) {
select {
case <-p.done:
return 0, p.readCloseError()
}
}
-func (p *pipe) readCloseError() error {
- rerr := p.rerr.Load()
- if werr := p.werr.Load(); rerr == nil && werr != nil {
- return werr
- }
- return ErrClosedPipe
-}
-
-func (p *pipe) CloseRead(err error) error {
+func (p *pipe) closeRead(err error) error {
if err == nil {
err = ErrClosedPipe
}
return nil
}
-func (p *pipe) Write(b []byte) (n int, err error) {
+func (p *pipe) write(b []byte) (n int, err error) {
select {
case <-p.done:
return 0, p.writeCloseError()
return n, nil
}
-func (p *pipe) writeCloseError() error {
- werr := p.werr.Load()
- if rerr := p.rerr.Load(); werr == nil && rerr != nil {
- return rerr
- }
- return ErrClosedPipe
-}
-
-func (p *pipe) CloseWrite(err error) error {
+func (p *pipe) closeWrite(err error) error {
if err == nil {
err = EOF
}
return nil
}
+// readCloseError is considered internal to the pipe type.
+func (p *pipe) readCloseError() error {
+ rerr := p.rerr.Load()
+ if werr := p.werr.Load(); rerr == nil && werr != nil {
+ return werr
+ }
+ return ErrClosedPipe
+}
+
+// writeCloseError is considered internal to the pipe type.
+func (p *pipe) writeCloseError() error {
+ werr := p.werr.Load()
+ if rerr := p.rerr.Load(); werr == nil && rerr != nil {
+ return rerr
+ }
+ return ErrClosedPipe
+}
+
// A PipeReader is the read half of a pipe.
type PipeReader struct {
p *pipe
// If the write end is closed with an error, that error is
// returned as err; otherwise err is EOF.
func (r *PipeReader) Read(data []byte) (n int, err error) {
- return r.p.Read(data)
+ return r.p.read(data)
}
// Close closes the reader; subsequent writes to the
// CloseWithError never overwrites the previous error if it exists
// and always returns nil.
func (r *PipeReader) CloseWithError(err error) error {
- return r.p.CloseRead(err)
+ return r.p.closeRead(err)
}
// A PipeWriter is the write half of a pipe.
// If the read end is closed with an error, that err is
// returned as err; otherwise err is ErrClosedPipe.
func (w *PipeWriter) Write(data []byte) (n int, err error) {
- return w.p.Write(data)
+ return w.p.write(data)
}
// Close closes the writer; subsequent reads from the
// CloseWithError never overwrites the previous error if it exists
// and always returns nil.
func (w *PipeWriter) CloseWithError(err error) error {
- return w.p.CloseWrite(err)
+ return w.p.closeWrite(err)
}
// Pipe creates a synchronous in-memory pipe.