51 lines
676 B
Go
51 lines
676 B
Go
|
package buffer
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bufio"
|
||
|
|
"bytes"
|
||
|
|
"io"
|
||
|
|
)
|
||
|
|
|
||
|
|
type (
|
||
|
|
Buffer struct {
|
||
|
|
input *bufio.Reader
|
||
|
|
ready chan rune
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewBytesBuffer(b []byte) *Buffer {
|
||
|
|
return NewReaderBuffer(bytes.NewReader(b))
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewReaderBuffer(rd io.Reader) *Buffer {
|
||
|
|
return New(bufio.NewReader(rd))
|
||
|
|
}
|
||
|
|
|
||
|
|
func New(input *bufio.Reader) *Buffer {
|
||
|
|
b := &Buffer{
|
||
|
|
input: input,
|
||
|
|
ready: make(chan rune, 100),
|
||
|
|
}
|
||
|
|
go b.stream()
|
||
|
|
return b
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Buffer) Next() rune {
|
||
|
|
if next, ok := <-b.ready; ok {
|
||
|
|
return next
|
||
|
|
} else {
|
||
|
|
return 0
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Buffer) stream() {
|
||
|
|
for {
|
||
|
|
if r, _, err := b.input.ReadRune(); err != nil {
|
||
|
|
close(b.ready)
|
||
|
|
break
|
||
|
|
} else {
|
||
|
|
b.ready <- r
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|