1337 lines
35 KiB
Go
1337 lines
35 KiB
Go
/*
|
|
* Licensed to the Apache Software Foundation (ASF) under one
|
|
* or more contributor license agreements. See the NOTICE file
|
|
* distributed with this work for additional information
|
|
* regarding copyright ownership. The ASF licenses this file
|
|
* to you under the Apache License, Version 2.0 (the
|
|
* "License"); you may not use this file except in compliance
|
|
* with the License. You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing,
|
|
* software distributed under the License is distributed on an
|
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
* KIND, either express or implied. See the License for the
|
|
* specific language governing permissions and limitations
|
|
* under the License.
|
|
*/
|
|
|
|
package thrift
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"encoding/base64"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"math"
|
|
"strconv"
|
|
)
|
|
|
|
type _ParseContext int
|
|
|
|
const (
|
|
_CONTEXT_IN_TOPLEVEL _ParseContext = 1
|
|
_CONTEXT_IN_LIST_FIRST _ParseContext = 2
|
|
_CONTEXT_IN_LIST _ParseContext = 3
|
|
_CONTEXT_IN_OBJECT_FIRST _ParseContext = 4
|
|
_CONTEXT_IN_OBJECT_NEXT_KEY _ParseContext = 5
|
|
_CONTEXT_IN_OBJECT_NEXT_VALUE _ParseContext = 6
|
|
)
|
|
|
|
func (p _ParseContext) String() string {
|
|
switch p {
|
|
case _CONTEXT_IN_TOPLEVEL:
|
|
return "TOPLEVEL"
|
|
case _CONTEXT_IN_LIST_FIRST:
|
|
return "LIST-FIRST"
|
|
case _CONTEXT_IN_LIST:
|
|
return "LIST"
|
|
case _CONTEXT_IN_OBJECT_FIRST:
|
|
return "OBJECT-FIRST"
|
|
case _CONTEXT_IN_OBJECT_NEXT_KEY:
|
|
return "OBJECT-NEXT-KEY"
|
|
case _CONTEXT_IN_OBJECT_NEXT_VALUE:
|
|
return "OBJECT-NEXT-VALUE"
|
|
}
|
|
return "UNKNOWN-PARSE-CONTEXT"
|
|
}
|
|
|
|
// JSON protocol implementation for thrift.
|
|
//
|
|
// This protocol produces/consumes a simple output format
|
|
// suitable for parsing by scripting languages. It should not be
|
|
// confused with the full-featured TJSONProtocol.
|
|
//
|
|
type TSimpleJSONProtocol struct {
|
|
trans TTransport
|
|
|
|
parseContextStack []int
|
|
dumpContext []int
|
|
|
|
writer *bufio.Writer
|
|
reader *bufio.Reader
|
|
}
|
|
|
|
// Constructor
|
|
func NewTSimpleJSONProtocol(t TTransport) *TSimpleJSONProtocol {
|
|
v := &TSimpleJSONProtocol{trans: t,
|
|
writer: bufio.NewWriter(t),
|
|
reader: bufio.NewReader(t),
|
|
}
|
|
v.parseContextStack = append(v.parseContextStack, int(_CONTEXT_IN_TOPLEVEL))
|
|
v.dumpContext = append(v.dumpContext, int(_CONTEXT_IN_TOPLEVEL))
|
|
return v
|
|
}
|
|
|
|
// Factory
|
|
type TSimpleJSONProtocolFactory struct{}
|
|
|
|
func (p *TSimpleJSONProtocolFactory) GetProtocol(trans TTransport) TProtocol {
|
|
return NewTSimpleJSONProtocol(trans)
|
|
}
|
|
|
|
func NewTSimpleJSONProtocolFactory() *TSimpleJSONProtocolFactory {
|
|
return &TSimpleJSONProtocolFactory{}
|
|
}
|
|
|
|
var (
|
|
JSON_COMMA []byte
|
|
JSON_COLON []byte
|
|
JSON_LBRACE []byte
|
|
JSON_RBRACE []byte
|
|
JSON_LBRACKET []byte
|
|
JSON_RBRACKET []byte
|
|
JSON_QUOTE byte
|
|
JSON_QUOTE_BYTES []byte
|
|
JSON_NULL []byte
|
|
JSON_TRUE []byte
|
|
JSON_FALSE []byte
|
|
JSON_INFINITY string
|
|
JSON_NEGATIVE_INFINITY string
|
|
JSON_NAN string
|
|
JSON_INFINITY_BYTES []byte
|
|
JSON_NEGATIVE_INFINITY_BYTES []byte
|
|
JSON_NAN_BYTES []byte
|
|
json_nonbase_map_elem_bytes []byte
|
|
)
|
|
|
|
func init() {
|
|
JSON_COMMA = []byte{','}
|
|
JSON_COLON = []byte{':'}
|
|
JSON_LBRACE = []byte{'{'}
|
|
JSON_RBRACE = []byte{'}'}
|
|
JSON_LBRACKET = []byte{'['}
|
|
JSON_RBRACKET = []byte{']'}
|
|
JSON_QUOTE = '"'
|
|
JSON_QUOTE_BYTES = []byte{'"'}
|
|
JSON_NULL = []byte{'n', 'u', 'l', 'l'}
|
|
JSON_TRUE = []byte{'t', 'r', 'u', 'e'}
|
|
JSON_FALSE = []byte{'f', 'a', 'l', 's', 'e'}
|
|
JSON_INFINITY = "Infinity"
|
|
JSON_NEGATIVE_INFINITY = "-Infinity"
|
|
JSON_NAN = "NaN"
|
|
JSON_INFINITY_BYTES = []byte{'I', 'n', 'f', 'i', 'n', 'i', 't', 'y'}
|
|
JSON_NEGATIVE_INFINITY_BYTES = []byte{'-', 'I', 'n', 'f', 'i', 'n', 'i', 't', 'y'}
|
|
JSON_NAN_BYTES = []byte{'N', 'a', 'N'}
|
|
json_nonbase_map_elem_bytes = []byte{']', ',', '['}
|
|
}
|
|
|
|
func jsonQuote(s string) string {
|
|
b, _ := json.Marshal(s)
|
|
s1 := string(b)
|
|
return s1
|
|
}
|
|
|
|
func jsonUnquote(s string) (string, bool) {
|
|
s1 := new(string)
|
|
err := json.Unmarshal([]byte(s), s1)
|
|
return *s1, err == nil
|
|
}
|
|
|
|
func mismatch(expected, actual string) error {
|
|
return fmt.Errorf("Expected '%s' but found '%s' while parsing JSON.", expected, actual)
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteMessageBegin(name string, typeId TMessageType, seqId int32) error {
|
|
p.resetContextStack() // THRIFT-3735
|
|
if e := p.OutputListBegin(); e != nil {
|
|
return e
|
|
}
|
|
if e := p.WriteString(name); e != nil {
|
|
return e
|
|
}
|
|
if e := p.WriteByte(int8(typeId)); e != nil {
|
|
return e
|
|
}
|
|
if e := p.WriteI32(seqId); e != nil {
|
|
return e
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteMessageEnd() error {
|
|
return p.OutputListEnd()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteStructBegin(name string) error {
|
|
if e := p.OutputObjectBegin(); e != nil {
|
|
return e
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteStructEnd() error {
|
|
return p.OutputObjectEnd()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteFieldBegin(name string, typeId TType, id int16) error {
|
|
if e := p.WriteString(name); e != nil {
|
|
return e
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteFieldEnd() error {
|
|
//return p.OutputListEnd()
|
|
return nil
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteFieldStop() error { return nil }
|
|
|
|
func (p *TSimpleJSONProtocol) WriteMapBegin(keyType TType, valueType TType, size int) error {
|
|
if e := p.OutputListBegin(); e != nil {
|
|
return e
|
|
}
|
|
if e := p.WriteByte(int8(keyType)); e != nil {
|
|
return e
|
|
}
|
|
if e := p.WriteByte(int8(valueType)); e != nil {
|
|
return e
|
|
}
|
|
return p.WriteI32(int32(size))
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteMapEnd() error {
|
|
return p.OutputListEnd()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteListBegin(elemType TType, size int) error {
|
|
return p.OutputElemListBegin(elemType, size)
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteListEnd() error {
|
|
return p.OutputListEnd()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteSetBegin(elemType TType, size int) error {
|
|
return p.OutputElemListBegin(elemType, size)
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteSetEnd() error {
|
|
return p.OutputListEnd()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteBool(b bool) error {
|
|
return p.OutputBool(b)
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteByte(b int8) error {
|
|
return p.WriteI32(int32(b))
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteI16(v int16) error {
|
|
return p.WriteI32(int32(v))
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteI32(v int32) error {
|
|
return p.OutputI64(int64(v))
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteI64(v int64) error {
|
|
return p.OutputI64(int64(v))
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteDouble(v float64) error {
|
|
return p.OutputF64(v)
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteString(v string) error {
|
|
return p.OutputString(v)
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) WriteBinary(v []byte) error {
|
|
// JSON library only takes in a string,
|
|
// not an arbitrary byte array, to ensure bytes are transmitted
|
|
// efficiently we must convert this into a valid JSON string
|
|
// therefore we use base64 encoding to avoid excessive escaping/quoting
|
|
if e := p.OutputPreValue(); e != nil {
|
|
return e
|
|
}
|
|
if _, e := p.write(JSON_QUOTE_BYTES); e != nil {
|
|
return NewTProtocolException(e)
|
|
}
|
|
writer := base64.NewEncoder(base64.StdEncoding, p.writer)
|
|
if _, e := writer.Write(v); e != nil {
|
|
p.writer.Reset(p.trans) // THRIFT-3735
|
|
return NewTProtocolException(e)
|
|
}
|
|
if e := writer.Close(); e != nil {
|
|
return NewTProtocolException(e)
|
|
}
|
|
if _, e := p.write(JSON_QUOTE_BYTES); e != nil {
|
|
return NewTProtocolException(e)
|
|
}
|
|
return p.OutputPostValue()
|
|
}
|
|
|
|
// Reading methods.
|
|
func (p *TSimpleJSONProtocol) ReadMessageBegin() (name string, typeId TMessageType, seqId int32, err error) {
|
|
p.resetContextStack() // THRIFT-3735
|
|
if isNull, err := p.ParseListBegin(); isNull || err != nil {
|
|
return name, typeId, seqId, err
|
|
}
|
|
if name, err = p.ReadString(); err != nil {
|
|
return name, typeId, seqId, err
|
|
}
|
|
bTypeId, err := p.ReadByte()
|
|
typeId = TMessageType(bTypeId)
|
|
if err != nil {
|
|
return name, typeId, seqId, err
|
|
}
|
|
if seqId, err = p.ReadI32(); err != nil {
|
|
return name, typeId, seqId, err
|
|
}
|
|
return name, typeId, seqId, nil
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ReadMessageEnd() error {
|
|
return p.ParseListEnd()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ReadStructBegin() (name string, err error) {
|
|
_, err = p.ParseObjectStart()
|
|
return "", err
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ReadStructEnd() error {
|
|
return p.ParseObjectEnd()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ReadFieldBegin() (string, TType, int16, error) {
|
|
if err := p.ParsePreValue(); err != nil {
|
|
return "", STOP, 0, err
|
|
}
|
|
b, _ := p.reader.Peek(1)
|
|
if len(b) > 0 {
|
|
switch b[0] {
|
|
case JSON_RBRACE[0]:
|
|
return "", STOP, 0, nil
|
|
case JSON_QUOTE:
|
|
p.reader.ReadByte()
|
|
name, err := p.ParseStringBody()
|
|
// simplejson is not meant to be read back into thrift
|
|
// - see http://wiki.apache.org/thrift/ThriftUsageJava
|
|
// - use JSON instead
|
|
if err != nil {
|
|
return name, STOP, 0, err
|
|
}
|
|
return name, STOP, -1, p.ParsePostValue()
|
|
/*
|
|
if err = p.ParsePostValue(); err != nil {
|
|
return name, STOP, 0, err
|
|
}
|
|
if isNull, err := p.ParseListBegin(); isNull || err != nil {
|
|
return name, STOP, 0, err
|
|
}
|
|
bType, err := p.ReadByte()
|
|
thetype := TType(bType)
|
|
if err != nil {
|
|
return name, thetype, 0, err
|
|
}
|
|
id, err := p.ReadI16()
|
|
return name, thetype, id, err
|
|
*/
|
|
}
|
|
e := fmt.Errorf("Expected \"}\" or '\"', but found: '%s'", string(b))
|
|
return "", STOP, 0, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
return "", STOP, 0, NewTProtocolException(io.EOF)
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ReadFieldEnd() error {
|
|
return nil
|
|
//return p.ParseListEnd()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ReadMapBegin() (keyType TType, valueType TType, size int, e error) {
|
|
if isNull, e := p.ParseListBegin(); isNull || e != nil {
|
|
return VOID, VOID, 0, e
|
|
}
|
|
|
|
// read keyType
|
|
bKeyType, e := p.ReadByte()
|
|
keyType = TType(bKeyType)
|
|
if e != nil {
|
|
return keyType, valueType, size, e
|
|
}
|
|
|
|
// read valueType
|
|
bValueType, e := p.ReadByte()
|
|
valueType = TType(bValueType)
|
|
if e != nil {
|
|
return keyType, valueType, size, e
|
|
}
|
|
|
|
// read size
|
|
iSize, err := p.ReadI64()
|
|
size = int(iSize)
|
|
return keyType, valueType, size, err
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ReadMapEnd() error {
|
|
return p.ParseListEnd()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ReadListBegin() (elemType TType, size int, e error) {
|
|
return p.ParseElemListBegin()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ReadListEnd() error {
|
|
return p.ParseListEnd()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ReadSetBegin() (elemType TType, size int, e error) {
|
|
return p.ParseElemListBegin()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ReadSetEnd() error {
|
|
return p.ParseListEnd()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ReadBool() (bool, error) {
|
|
var value bool
|
|
|
|
if err := p.ParsePreValue(); err != nil {
|
|
return value, err
|
|
}
|
|
f, _ := p.reader.Peek(1)
|
|
if len(f) > 0 {
|
|
switch f[0] {
|
|
case JSON_TRUE[0]:
|
|
b := make([]byte, len(JSON_TRUE))
|
|
_, err := p.reader.Read(b)
|
|
if err != nil {
|
|
return false, NewTProtocolException(err)
|
|
}
|
|
if string(b) == string(JSON_TRUE) {
|
|
value = true
|
|
} else {
|
|
e := fmt.Errorf("Expected \"true\" but found: %s", string(b))
|
|
return value, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
break
|
|
case JSON_FALSE[0]:
|
|
b := make([]byte, len(JSON_FALSE))
|
|
_, err := p.reader.Read(b)
|
|
if err != nil {
|
|
return false, NewTProtocolException(err)
|
|
}
|
|
if string(b) == string(JSON_FALSE) {
|
|
value = false
|
|
} else {
|
|
e := fmt.Errorf("Expected \"false\" but found: %s", string(b))
|
|
return value, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
break
|
|
case JSON_NULL[0]:
|
|
b := make([]byte, len(JSON_NULL))
|
|
_, err := p.reader.Read(b)
|
|
if err != nil {
|
|
return false, NewTProtocolException(err)
|
|
}
|
|
if string(b) == string(JSON_NULL) {
|
|
value = false
|
|
} else {
|
|
e := fmt.Errorf("Expected \"null\" but found: %s", string(b))
|
|
return value, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
default:
|
|
e := fmt.Errorf("Expected \"true\", \"false\", or \"null\" but found: %s", string(f))
|
|
return value, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
}
|
|
return value, p.ParsePostValue()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ReadByte() (int8, error) {
|
|
v, err := p.ReadI64()
|
|
return int8(v), err
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ReadI16() (int16, error) {
|
|
v, err := p.ReadI64()
|
|
return int16(v), err
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ReadI32() (int32, error) {
|
|
v, err := p.ReadI64()
|
|
return int32(v), err
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ReadI64() (int64, error) {
|
|
v, _, err := p.ParseI64()
|
|
return v, err
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ReadDouble() (float64, error) {
|
|
v, _, err := p.ParseF64()
|
|
return v, err
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ReadString() (string, error) {
|
|
var v string
|
|
if err := p.ParsePreValue(); err != nil {
|
|
return v, err
|
|
}
|
|
f, _ := p.reader.Peek(1)
|
|
if len(f) > 0 && f[0] == JSON_QUOTE {
|
|
p.reader.ReadByte()
|
|
value, err := p.ParseStringBody()
|
|
v = value
|
|
if err != nil {
|
|
return v, err
|
|
}
|
|
} else if len(f) > 0 && f[0] == JSON_NULL[0] {
|
|
b := make([]byte, len(JSON_NULL))
|
|
_, err := p.reader.Read(b)
|
|
if err != nil {
|
|
return v, NewTProtocolException(err)
|
|
}
|
|
if string(b) != string(JSON_NULL) {
|
|
e := fmt.Errorf("Expected a JSON string, found unquoted data started with %s", string(b))
|
|
return v, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
} else {
|
|
e := fmt.Errorf("Expected a JSON string, found unquoted data started with %s", string(f))
|
|
return v, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
return v, p.ParsePostValue()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ReadBinary() ([]byte, error) {
|
|
var v []byte
|
|
if err := p.ParsePreValue(); err != nil {
|
|
return nil, err
|
|
}
|
|
f, _ := p.reader.Peek(1)
|
|
if len(f) > 0 && f[0] == JSON_QUOTE {
|
|
p.reader.ReadByte()
|
|
value, err := p.ParseBase64EncodedBody()
|
|
v = value
|
|
if err != nil {
|
|
return v, err
|
|
}
|
|
} else if len(f) > 0 && f[0] == JSON_NULL[0] {
|
|
b := make([]byte, len(JSON_NULL))
|
|
_, err := p.reader.Read(b)
|
|
if err != nil {
|
|
return v, NewTProtocolException(err)
|
|
}
|
|
if string(b) != string(JSON_NULL) {
|
|
e := fmt.Errorf("Expected a JSON string, found unquoted data started with %s", string(b))
|
|
return v, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
} else {
|
|
e := fmt.Errorf("Expected a JSON string, found unquoted data started with %s", string(f))
|
|
return v, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
|
|
return v, p.ParsePostValue()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) Flush() (err error) {
|
|
return NewTProtocolException(p.writer.Flush())
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) Skip(fieldType TType) (err error) {
|
|
return SkipDefaultDepth(p, fieldType)
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) Transport() TTransport {
|
|
return p.trans
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) OutputPreValue() error {
|
|
cxt := _ParseContext(p.dumpContext[len(p.dumpContext)-1])
|
|
switch cxt {
|
|
case _CONTEXT_IN_LIST, _CONTEXT_IN_OBJECT_NEXT_KEY:
|
|
if _, e := p.write(JSON_COMMA); e != nil {
|
|
return NewTProtocolException(e)
|
|
}
|
|
break
|
|
case _CONTEXT_IN_OBJECT_NEXT_VALUE:
|
|
if _, e := p.write(JSON_COLON); e != nil {
|
|
return NewTProtocolException(e)
|
|
}
|
|
break
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) OutputPostValue() error {
|
|
cxt := _ParseContext(p.dumpContext[len(p.dumpContext)-1])
|
|
switch cxt {
|
|
case _CONTEXT_IN_LIST_FIRST:
|
|
p.dumpContext = p.dumpContext[:len(p.dumpContext)-1]
|
|
p.dumpContext = append(p.dumpContext, int(_CONTEXT_IN_LIST))
|
|
break
|
|
case _CONTEXT_IN_OBJECT_FIRST:
|
|
p.dumpContext = p.dumpContext[:len(p.dumpContext)-1]
|
|
p.dumpContext = append(p.dumpContext, int(_CONTEXT_IN_OBJECT_NEXT_VALUE))
|
|
break
|
|
case _CONTEXT_IN_OBJECT_NEXT_KEY:
|
|
p.dumpContext = p.dumpContext[:len(p.dumpContext)-1]
|
|
p.dumpContext = append(p.dumpContext, int(_CONTEXT_IN_OBJECT_NEXT_VALUE))
|
|
break
|
|
case _CONTEXT_IN_OBJECT_NEXT_VALUE:
|
|
p.dumpContext = p.dumpContext[:len(p.dumpContext)-1]
|
|
p.dumpContext = append(p.dumpContext, int(_CONTEXT_IN_OBJECT_NEXT_KEY))
|
|
break
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) OutputBool(value bool) error {
|
|
if e := p.OutputPreValue(); e != nil {
|
|
return e
|
|
}
|
|
var v string
|
|
if value {
|
|
v = string(JSON_TRUE)
|
|
} else {
|
|
v = string(JSON_FALSE)
|
|
}
|
|
switch _ParseContext(p.dumpContext[len(p.dumpContext)-1]) {
|
|
case _CONTEXT_IN_OBJECT_FIRST, _CONTEXT_IN_OBJECT_NEXT_KEY:
|
|
v = jsonQuote(v)
|
|
default:
|
|
}
|
|
if e := p.OutputStringData(v); e != nil {
|
|
return e
|
|
}
|
|
return p.OutputPostValue()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) OutputNull() error {
|
|
if e := p.OutputPreValue(); e != nil {
|
|
return e
|
|
}
|
|
if _, e := p.write(JSON_NULL); e != nil {
|
|
return NewTProtocolException(e)
|
|
}
|
|
return p.OutputPostValue()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) OutputF64(value float64) error {
|
|
if e := p.OutputPreValue(); e != nil {
|
|
return e
|
|
}
|
|
var v string
|
|
if math.IsNaN(value) {
|
|
v = string(JSON_QUOTE) + JSON_NAN + string(JSON_QUOTE)
|
|
} else if math.IsInf(value, 1) {
|
|
v = string(JSON_QUOTE) + JSON_INFINITY + string(JSON_QUOTE)
|
|
} else if math.IsInf(value, -1) {
|
|
v = string(JSON_QUOTE) + JSON_NEGATIVE_INFINITY + string(JSON_QUOTE)
|
|
} else {
|
|
v = strconv.FormatFloat(value, 'g', -1, 64)
|
|
switch _ParseContext(p.dumpContext[len(p.dumpContext)-1]) {
|
|
case _CONTEXT_IN_OBJECT_FIRST, _CONTEXT_IN_OBJECT_NEXT_KEY:
|
|
v = string(JSON_QUOTE) + v + string(JSON_QUOTE)
|
|
default:
|
|
}
|
|
}
|
|
if e := p.OutputStringData(v); e != nil {
|
|
return e
|
|
}
|
|
return p.OutputPostValue()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) OutputI64(value int64) error {
|
|
if e := p.OutputPreValue(); e != nil {
|
|
return e
|
|
}
|
|
v := strconv.FormatInt(value, 10)
|
|
switch _ParseContext(p.dumpContext[len(p.dumpContext)-1]) {
|
|
case _CONTEXT_IN_OBJECT_FIRST, _CONTEXT_IN_OBJECT_NEXT_KEY:
|
|
v = jsonQuote(v)
|
|
default:
|
|
}
|
|
if e := p.OutputStringData(v); e != nil {
|
|
return e
|
|
}
|
|
return p.OutputPostValue()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) OutputString(s string) error {
|
|
if e := p.OutputPreValue(); e != nil {
|
|
return e
|
|
}
|
|
if e := p.OutputStringData(jsonQuote(s)); e != nil {
|
|
return e
|
|
}
|
|
return p.OutputPostValue()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) OutputStringData(s string) error {
|
|
_, e := p.write([]byte(s))
|
|
return NewTProtocolException(e)
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) OutputObjectBegin() error {
|
|
if e := p.OutputPreValue(); e != nil {
|
|
return e
|
|
}
|
|
if _, e := p.write(JSON_LBRACE); e != nil {
|
|
return NewTProtocolException(e)
|
|
}
|
|
p.dumpContext = append(p.dumpContext, int(_CONTEXT_IN_OBJECT_FIRST))
|
|
return nil
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) OutputObjectEnd() error {
|
|
if _, e := p.write(JSON_RBRACE); e != nil {
|
|
return NewTProtocolException(e)
|
|
}
|
|
p.dumpContext = p.dumpContext[:len(p.dumpContext)-1]
|
|
if e := p.OutputPostValue(); e != nil {
|
|
return e
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) OutputListBegin() error {
|
|
if e := p.OutputPreValue(); e != nil {
|
|
return e
|
|
}
|
|
if _, e := p.write(JSON_LBRACKET); e != nil {
|
|
return NewTProtocolException(e)
|
|
}
|
|
p.dumpContext = append(p.dumpContext, int(_CONTEXT_IN_LIST_FIRST))
|
|
return nil
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) OutputListEnd() error {
|
|
if _, e := p.write(JSON_RBRACKET); e != nil {
|
|
return NewTProtocolException(e)
|
|
}
|
|
p.dumpContext = p.dumpContext[:len(p.dumpContext)-1]
|
|
if e := p.OutputPostValue(); e != nil {
|
|
return e
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) OutputElemListBegin(elemType TType, size int) error {
|
|
if e := p.OutputListBegin(); e != nil {
|
|
return e
|
|
}
|
|
if e := p.WriteByte(int8(elemType)); e != nil {
|
|
return e
|
|
}
|
|
if e := p.WriteI64(int64(size)); e != nil {
|
|
return e
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ParsePreValue() error {
|
|
if e := p.readNonSignificantWhitespace(); e != nil {
|
|
return NewTProtocolException(e)
|
|
}
|
|
cxt := _ParseContext(p.parseContextStack[len(p.parseContextStack)-1])
|
|
b, _ := p.reader.Peek(1)
|
|
switch cxt {
|
|
case _CONTEXT_IN_LIST:
|
|
if len(b) > 0 {
|
|
switch b[0] {
|
|
case JSON_RBRACKET[0]:
|
|
return nil
|
|
case JSON_COMMA[0]:
|
|
p.reader.ReadByte()
|
|
if e := p.readNonSignificantWhitespace(); e != nil {
|
|
return NewTProtocolException(e)
|
|
}
|
|
return nil
|
|
default:
|
|
e := fmt.Errorf("Expected \"]\" or \",\" in list context, but found \"%s\"", string(b))
|
|
return NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
}
|
|
break
|
|
case _CONTEXT_IN_OBJECT_NEXT_KEY:
|
|
if len(b) > 0 {
|
|
switch b[0] {
|
|
case JSON_RBRACE[0]:
|
|
return nil
|
|
case JSON_COMMA[0]:
|
|
p.reader.ReadByte()
|
|
if e := p.readNonSignificantWhitespace(); e != nil {
|
|
return NewTProtocolException(e)
|
|
}
|
|
return nil
|
|
default:
|
|
e := fmt.Errorf("Expected \"}\" or \",\" in object context, but found \"%s\"", string(b))
|
|
return NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
}
|
|
break
|
|
case _CONTEXT_IN_OBJECT_NEXT_VALUE:
|
|
if len(b) > 0 {
|
|
switch b[0] {
|
|
case JSON_COLON[0]:
|
|
p.reader.ReadByte()
|
|
if e := p.readNonSignificantWhitespace(); e != nil {
|
|
return NewTProtocolException(e)
|
|
}
|
|
return nil
|
|
default:
|
|
e := fmt.Errorf("Expected \":\" in object context, but found \"%s\"", string(b))
|
|
return NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
}
|
|
break
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ParsePostValue() error {
|
|
if e := p.readNonSignificantWhitespace(); e != nil {
|
|
return NewTProtocolException(e)
|
|
}
|
|
cxt := _ParseContext(p.parseContextStack[len(p.parseContextStack)-1])
|
|
switch cxt {
|
|
case _CONTEXT_IN_LIST_FIRST:
|
|
p.parseContextStack = p.parseContextStack[:len(p.parseContextStack)-1]
|
|
p.parseContextStack = append(p.parseContextStack, int(_CONTEXT_IN_LIST))
|
|
break
|
|
case _CONTEXT_IN_OBJECT_FIRST, _CONTEXT_IN_OBJECT_NEXT_KEY:
|
|
p.parseContextStack = p.parseContextStack[:len(p.parseContextStack)-1]
|
|
p.parseContextStack = append(p.parseContextStack, int(_CONTEXT_IN_OBJECT_NEXT_VALUE))
|
|
break
|
|
case _CONTEXT_IN_OBJECT_NEXT_VALUE:
|
|
p.parseContextStack = p.parseContextStack[:len(p.parseContextStack)-1]
|
|
p.parseContextStack = append(p.parseContextStack, int(_CONTEXT_IN_OBJECT_NEXT_KEY))
|
|
break
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) readNonSignificantWhitespace() error {
|
|
for {
|
|
b, _ := p.reader.Peek(1)
|
|
if len(b) < 1 {
|
|
return nil
|
|
}
|
|
switch b[0] {
|
|
case ' ', '\r', '\n', '\t':
|
|
p.reader.ReadByte()
|
|
continue
|
|
default:
|
|
break
|
|
}
|
|
break
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ParseStringBody() (string, error) {
|
|
line, err := p.reader.ReadString(JSON_QUOTE)
|
|
if err != nil {
|
|
return "", NewTProtocolException(err)
|
|
}
|
|
l := len(line)
|
|
// count number of escapes to see if we need to keep going
|
|
i := 1
|
|
for ; i < l; i++ {
|
|
if line[l-i-1] != '\\' {
|
|
break
|
|
}
|
|
}
|
|
if i&0x01 == 1 {
|
|
v, ok := jsonUnquote(string(JSON_QUOTE) + line)
|
|
if !ok {
|
|
return "", NewTProtocolException(err)
|
|
}
|
|
return v, nil
|
|
}
|
|
s, err := p.ParseQuotedStringBody()
|
|
if err != nil {
|
|
return "", NewTProtocolException(err)
|
|
}
|
|
str := string(JSON_QUOTE) + line + s
|
|
v, ok := jsonUnquote(str)
|
|
if !ok {
|
|
e := fmt.Errorf("Unable to parse as JSON string %s", str)
|
|
return "", NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
return v, nil
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ParseQuotedStringBody() (string, error) {
|
|
line, err := p.reader.ReadString(JSON_QUOTE)
|
|
if err != nil {
|
|
return "", NewTProtocolException(err)
|
|
}
|
|
l := len(line)
|
|
// count number of escapes to see if we need to keep going
|
|
i := 1
|
|
for ; i < l; i++ {
|
|
if line[l-i-1] != '\\' {
|
|
break
|
|
}
|
|
}
|
|
if i&0x01 == 1 {
|
|
return line, nil
|
|
}
|
|
s, err := p.ParseQuotedStringBody()
|
|
if err != nil {
|
|
return "", NewTProtocolException(err)
|
|
}
|
|
v := line + s
|
|
return v, nil
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ParseBase64EncodedBody() ([]byte, error) {
|
|
line, err := p.reader.ReadBytes(JSON_QUOTE)
|
|
if err != nil {
|
|
return line, NewTProtocolException(err)
|
|
}
|
|
line2 := line[0 : len(line)-1]
|
|
l := len(line2)
|
|
if (l % 4) != 0 {
|
|
pad := 4 - (l % 4)
|
|
fill := [...]byte{'=', '=', '='}
|
|
line2 = append(line2, fill[:pad]...)
|
|
l = len(line2)
|
|
}
|
|
output := make([]byte, base64.StdEncoding.DecodedLen(l))
|
|
n, err := base64.StdEncoding.Decode(output, line2)
|
|
return output[0:n], NewTProtocolException(err)
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ParseI64() (int64, bool, error) {
|
|
if err := p.ParsePreValue(); err != nil {
|
|
return 0, false, err
|
|
}
|
|
var value int64
|
|
var isnull bool
|
|
if p.safePeekContains(JSON_NULL) {
|
|
p.reader.Read(make([]byte, len(JSON_NULL)))
|
|
isnull = true
|
|
} else {
|
|
num, err := p.readNumeric()
|
|
isnull = (num == nil)
|
|
if !isnull {
|
|
value = num.Int64()
|
|
}
|
|
if err != nil {
|
|
return value, isnull, err
|
|
}
|
|
}
|
|
return value, isnull, p.ParsePostValue()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ParseF64() (float64, bool, error) {
|
|
if err := p.ParsePreValue(); err != nil {
|
|
return 0, false, err
|
|
}
|
|
var value float64
|
|
var isnull bool
|
|
if p.safePeekContains(JSON_NULL) {
|
|
p.reader.Read(make([]byte, len(JSON_NULL)))
|
|
isnull = true
|
|
} else {
|
|
num, err := p.readNumeric()
|
|
isnull = (num == nil)
|
|
if !isnull {
|
|
value = num.Float64()
|
|
}
|
|
if err != nil {
|
|
return value, isnull, err
|
|
}
|
|
}
|
|
return value, isnull, p.ParsePostValue()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ParseObjectStart() (bool, error) {
|
|
if err := p.ParsePreValue(); err != nil {
|
|
return false, err
|
|
}
|
|
var b []byte
|
|
b, err := p.reader.Peek(1)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
if len(b) > 0 && b[0] == JSON_LBRACE[0] {
|
|
p.reader.ReadByte()
|
|
p.parseContextStack = append(p.parseContextStack, int(_CONTEXT_IN_OBJECT_FIRST))
|
|
return false, nil
|
|
} else if p.safePeekContains(JSON_NULL) {
|
|
return true, nil
|
|
}
|
|
e := fmt.Errorf("Expected '{' or null, but found '%s'", string(b))
|
|
return false, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ParseObjectEnd() error {
|
|
if isNull, err := p.readIfNull(); isNull || err != nil {
|
|
return err
|
|
}
|
|
cxt := _ParseContext(p.parseContextStack[len(p.parseContextStack)-1])
|
|
if (cxt != _CONTEXT_IN_OBJECT_FIRST) && (cxt != _CONTEXT_IN_OBJECT_NEXT_KEY) {
|
|
e := fmt.Errorf("Expected to be in the Object Context, but not in Object Context (%d)", cxt)
|
|
return NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
line, err := p.reader.ReadString(JSON_RBRACE[0])
|
|
if err != nil {
|
|
return NewTProtocolException(err)
|
|
}
|
|
for _, char := range line {
|
|
switch char {
|
|
default:
|
|
e := fmt.Errorf("Expecting end of object \"}\", but found: \"%s\"", line)
|
|
return NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
case ' ', '\n', '\r', '\t', '}':
|
|
break
|
|
}
|
|
}
|
|
p.parseContextStack = p.parseContextStack[:len(p.parseContextStack)-1]
|
|
return p.ParsePostValue()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ParseListBegin() (isNull bool, err error) {
|
|
if e := p.ParsePreValue(); e != nil {
|
|
return false, e
|
|
}
|
|
var b []byte
|
|
b, err = p.reader.Peek(1)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
if len(b) >= 1 && b[0] == JSON_LBRACKET[0] {
|
|
p.parseContextStack = append(p.parseContextStack, int(_CONTEXT_IN_LIST_FIRST))
|
|
p.reader.ReadByte()
|
|
isNull = false
|
|
} else if p.safePeekContains(JSON_NULL) {
|
|
isNull = true
|
|
} else {
|
|
err = fmt.Errorf("Expected \"null\" or \"[\", received %q", b)
|
|
}
|
|
return isNull, NewTProtocolExceptionWithType(INVALID_DATA, err)
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ParseElemListBegin() (elemType TType, size int, e error) {
|
|
if isNull, e := p.ParseListBegin(); isNull || e != nil {
|
|
return VOID, 0, e
|
|
}
|
|
bElemType, err := p.ReadByte()
|
|
elemType = TType(bElemType)
|
|
if err != nil {
|
|
return elemType, size, err
|
|
}
|
|
nSize, err2 := p.ReadI64()
|
|
size = int(nSize)
|
|
return elemType, size, err2
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) ParseListEnd() error {
|
|
if isNull, err := p.readIfNull(); isNull || err != nil {
|
|
return err
|
|
}
|
|
cxt := _ParseContext(p.parseContextStack[len(p.parseContextStack)-1])
|
|
if cxt != _CONTEXT_IN_LIST {
|
|
e := fmt.Errorf("Expected to be in the List Context, but not in List Context (%d)", cxt)
|
|
return NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
line, err := p.reader.ReadString(JSON_RBRACKET[0])
|
|
if err != nil {
|
|
return NewTProtocolException(err)
|
|
}
|
|
for _, char := range line {
|
|
switch char {
|
|
default:
|
|
e := fmt.Errorf("Expecting end of list \"]\", but found: \"%s\"", line)
|
|
return NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
case ' ', '\n', '\r', '\t', rune(JSON_RBRACKET[0]):
|
|
break
|
|
}
|
|
}
|
|
p.parseContextStack = p.parseContextStack[:len(p.parseContextStack)-1]
|
|
if _ParseContext(p.parseContextStack[len(p.parseContextStack)-1]) == _CONTEXT_IN_TOPLEVEL {
|
|
return nil
|
|
}
|
|
return p.ParsePostValue()
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) readSingleValue() (interface{}, TType, error) {
|
|
e := p.readNonSignificantWhitespace()
|
|
if e != nil {
|
|
return nil, VOID, NewTProtocolException(e)
|
|
}
|
|
b, e := p.reader.Peek(1)
|
|
if len(b) > 0 {
|
|
c := b[0]
|
|
switch c {
|
|
case JSON_NULL[0]:
|
|
buf := make([]byte, len(JSON_NULL))
|
|
_, e := p.reader.Read(buf)
|
|
if e != nil {
|
|
return nil, VOID, NewTProtocolException(e)
|
|
}
|
|
if string(JSON_NULL) != string(buf) {
|
|
e = mismatch(string(JSON_NULL), string(buf))
|
|
return nil, VOID, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
return nil, VOID, nil
|
|
case JSON_QUOTE:
|
|
p.reader.ReadByte()
|
|
v, e := p.ParseStringBody()
|
|
if e != nil {
|
|
return v, UTF8, NewTProtocolException(e)
|
|
}
|
|
if v == JSON_INFINITY {
|
|
return INFINITY, DOUBLE, nil
|
|
} else if v == JSON_NEGATIVE_INFINITY {
|
|
return NEGATIVE_INFINITY, DOUBLE, nil
|
|
} else if v == JSON_NAN {
|
|
return NAN, DOUBLE, nil
|
|
}
|
|
return v, UTF8, nil
|
|
case JSON_TRUE[0]:
|
|
buf := make([]byte, len(JSON_TRUE))
|
|
_, e := p.reader.Read(buf)
|
|
if e != nil {
|
|
return true, BOOL, NewTProtocolException(e)
|
|
}
|
|
if string(JSON_TRUE) != string(buf) {
|
|
e := mismatch(string(JSON_TRUE), string(buf))
|
|
return true, BOOL, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
return true, BOOL, nil
|
|
case JSON_FALSE[0]:
|
|
buf := make([]byte, len(JSON_FALSE))
|
|
_, e := p.reader.Read(buf)
|
|
if e != nil {
|
|
return false, BOOL, NewTProtocolException(e)
|
|
}
|
|
if string(JSON_FALSE) != string(buf) {
|
|
e := mismatch(string(JSON_FALSE), string(buf))
|
|
return false, BOOL, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
return false, BOOL, nil
|
|
case JSON_LBRACKET[0]:
|
|
_, e := p.reader.ReadByte()
|
|
return make([]interface{}, 0), LIST, NewTProtocolException(e)
|
|
case JSON_LBRACE[0]:
|
|
_, e := p.reader.ReadByte()
|
|
return make(map[string]interface{}), STRUCT, NewTProtocolException(e)
|
|
case '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'e', 'E', '.', '+', '-', JSON_INFINITY[0], JSON_NAN[0]:
|
|
// assume numeric
|
|
v, e := p.readNumeric()
|
|
return v, DOUBLE, e
|
|
default:
|
|
e := fmt.Errorf("Expected element in list but found '%s' while parsing JSON.", string(c))
|
|
return nil, VOID, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
}
|
|
e = fmt.Errorf("Cannot read a single element while parsing JSON.")
|
|
return nil, VOID, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) readIfNull() (bool, error) {
|
|
cont := true
|
|
for cont {
|
|
b, _ := p.reader.Peek(1)
|
|
if len(b) < 1 {
|
|
return false, nil
|
|
}
|
|
switch b[0] {
|
|
default:
|
|
return false, nil
|
|
case JSON_NULL[0]:
|
|
cont = false
|
|
break
|
|
case ' ', '\n', '\r', '\t':
|
|
p.reader.ReadByte()
|
|
break
|
|
}
|
|
}
|
|
if p.safePeekContains(JSON_NULL) {
|
|
p.reader.Read(make([]byte, len(JSON_NULL)))
|
|
return true, nil
|
|
}
|
|
return false, nil
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) readQuoteIfNext() {
|
|
b, _ := p.reader.Peek(1)
|
|
if len(b) > 0 && b[0] == JSON_QUOTE {
|
|
p.reader.ReadByte()
|
|
}
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) readNumeric() (Numeric, error) {
|
|
isNull, err := p.readIfNull()
|
|
if isNull || err != nil {
|
|
return NUMERIC_NULL, err
|
|
}
|
|
hasDecimalPoint := false
|
|
nextCanBeSign := true
|
|
hasE := false
|
|
MAX_LEN := 40
|
|
buf := bytes.NewBuffer(make([]byte, 0, MAX_LEN))
|
|
continueFor := true
|
|
inQuotes := false
|
|
for continueFor {
|
|
c, err := p.reader.ReadByte()
|
|
if err != nil {
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
return NUMERIC_NULL, NewTProtocolException(err)
|
|
}
|
|
switch c {
|
|
case '0', '1', '2', '3', '4', '5', '6', '7', '8', '9':
|
|
buf.WriteByte(c)
|
|
nextCanBeSign = false
|
|
case '.':
|
|
if hasDecimalPoint {
|
|
e := fmt.Errorf("Unable to parse number with multiple decimal points '%s.'", buf.String())
|
|
return NUMERIC_NULL, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
if hasE {
|
|
e := fmt.Errorf("Unable to parse number with decimal points in the exponent '%s.'", buf.String())
|
|
return NUMERIC_NULL, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
buf.WriteByte(c)
|
|
hasDecimalPoint, nextCanBeSign = true, false
|
|
case 'e', 'E':
|
|
if hasE {
|
|
e := fmt.Errorf("Unable to parse number with multiple exponents '%s%c'", buf.String(), c)
|
|
return NUMERIC_NULL, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
buf.WriteByte(c)
|
|
hasE, nextCanBeSign = true, true
|
|
case '-', '+':
|
|
if !nextCanBeSign {
|
|
e := fmt.Errorf("Negative sign within number")
|
|
return NUMERIC_NULL, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
buf.WriteByte(c)
|
|
nextCanBeSign = false
|
|
case ' ', 0, '\t', '\n', '\r', JSON_RBRACE[0], JSON_RBRACKET[0], JSON_COMMA[0], JSON_COLON[0]:
|
|
p.reader.UnreadByte()
|
|
continueFor = false
|
|
case JSON_NAN[0]:
|
|
if buf.Len() == 0 {
|
|
buffer := make([]byte, len(JSON_NAN))
|
|
buffer[0] = c
|
|
_, e := p.reader.Read(buffer[1:])
|
|
if e != nil {
|
|
return NUMERIC_NULL, NewTProtocolException(e)
|
|
}
|
|
if JSON_NAN != string(buffer) {
|
|
e := mismatch(JSON_NAN, string(buffer))
|
|
return NUMERIC_NULL, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
if inQuotes {
|
|
p.readQuoteIfNext()
|
|
}
|
|
return NAN, nil
|
|
} else {
|
|
e := fmt.Errorf("Unable to parse number starting with character '%c'", c)
|
|
return NUMERIC_NULL, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
case JSON_INFINITY[0]:
|
|
if buf.Len() == 0 || (buf.Len() == 1 && buf.Bytes()[0] == '+') {
|
|
buffer := make([]byte, len(JSON_INFINITY))
|
|
buffer[0] = c
|
|
_, e := p.reader.Read(buffer[1:])
|
|
if e != nil {
|
|
return NUMERIC_NULL, NewTProtocolException(e)
|
|
}
|
|
if JSON_INFINITY != string(buffer) {
|
|
e := mismatch(JSON_INFINITY, string(buffer))
|
|
return NUMERIC_NULL, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
if inQuotes {
|
|
p.readQuoteIfNext()
|
|
}
|
|
return INFINITY, nil
|
|
} else if buf.Len() == 1 && buf.Bytes()[0] == JSON_NEGATIVE_INFINITY[0] {
|
|
buffer := make([]byte, len(JSON_NEGATIVE_INFINITY))
|
|
buffer[0] = JSON_NEGATIVE_INFINITY[0]
|
|
buffer[1] = c
|
|
_, e := p.reader.Read(buffer[2:])
|
|
if e != nil {
|
|
return NUMERIC_NULL, NewTProtocolException(e)
|
|
}
|
|
if JSON_NEGATIVE_INFINITY != string(buffer) {
|
|
e := mismatch(JSON_NEGATIVE_INFINITY, string(buffer))
|
|
return NUMERIC_NULL, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
if inQuotes {
|
|
p.readQuoteIfNext()
|
|
}
|
|
return NEGATIVE_INFINITY, nil
|
|
} else {
|
|
e := fmt.Errorf("Unable to parse number starting with character '%c' due to existing buffer %s", c, buf.String())
|
|
return NUMERIC_NULL, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
case JSON_QUOTE:
|
|
if !inQuotes {
|
|
inQuotes = true
|
|
} else {
|
|
break
|
|
}
|
|
default:
|
|
e := fmt.Errorf("Unable to parse number starting with character '%c'", c)
|
|
return NUMERIC_NULL, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
}
|
|
if buf.Len() == 0 {
|
|
e := fmt.Errorf("Unable to parse number from empty string ''")
|
|
return NUMERIC_NULL, NewTProtocolExceptionWithType(INVALID_DATA, e)
|
|
}
|
|
return NewNumericFromJSONString(buf.String(), false), nil
|
|
}
|
|
|
|
// Safely peeks into the buffer, reading only what is necessary
|
|
func (p *TSimpleJSONProtocol) safePeekContains(b []byte) bool {
|
|
for i := 0; i < len(b); i++ {
|
|
a, _ := p.reader.Peek(i + 1)
|
|
if len(a) == 0 || a[i] != b[i] {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
// Reset the context stack to its initial state.
|
|
func (p *TSimpleJSONProtocol) resetContextStack() {
|
|
p.parseContextStack = []int{int(_CONTEXT_IN_TOPLEVEL)}
|
|
p.dumpContext = []int{int(_CONTEXT_IN_TOPLEVEL)}
|
|
}
|
|
|
|
func (p *TSimpleJSONProtocol) write(b []byte) (int, error) {
|
|
n, err := p.writer.Write(b)
|
|
if err != nil {
|
|
p.writer.Reset(p.trans) // THRIFT-3735
|
|
}
|
|
return n, err
|
|
}
|