mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2024-12-25 03:32:12 +00:00
61 lines
793 B
Go
61 lines
793 B
Go
|
package pgdialect
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"io"
|
||
|
)
|
||
|
|
||
|
type streamParser struct {
|
||
|
b []byte
|
||
|
i int
|
||
|
|
||
|
buf []byte
|
||
|
}
|
||
|
|
||
|
func newStreamParser(b []byte, start int) *streamParser {
|
||
|
return &streamParser{
|
||
|
b: b,
|
||
|
i: start,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (p *streamParser) valid() bool {
|
||
|
return p.i < len(p.b)
|
||
|
}
|
||
|
|
||
|
func (p *streamParser) skipByte(skip byte) error {
|
||
|
c, err := p.readByte()
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
if c == skip {
|
||
|
return nil
|
||
|
}
|
||
|
p.unreadByte()
|
||
|
return fmt.Errorf("got %q, wanted %q", c, skip)
|
||
|
}
|
||
|
|
||
|
func (p *streamParser) readByte() (byte, error) {
|
||
|
if p.valid() {
|
||
|
c := p.b[p.i]
|
||
|
p.i++
|
||
|
return c, nil
|
||
|
}
|
||
|
return 0, io.EOF
|
||
|
}
|
||
|
|
||
|
func (p *streamParser) unreadByte() {
|
||
|
p.i--
|
||
|
}
|
||
|
|
||
|
func (p *streamParser) peek() byte {
|
||
|
if p.valid() {
|
||
|
return p.b[p.i]
|
||
|
}
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
func (p *streamParser) skipNext() {
|
||
|
p.i++
|
||
|
}
|