diff --git a/hosts/comment-reader.go b/hosts/comment-reader.go index 14673f4..648c0c2 100644 --- a/hosts/comment-reader.go +++ b/hosts/comment-reader.go @@ -9,17 +9,16 @@ type CommentReader struct { r io.Reader over []byte mark byte - hit bool } var _ io.Reader = &CommentReader{} func NewCommentReader(r io.Reader, mark byte) *CommentReader { - return &CommentReader{r, nil, mark, false} + return &CommentReader{r, nil, mark} } func (c *CommentReader) Read(p []byte) (n int, err error) { - if c.hit { + if c.over != nil { return 0, io.EOF } n, err = c.r.Read(p)