| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253 |
- // Copyright (c) 2012-2018 Ugorji Nwoke. All rights reserved.
- // Use of this source code is governed by a MIT license found in the LICENSE file.
- package codec
- import "io"
- /*
- // decReader abstracts the reading source, allowing implementations that can
- // read from an io.Reader or directly off a byte slice with zero-copying.
- //
- // Deprecated: Use decReaderSwitch instead.
- type decReader interface {
- unreadn1()
- // readx will use the implementation scratch buffer if possible i.e. n < len(scratchbuf), OR
- // just return a view of the []byte being decoded from.
- // Ensure you call detachZeroCopyBytes later if this needs to be sent outside codec control.
- readx(n int) []byte
- readb([]byte)
- readn1() uint8
- numread() uint // number of bytes read
- track()
- stopTrack() []byte
- // skip will skip any byte that matches, and return the first non-matching byte
- skip(accept *bitset256) (token byte)
- // readTo will read any byte that matches, stopping once no-longer matching.
- readTo(accept *bitset256) (out []byte)
- // readUntil will read, only stopping once it matches the 'stop' byte.
- readUntil(stop byte) (out []byte)
- }
- */
- // ------------------------------------------------
- type unreadByteStatus uint8
- // unreadByteStatus goes from
- // undefined (when initialized) -- (read) --> canUnread -- (unread) --> canRead ...
- const (
- unreadByteUndefined unreadByteStatus = iota
- unreadByteCanRead
- unreadByteCanUnread
- )
- // func appendPool(bs []byte, b byte, bufp *bytesBufPooler) []byte {
- // if cap(bs)-len(bs) < 1 {
- // bs = bufp.ensureCap(len(bs)+1, bs)
- // }
- // bs = append(bs, b)
- // return bs
- // }
- // func appendPoolMany(bs []byte, b []byte, bufp *bytesBufPooler) []byte {
- // if cap(bs)-len(bs) < 1 {
- // bs = bufp.ensureCap(len(bs)+1, bs)
- // }
- // bs = append(bs, b...)
- // return bs
- // }
- // --------------------
- type ioDecReaderCommon struct {
- r io.Reader // the reader passed in
- n uint // num read
- l byte // last byte
- ls unreadByteStatus // last byte status
- trb bool // tracking bytes turned on
- _ bool
- b [4]byte // tiny buffer for reading single bytes
- blist *bytesFreelist
- tr []byte // buffer for tracking bytes
- bufr []byte // buffer for readTo/readUntil
- }
- func (z *ioDecReaderCommon) reset(r io.Reader, blist *bytesFreelist) {
- z.blist = blist
- z.r = r
- z.ls = unreadByteUndefined
- z.l, z.n = 0, 0
- z.trb = false
- }
- func (z *ioDecReaderCommon) numread() uint {
- return z.n
- }
- func (z *ioDecReaderCommon) track() {
- z.tr = z.blist.check(z.tr, 256)[:0]
- z.trb = true
- }
- func (z *ioDecReaderCommon) stopTrack() (bs []byte) {
- z.trb = false
- return z.tr
- }
- // func (z *ioDecReaderCommon) resetBufr() {
- // if cap(z.bufr) < 128 {
- // blist.put(z.bufr)
- // z.bufr = blist.get(128)
- // }
- // z.bufr = z.bufr[:0]
- // }
- // func (z *ioDecReaderCommon) release() {
- // z.tr.end()
- // z.bufr.end()
- // }
- // ------------------------------------------
- // ioDecReader is a decReader that reads off an io.Reader.
- //
- // It also has a fallback implementation of ByteScanner if needed.
- type ioDecReader struct {
- ioDecReaderCommon
- rr io.Reader
- br io.ByteScanner
- x [scratchByteArrayLen]byte // for: get struct field name, swallow valueTypeBytes, etc
- // _ [1]uint64 // padding
- }
- func (z *ioDecReader) reset(r io.Reader, blist *bytesFreelist) {
- z.ioDecReaderCommon.reset(r, blist)
- var ok bool
- z.rr = r
- z.br, ok = r.(io.ByteScanner)
- if !ok {
- z.br = z
- z.rr = z
- }
- }
- func (z *ioDecReader) Read(p []byte) (n int, err error) {
- if len(p) == 0 {
- return
- }
- var firstByte bool
- if z.ls == unreadByteCanRead {
- z.ls = unreadByteCanUnread
- p[0] = z.l
- if len(p) == 1 {
- n = 1
- return
- }
- firstByte = true
- p = p[1:]
- }
- n, err = z.r.Read(p)
- if n > 0 {
- if err == io.EOF && n == len(p) {
- err = nil // read was successful, so postpone EOF (till next time)
- }
- z.l = p[n-1]
- z.ls = unreadByteCanUnread
- }
- if firstByte {
- n++
- }
- return
- }
- func (z *ioDecReader) ReadByte() (c byte, err error) {
- n, err := z.Read(z.b[:1])
- if n == 1 {
- c = z.b[0]
- if err == io.EOF {
- err = nil // read was successful, so postpone EOF (till next time)
- }
- }
- return
- }
- func (z *ioDecReader) UnreadByte() (err error) {
- switch z.ls {
- case unreadByteCanUnread:
- z.ls = unreadByteCanRead
- case unreadByteCanRead:
- err = errDecUnreadByteLastByteNotRead
- case unreadByteUndefined:
- err = errDecUnreadByteNothingToRead
- default:
- err = errDecUnreadByteUnknown
- }
- return
- }
- func (z *ioDecReader) readx(n uint) (bs []byte) {
- if n == 0 {
- return
- }
- if n < uint(len(z.x)) {
- bs = z.x[:n]
- } else {
- bs = make([]byte, n)
- }
- if _, err := decReadFull(z.rr, bs); err != nil {
- panic(err)
- }
- z.n += uint(len(bs))
- if z.trb {
- z.tr = append(z.tr, bs...)
- }
- return
- }
- func (z *ioDecReader) readb(bs []byte) {
- if len(bs) == 0 {
- return
- }
- if _, err := decReadFull(z.rr, bs); err != nil {
- panic(err)
- }
- z.n += uint(len(bs))
- if z.trb {
- z.tr = append(z.tr, bs...)
- }
- }
- func (z *ioDecReader) readn1eof() (b uint8, eof bool) {
- b, err := z.br.ReadByte()
- if err == nil {
- z.n++
- if z.trb {
- z.tr = append(z.tr, b)
- }
- } else if err == io.EOF {
- eof = true
- } else {
- panic(err)
- }
- return
- }
- func (z *ioDecReader) readn1() (b uint8) {
- b, err := z.br.ReadByte()
- if err == nil {
- z.n++
- if z.trb {
- z.tr = append(z.tr, b)
- }
- return
- }
- panic(err)
- }
- func (z *ioDecReader) skip(accept *bitset256) (token byte) {
- var eof bool
- // for {
- // token, eof = z.readn1eof()
- // if eof {
- // return
- // }
- // if accept.isset(token) {
- // continue
- // }
- // return
- // }
- LOOP:
- token, eof = z.readn1eof()
- if eof {
- return
- }
- if accept.isset(token) {
- goto LOOP
- }
- return
- }
- func (z *ioDecReader) readTo(accept *bitset256) []byte {
- // out = in
- // for {
- // token, eof := z.readn1eof()
- // if eof {
- // return
- // }
- // if accept.isset(token) {
- // out = append(out, token)
- // } else {
- // z.unreadn1()
- // return
- // }
- // }
- z.bufr = z.blist.check(z.bufr, 256)[:0]
- LOOP:
- token, eof := z.readn1eof()
- if eof {
- return z.bufr
- }
- if accept.isset(token) {
- // out = append(out, token)
- z.bufr = append(z.bufr, token)
- goto LOOP
- }
- z.unreadn1()
- return z.bufr
- }
- func (z *ioDecReader) readUntil(stop byte) []byte {
- // for {
- // token, eof := z.readn1eof()
- // if eof {
- // panic(io.EOF)
- // }
- // out = append(out, token)
- // if token == stop {
- // return
- // }
- // }
- z.bufr = z.blist.check(z.bufr, 256)[:0]
- LOOP:
- token, eof := z.readn1eof()
- if eof {
- panic(io.EOF)
- }
- z.bufr = append(z.bufr, token)
- if token == stop {
- return z.bufr
- }
- goto LOOP
- }
- //go:noinline
- func (z *ioDecReader) unreadn1() {
- err := z.br.UnreadByte()
- if err != nil {
- panic(err)
- }
- z.n--
- if z.trb {
- if l := len(z.tr) - 1; l >= 0 {
- z.tr = z.tr[:l]
- }
- }
- }
- // ------------------------------------
- type bufioDecReader struct {
- ioDecReaderCommon
- // _ uint64 // padding (cache-aligned)
- c uint // cursor
- buf []byte
- // bp bytesBufSlicePooler
- // err error
- }
- func (z *bufioDecReader) reset(r io.Reader, bufsize int, blist *bytesFreelist) {
- z.ioDecReaderCommon.reset(r, blist)
- z.c = 0
- if cap(z.buf) < bufsize {
- // z.bp.get(bufsize)
- // z.buf = make([]byte, 0, bufsize)
- z.buf = blist.get(bufsize)
- }
- z.buf = z.buf[:0]
- }
- // func (z *bufioDecReader) release() {
- // z.ioDecReaderCommon.release()
- // // z.bp.end()
- // }
- func (z *bufioDecReader) readb(p []byte) {
- var n = uint(copy(p, z.buf[z.c:]))
- z.n += n
- z.c += n
- if len(p) == int(n) {
- if z.trb {
- z.tr = append(z.tr, p...)
- }
- } else {
- z.readbFill(p, n)
- }
- }
- //go:noinline - fallback when z.buf is consumed
- func (z *bufioDecReader) readbFill(p0 []byte, n uint) {
- // at this point, there's nothing in z.buf to read (z.buf is fully consumed)
- p := p0[n:]
- var n2 uint
- var err error
- if len(p) > cap(z.buf) {
- n2, err = decReadFull(z.r, p)
- if err != nil {
- panic(err)
- }
- n += n2
- z.n += n2
- // always keep last byte in z.buf
- z.buf = z.buf[:1]
- z.buf[0] = p[len(p)-1]
- z.c = 1
- if z.trb {
- z.tr = append(z.tr, p0[:n]...)
- }
- return
- }
- // z.c is now 0, and len(p) <= cap(z.buf)
- LOOP:
- // for len(p) > 0 && z.err == nil {
- if len(p) > 0 {
- z.buf = z.buf[0:cap(z.buf)]
- var n1 int
- n1, err = z.r.Read(z.buf)
- n2 = uint(n1)
- if n2 == 0 && err != nil {
- panic(err)
- }
- z.buf = z.buf[:n2]
- n2 = uint(copy(p, z.buf))
- z.c = n2
- n += n2
- z.n += n2
- p = p[n2:]
- goto LOOP
- }
- if z.c == 0 {
- z.buf = z.buf[:1]
- z.buf[0] = p[len(p)-1]
- z.c = 1
- }
- if z.trb {
- z.tr = append(z.tr, p0[:n]...)
- }
- }
- func (z *bufioDecReader) readn1() (b byte) {
- // fast-path, so we elide calling into Read() most of the time
- if z.c < uint(len(z.buf)) {
- b = z.buf[z.c]
- z.c++
- z.n++
- if z.trb {
- z.tr = append(z.tr, b)
- }
- } else { // meaning z.c == len(z.buf) or greater ... so need to fill
- z.readbFill(z.b[:1], 0)
- b = z.b[0]
- }
- return
- }
- func (z *bufioDecReader) unreadn1() {
- if z.c == 0 {
- panic(errDecUnreadByteNothingToRead)
- }
- z.c--
- z.n--
- if z.trb {
- z.tr = z.tr[:len(z.tr)-1]
- }
- }
- func (z *bufioDecReader) readx(n uint) (bs []byte) {
- if n == 0 {
- // return
- } else if z.c+n <= uint(len(z.buf)) {
- bs = z.buf[z.c : z.c+n]
- z.n += n
- z.c += n
- if z.trb {
- z.tr = append(z.tr, bs...)
- }
- } else {
- bs = make([]byte, n)
- // n no longer used - can reuse
- n = uint(copy(bs, z.buf[z.c:]))
- z.n += n
- z.c += n
- z.readbFill(bs, n)
- }
- return
- }
- // func (z *bufioDecReader) doTrack(y uint) {
- // z.tr = append(z.tr, z.buf[z.c:y]...) // cost=14???
- // }
- // func (z *bufioDecReader) skipLoopFn(i uint) {
- // z.n += (i - z.c) - 1
- // i++
- // if z.trb {
- // // z.tr = append(z.tr, z.buf[z.c:i]...)
- // z.doTrack(i)
- // }
- // z.c = i
- // }
- func (z *bufioDecReader) skip(accept *bitset256) (token byte) {
- // token, _ = z.search(nil, accept, 0, 1); return
- // for i := z.c; i < len(z.buf); i++ {
- // if token = z.buf[i]; !accept.isset(token) {
- // z.skipLoopFn(i)
- // return
- // }
- // }
- i := z.c
- LOOP:
- if i < uint(len(z.buf)) {
- // inline z.skipLoopFn(i) and refactor, so cost is within inline budget
- token = z.buf[i]
- i++
- if accept.isset(token) {
- goto LOOP
- }
- z.n += i - 2 - z.c
- if z.trb {
- z.tr = append(z.tr, z.buf[z.c:i]...) // z.doTrack(i)
- }
- z.c = i
- return
- }
- return z.skipFill(accept)
- }
- func (z *bufioDecReader) skipFill(accept *bitset256) (token byte) {
- // defer func() { xdebugf("skipFill '%c'", token) }()
- z.n += uint(len(z.buf)) - z.c
- if z.trb {
- z.tr = append(z.tr, z.buf[z.c:]...)
- }
- var i, n2 int
- var err error
- for {
- z.c = 0
- z.buf = z.buf[0:cap(z.buf)]
- n2, err = z.r.Read(z.buf)
- if n2 == 0 && err != nil {
- panic(err)
- }
- z.buf = z.buf[:n2]
- for i, token = range z.buf {
- if !accept.isset(token) {
- z.n += (uint(i) - z.c) - 1
- z.loopFn(uint(i + 1))
- return
- }
- }
- // for i := 0; i < n2; i++ {
- // if token = z.buf[i]; !accept.isset(token) {
- // z.skipLoopFn(i)
- // return
- // }
- // }
- z.n += uint(n2)
- if z.trb {
- z.tr = append(z.tr, z.buf...)
- }
- }
- }
- // func (z *bufioDecReader) readLoopFn(i uint, out0 []byte) (out []byte) {
- // out = appendPool(out0, z.buf[z.c:i]...)
- // z.loopFn(i)
- // return
- // }
- func (z *bufioDecReader) loopFn(i uint) {
- if z.trb {
- z.tr = append(z.tr, z.buf[z.c:i]...) // z.doTrack(i)
- }
- z.c = i
- }
- // func (z *bufioDecReader) readToLoopFn(i uint, out0 []byte) (out []byte) {
- // // out0 is never nil
- // z.n += (i - z.c) - 1
- // return z.readLoopFn(i, out0)
- // }
- func (z *bufioDecReader) readTo(accept *bitset256) (out []byte) {
- // defer func() { xdebug2f("bufio: readTo: %s", out) }()
- // _, out = z.search(in, accept, 0, 2); return
- // for i := z.c; i < len(z.buf); i++ {
- // if !accept.isset(z.buf[i]) {
- // return z.readToLoopFn(i, nil)
- // }
- // }
- i := z.c
- LOOP:
- if i < uint(len(z.buf)) {
- if !accept.isset(z.buf[i]) {
- // return z.readToLoopFn(i, nil)
- // inline readToLoopFn here (for performance)
- z.n += (i - z.c) - 1
- out = z.buf[z.c:i]
- if z.trb {
- z.tr = append(z.tr, z.buf[z.c:i]...) // z.doTrack(i)
- }
- z.c = i
- return
- }
- i++
- goto LOOP
- }
- return z.readToFill(accept)
- }
- func (z *bufioDecReader) readToFill(accept *bitset256) []byte {
- z.bufr = z.blist.check(z.bufr, 256)[:0]
- z.n += uint(len(z.buf)) - z.c
- z.bufr = append(z.bufr, z.buf[z.c:]...)
- if z.trb {
- z.tr = append(z.tr, z.buf[z.c:]...)
- }
- var n2 int
- var err error
- for {
- z.c = 0
- z.buf = z.buf[:cap(z.buf)]
- n2, err = z.r.Read(z.buf)
- if n2 == 0 && err != nil {
- if err == io.EOF {
- return z.bufr // readTo should read until it matches or end is reached
- }
- panic(err)
- }
- z.buf = z.buf[:n2]
- for i, token := range z.buf {
- if !accept.isset(token) {
- z.n += (uint(i) - z.c) - 1
- z.bufr = append(z.bufr, z.buf[z.c:i]...)
- z.loopFn(uint(i))
- return z.bufr
- }
- }
- // for i := 0; i < n2; i++ {
- // if !accept.isset(z.buf[i]) {
- // return z.readToLoopFn(i, out)
- // }
- // }
- z.bufr = append(z.bufr, z.buf...)
- z.n += uint(n2)
- if z.trb {
- z.tr = append(z.tr, z.buf...)
- }
- }
- }
- // func (z *bufioDecReader) readUntilLoopFn(i uint, out0 []byte) (out []byte) {
- // z.n += (i - z.c) - 1
- // return z.readLoopFn(i+1, out0)
- // }
- func (z *bufioDecReader) readUntil(stop byte) (out []byte) {
- // defer func() { xdebug2f("bufio: readUntil: %s", out) }()
- // _, out = z.search(in, nil, stop, 4); return
- // for i := z.c; i < len(z.buf); i++ {
- // if z.buf[i] == stop {
- // return z.readUntilLoopFn(i, nil)
- // }
- // }
- i := z.c
- LOOP:
- if i < uint(len(z.buf)) {
- if z.buf[i] == stop {
- // inline readUntilLoopFn
- // return z.readUntilLoopFn(i, nil)
- z.n += (i - z.c) - 1
- i++
- out = z.buf[z.c:i]
- if z.trb {
- z.tr = append(z.tr, z.buf[z.c:i]...) // z.doTrack(i)
- }
- z.c = i
- return
- }
- i++
- goto LOOP
- }
- return z.readUntilFill(stop)
- }
- func (z *bufioDecReader) readUntilFill(stop byte) []byte {
- z.bufr = z.blist.check(z.bufr, 256)[:0]
- z.n += uint(len(z.buf)) - z.c
- z.bufr = append(z.bufr, z.buf[z.c:]...)
- if z.trb {
- z.tr = append(z.tr, z.buf[z.c:]...)
- }
- for {
- z.c = 0
- z.buf = z.buf[0:cap(z.buf)]
- n1, err := z.r.Read(z.buf)
- if n1 == 0 && err != nil {
- panic(err)
- }
- n2 := uint(n1)
- z.buf = z.buf[:n2]
- for i, token := range z.buf {
- if token == stop {
- z.n += (uint(i) - z.c) - 1
- z.bufr = append(z.bufr, z.buf[z.c:i+1]...)
- z.loopFn(uint(i + 1))
- return z.bufr
- }
- }
- // for i := 0; i < n2; i++ {
- // if z.buf[i] == stop {
- // return z.readUntilLoopFn(i, out)
- // }
- // }
- z.bufr = append(z.bufr, z.buf...)
- z.n += n2
- if z.trb {
- z.tr = append(z.tr, z.buf...)
- }
- }
- }
- // ------------------------------------
- // bytesDecReader is a decReader that reads off a byte slice with zero copying
- type bytesDecReader struct {
- b []byte // data
- c uint // cursor
- t uint // track start
- // a int // available
- }
- func (z *bytesDecReader) reset(in []byte) {
- z.b = in
- // z.a = len(in)
- z.c = 0
- z.t = 0
- }
- func (z *bytesDecReader) numread() uint {
- return z.c
- }
- func (z *bytesDecReader) unreadn1() {
- if z.c == 0 || len(z.b) == 0 {
- panic(errBytesDecReaderCannotUnread)
- }
- z.c--
- // z.a++
- }
- func (z *bytesDecReader) readx(n uint) (bs []byte) {
- // slicing from a non-constant start position is more expensive,
- // as more computation is required to decipher the pointer start position.
- // However, we do it only once, and it's better than reslicing both z.b and return value.
- // if n <= 0 {
- // } else if z.a == 0 {
- // panic(io.EOF)
- // } else if n > z.a {
- // panic(io.ErrUnexpectedEOF)
- // } else {
- // c0 := z.c
- // z.c = c0 + n
- // z.a = z.a - n
- // bs = z.b[c0:z.c]
- // }
- // return
- if n != 0 {
- z.c += n
- if z.c > uint(len(z.b)) {
- z.c = uint(len(z.b))
- panic(io.EOF)
- }
- bs = z.b[z.c-n : z.c]
- }
- return
- // if n == 0 {
- // } else if z.c+n > uint(len(z.b)) {
- // z.c = uint(len(z.b))
- // panic(io.EOF)
- // } else {
- // z.c += n
- // bs = z.b[z.c-n : z.c]
- // }
- // return
- // if n == 0 {
- // return
- // }
- // if z.c == uint(len(z.b)) {
- // panic(io.EOF)
- // }
- // if z.c+n > uint(len(z.b)) {
- // panic(io.ErrUnexpectedEOF)
- // }
- // // z.a -= n
- // z.c += n
- // return z.b[z.c-n : z.c]
- }
- func (z *bytesDecReader) readb(bs []byte) {
- copy(bs, z.readx(uint(len(bs))))
- }
- func (z *bytesDecReader) readn1() (v uint8) {
- if z.c == uint(len(z.b)) {
- panic(io.EOF)
- }
- v = z.b[z.c]
- z.c++
- // z.a--
- return
- }
- // func (z *bytesDecReader) readn1eof() (v uint8, eof bool) {
- // if z.a == 0 {
- // eof = true
- // return
- // }
- // v = z.b[z.c]
- // z.c++
- // z.a--
- // return
- // }
- func (z *bytesDecReader) skip(accept *bitset256) (token byte) {
- i := z.c
- // if i == len(z.b) {
- // goto END
- // // panic(io.EOF)
- // }
- // Replace loop with goto construct, so that this can be inlined
- // for i := z.c; i < blen; i++ {
- // if !accept.isset(z.b[i]) {
- // token = z.b[i]
- // i++
- // z.a -= (i - z.c)
- // z.c = i
- // return
- // }
- // }
- // i := z.c
- LOOP:
- if i < uint(len(z.b)) {
- token = z.b[i]
- i++
- if accept.isset(token) {
- goto LOOP
- }
- // z.a -= (i - z.c)
- z.c = i
- return
- }
- // END:
- panic(io.EOF)
- // // z.a = 0
- // z.c = blen
- // return
- }
- func (z *bytesDecReader) readTo(accept *bitset256) (out []byte) {
- i := z.c
- if i == uint(len(z.b)) {
- panic(io.EOF)
- }
- // Replace loop with goto construct, so that this can be inlined
- // for i := z.c; i < blen; i++ {
- // if !accept.isset(z.b[i]) {
- // out = z.b[z.c:i]
- // z.a -= (i - z.c)
- // z.c = i
- // return
- // }
- // }
- // out = z.b[z.c:]
- // z.a, z.c = 0, blen
- // return
- // i := z.c
- // LOOP:
- // if i < blen {
- // if accept.isset(z.b[i]) {
- // i++
- // goto LOOP
- // }
- // out = z.b[z.c:i]
- // z.a -= (i - z.c)
- // z.c = i
- // return
- // }
- // out = z.b[z.c:]
- // // z.a, z.c = 0, blen
- // z.a = 0
- // z.c = blen
- // return
- // c := i
- LOOP:
- if i < uint(len(z.b)) {
- if accept.isset(z.b[i]) {
- i++
- goto LOOP
- }
- }
- out = z.b[z.c:i]
- // z.a -= (i - z.c)
- z.c = i
- return // z.b[c:i]
- // z.c, i = i, z.c
- // return z.b[i:z.c]
- }
- func (z *bytesDecReader) readUntil(stop byte) (out []byte) {
- i := z.c
- // if i == len(z.b) {
- // panic(io.EOF)
- // }
- // Replace loop with goto construct, so that this can be inlined
- // for i := z.c; i < blen; i++ {
- // if z.b[i] == stop {
- // i++
- // out = z.b[z.c:i]
- // z.a -= (i - z.c)
- // z.c = i
- // return
- // }
- // }
- LOOP:
- if i < uint(len(z.b)) {
- if z.b[i] == stop {
- i++
- out = z.b[z.c:i]
- // z.a -= (i - z.c)
- z.c = i
- return
- }
- i++
- goto LOOP
- }
- // z.a = 0
- // z.c = blen
- panic(io.EOF)
- }
- func (z *bytesDecReader) track() {
- z.t = z.c
- }
- func (z *bytesDecReader) stopTrack() (bs []byte) {
- return z.b[z.t:z.c]
- }
- // --------------
- type decReaderSwitch struct {
- esep bool // has elem separators
- mtr, str bool // whether maptype or slicetype are known types
- be bool // is binary encoding
- js bool // is json handle
- jsms bool // is json handle, and MapKeyAsString
- // typ entryType
- bytes bool // is bytes reader
- bufio bool // is this a bufioDecReader?
- rb bytesDecReader
- ri *ioDecReader
- bi *bufioDecReader
- }
- // numread, track and stopTrack are always inlined, as they just check int fields, etc.
- /*
- func (z *decReaderSwitch) numread() int {
- switch z.typ {
- case entryTypeBytes:
- return z.rb.numread()
- case entryTypeIo:
- return z.ri.numread()
- default:
- return z.bi.numread()
- }
- }
- func (z *decReaderSwitch) track() {
- switch z.typ {
- case entryTypeBytes:
- z.rb.track()
- case entryTypeIo:
- z.ri.track()
- default:
- z.bi.track()
- }
- }
- func (z *decReaderSwitch) stopTrack() []byte {
- switch z.typ {
- case entryTypeBytes:
- return z.rb.stopTrack()
- case entryTypeIo:
- return z.ri.stopTrack()
- default:
- return z.bi.stopTrack()
- }
- }
- func (z *decReaderSwitch) unreadn1() {
- switch z.typ {
- case entryTypeBytes:
- z.rb.unreadn1()
- case entryTypeIo:
- z.ri.unreadn1()
- default:
- z.bi.unreadn1()
- }
- }
- func (z *decReaderSwitch) readx(n int) []byte {
- switch z.typ {
- case entryTypeBytes:
- return z.rb.readx(n)
- case entryTypeIo:
- return z.ri.readx(n)
- default:
- return z.bi.readx(n)
- }
- }
- func (z *decReaderSwitch) readb(s []byte) {
- switch z.typ {
- case entryTypeBytes:
- z.rb.readb(s)
- case entryTypeIo:
- z.ri.readb(s)
- default:
- z.bi.readb(s)
- }
- }
- func (z *decReaderSwitch) readn1() uint8 {
- switch z.typ {
- case entryTypeBytes:
- return z.rb.readn1()
- case entryTypeIo:
- return z.ri.readn1()
- default:
- return z.bi.readn1()
- }
- }
- func (z *decReaderSwitch) skip(accept *bitset256) (token byte) {
- switch z.typ {
- case entryTypeBytes:
- return z.rb.skip(accept)
- case entryTypeIo:
- return z.ri.skip(accept)
- default:
- return z.bi.skip(accept)
- }
- }
- func (z *decReaderSwitch) readTo(accept *bitset256) (out []byte) {
- switch z.typ {
- case entryTypeBytes:
- return z.rb.readTo(accept)
- case entryTypeIo:
- return z.ri.readTo(accept)
- default:
- return z.bi.readTo(accept)
- }
- }
- func (z *decReaderSwitch) readUntil(stop byte) (out []byte) {
- switch z.typ {
- case entryTypeBytes:
- return z.rb.readUntil(stop)
- case entryTypeIo:
- return z.ri.readUntil(stop)
- default:
- return z.bi.readUntil(stop)
- }
- }
- */
- // the if/else-if/else block is expensive to inline.
- // Each node of this construct costs a lot and dominates the budget.
- // Best to only do an if fast-path else block (so fast-path is inlined).
- // This is irrespective of inlineExtraCallCost set in $GOROOT/src/cmd/compile/internal/gc/inl.go
- //
- // In decReaderSwitch methods below, we delegate all IO functions into their own methods.
- // This allows for the inlining of the common path when z.bytes=true.
- // Go 1.12+ supports inlining methods with up to 1 inlined function (or 2 if no other constructs).
- // func (z *decReaderSwitch) release() {
- // if z.bytes {
- // } else if z.bufio {
- // z.bi.release()
- // } else {
- // z.ri.release()
- // }
- // }
- func (z *decReaderSwitch) numread() uint {
- if z.bytes {
- return z.rb.numread()
- } else if z.bufio {
- return z.bi.numread()
- } else {
- return z.ri.numread()
- }
- }
- func (z *decReaderSwitch) track() {
- if z.bytes {
- z.rb.track()
- } else if z.bufio {
- z.bi.track()
- } else {
- z.ri.track()
- }
- }
- func (z *decReaderSwitch) stopTrack() []byte {
- if z.bytes {
- return z.rb.stopTrack()
- } else if z.bufio {
- return z.bi.stopTrack()
- } else {
- return z.ri.stopTrack()
- }
- }
- // func (z *decReaderSwitch) unreadn1() {
- // if z.bytes {
- // z.rb.unreadn1()
- // } else {
- // z.unreadn1IO()
- // }
- // }
- // func (z *decReaderSwitch) unreadn1IO() {
- // if z.bufio {
- // z.bi.unreadn1()
- // } else {
- // z.ri.unreadn1()
- // }
- // }
- func (z *decReaderSwitch) unreadn1() {
- if z.bytes {
- z.rb.unreadn1()
- } else if z.bufio {
- z.bi.unreadn1()
- } else {
- z.ri.unreadn1() // not inlined
- }
- }
- func (z *decReaderSwitch) readx(n uint) []byte {
- if z.bytes {
- return z.rb.readx(n)
- }
- return z.readxIO(n)
- }
- func (z *decReaderSwitch) readxIO(n uint) []byte {
- if z.bufio {
- return z.bi.readx(n)
- }
- return z.ri.readx(n)
- }
- func (z *decReaderSwitch) readb(s []byte) {
- if z.bytes {
- z.rb.readb(s)
- } else {
- z.readbIO(s)
- }
- }
- //go:noinline - fallback for io, ensures z.bytes path is inlined
- func (z *decReaderSwitch) readbIO(s []byte) {
- if z.bufio {
- z.bi.readb(s)
- } else {
- z.ri.readb(s)
- }
- }
- func (z *decReaderSwitch) readn1() uint8 {
- if z.bytes {
- return z.rb.readn1()
- }
- return z.readn1IO()
- }
- func (z *decReaderSwitch) readn1IO() uint8 {
- if z.bufio {
- return z.bi.readn1()
- }
- return z.ri.readn1()
- }
- func (z *decReaderSwitch) skip(accept *bitset256) (token byte) {
- if z.bytes {
- return z.rb.skip(accept)
- }
- return z.skipIO(accept)
- }
- func (z *decReaderSwitch) skipIO(accept *bitset256) (token byte) {
- if z.bufio {
- return z.bi.skip(accept)
- }
- return z.ri.skip(accept)
- }
- func (z *decReaderSwitch) readTo(accept *bitset256) (out []byte) {
- if z.bytes {
- return z.rb.readTo(accept)
- }
- return z.readToIO(accept)
- }
- //go:noinline - fallback for io, ensures z.bytes path is inlined
- func (z *decReaderSwitch) readToIO(accept *bitset256) (out []byte) {
- if z.bufio {
- return z.bi.readTo(accept)
- }
- return z.ri.readTo(accept)
- }
- func (z *decReaderSwitch) readUntil(stop byte) (out []byte) {
- if z.bytes {
- return z.rb.readUntil(stop)
- }
- return z.readUntilIO(stop)
- }
- func (z *decReaderSwitch) readUntilIO(stop byte) (out []byte) {
- if z.bufio {
- return z.bi.readUntil(stop)
- }
- return z.ri.readUntil(stop)
- }
- // // register these here, so that staticcheck stops barfing
- // var _ = (*bytesDecReader).readTo
- // var _ = (*bytesDecReader).readUntil
|