37 lines
687 B
Go
37 lines
687 B
Go
|
// Copyright (C) 2018 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package streams
|
||
|
|
|
||
|
|
import "io"
|
||
|
|
|
||
|
|
// EOFReader holds reader and status of EOF
|
||
|
|
type EOFReader struct {
|
||
|
|
reader io.Reader
|
||
|
|
eof bool
|
||
|
|
err error
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewEOFReader keeps track of the state, has the internal reader reached EOF
|
||
|
|
func NewEOFReader(r io.Reader) *EOFReader {
|
||
|
|
return &EOFReader{reader: r}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *EOFReader) Read(p []byte) (n int, err error) {
|
||
|
|
n, err = r.reader.Read(p)
|
||
|
|
if err == io.EOF {
|
||
|
|
r.eof = true
|
||
|
|
} else if err != nil && r.err == nil {
|
||
|
|
r.err = err
|
||
|
|
}
|
||
|
|
return n, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *EOFReader) isEOF() bool {
|
||
|
|
return r.eof
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *EOFReader) hasError() bool {
|
||
|
|
return r.err != nil
|
||
|
|
}
|