mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2024-12-19 08:51:06 +00:00
29007b1b88
* update bun libraries to v1.2.5 * pin old v1.29.0 of otel
94 lines
1.2 KiB
Go
94 lines
1.2 KiB
Go
package pgdialect
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
)
|
|
|
|
type arrayParser struct {
|
|
p pgparser
|
|
|
|
elem []byte
|
|
err error
|
|
}
|
|
|
|
func newArrayParser(b []byte) *arrayParser {
|
|
p := new(arrayParser)
|
|
|
|
if len(b) < 2 || b[0] != '{' || b[len(b)-1] != '}' {
|
|
p.err = fmt.Errorf("pgdialect: can't parse array: %q", b)
|
|
return p
|
|
}
|
|
|
|
p.p.Reset(b[1 : len(b)-1])
|
|
return p
|
|
}
|
|
|
|
func (p *arrayParser) Next() bool {
|
|
if p.err != nil {
|
|
return false
|
|
}
|
|
p.err = p.readNext()
|
|
return p.err == nil
|
|
}
|
|
|
|
func (p *arrayParser) Err() error {
|
|
if p.err != io.EOF {
|
|
return p.err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *arrayParser) Elem() []byte {
|
|
return p.elem
|
|
}
|
|
|
|
func (p *arrayParser) readNext() error {
|
|
ch := p.p.Read()
|
|
if ch == 0 {
|
|
return io.EOF
|
|
}
|
|
|
|
switch ch {
|
|
case '}':
|
|
return io.EOF
|
|
case '"':
|
|
b, err := p.p.ReadSubstring(ch)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if p.p.Peek() == ',' {
|
|
p.p.Advance()
|
|
}
|
|
|
|
p.elem = b
|
|
return nil
|
|
case '[', '(':
|
|
rng, err := p.p.ReadRange(ch)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if p.p.Peek() == ',' {
|
|
p.p.Advance()
|
|
}
|
|
|
|
p.elem = rng
|
|
return nil
|
|
default:
|
|
lit := p.p.ReadLiteral(ch)
|
|
if bytes.Equal(lit, []byte("NULL")) {
|
|
lit = nil
|
|
}
|
|
|
|
if p.p.Peek() == ',' {
|
|
p.p.Advance()
|
|
}
|
|
|
|
p.elem = lit
|
|
return nil
|
|
}
|
|
}
|