Skip to content

Instantly share code, notes, and snippets.

@arosh
Created December 30, 2016 16:58
Show Gist options
  • Star 3 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save arosh/98af4d3c490ad5e7c79df3a5a2af8672 to your computer and use it in GitHub Desktop.
Save arosh/98af4d3c490ad5e7c79df3a5a2af8672 to your computer and use it in GitHub Desktop.
read until EOF
package main
import (
"bufio"
"fmt"
"os"
"io"
"log"
)
type LineIterator struct {
reader *bufio.Reader
}
func NewLineIterator(rd io.Reader) *LineIterator {
return &LineIterator{
reader: bufio.NewReader(rd),
}
}
func (ln *LineIterator) Next() ([]byte, error) {
var bytes []byte
for {
line, isPrefix, err := ln.reader.ReadLine()
if err != nil {
return nil, err
}
bytes = append(bytes, line...)
if !isPrefix {
break
}
}
return bytes, nil
}
// 肝心のmain関数があまり短くならないので意味があるか微妙…
func main() {
ln := NewLineIterator(os.Stdin)
for {
line, err := ln.Next()
if err != nil {
if err == io.EOF {
break
} else {
log.Fatal(err)
}
}
log.Printf("line = %s", line)
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment