56 lines
1.0 KiB
Go
56 lines
1.0 KiB
Go
|
package jstream
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"encoding/base64"
|
||
|
|
"io"
|
||
|
|
)
|
||
|
|
|
||
|
|
type delimReader struct {
|
||
|
|
r io.Reader
|
||
|
|
delim byte
|
||
|
|
rest []byte
|
||
|
|
}
|
||
|
|
|
||
|
|
func (dr *delimReader) Read(b []byte) (int, error) {
|
||
|
|
if dr.delim != 0 {
|
||
|
|
return 0, io.EOF
|
||
|
|
}
|
||
|
|
n, err := dr.r.Read(b)
|
||
|
|
if i := bytes.IndexAny(b[:n], bbDelims); i >= 0 {
|
||
|
|
dr.delim = b[i]
|
||
|
|
dr.rest = append([]byte(nil), b[i+1:n]...)
|
||
|
|
return i, err
|
||
|
|
}
|
||
|
|
return n, err
|
||
|
|
}
|
||
|
|
|
||
|
|
type byteBlobReader struct {
|
||
|
|
dr *delimReader
|
||
|
|
dec io.Reader
|
||
|
|
}
|
||
|
|
|
||
|
|
func newByteBlobReader(r io.Reader) *byteBlobReader {
|
||
|
|
dr := &delimReader{r: r}
|
||
|
|
return &byteBlobReader{
|
||
|
|
dr: dr,
|
||
|
|
dec: base64.NewDecoder(base64.StdEncoding, dr),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (bbr *byteBlobReader) Read(into []byte) (int, error) {
|
||
|
|
n, err := bbr.dec.Read(into)
|
||
|
|
if bbr.dr.delim == bbEnd {
|
||
|
|
return n, io.EOF
|
||
|
|
} else if bbr.dr.delim == bbCancel {
|
||
|
|
return n, ErrCanceled
|
||
|
|
}
|
||
|
|
return n, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// returns the bytes which were read off the underlying io.Reader but which
|
||
|
|
// haven't been consumed yet.
|
||
|
|
func (bbr *byteBlobReader) buffered() io.Reader {
|
||
|
|
return bytes.NewBuffer(bbr.dr.rest)
|
||
|
|
}
|