forked from wundergraph/graphql-go-tools
/
jsonvaluetype.go
84 lines (74 loc) · 1.92 KB
/
jsonvaluetype.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
//go:generate stringer -type=JSONValueType
package execution
import (
"bytes"
"fmt"
"io"
"strconv"
"github.com/dchukmasov/graphql-go-tools/internal/pkg/unsafebytes"
"github.com/dchukmasov/graphql-go-tools/pkg/lexer/literal"
)
type JSONValueType int
const (
UnknownValueType JSONValueType = iota
StringValueType
IntegerValueType
FloatValueType
BooleanValueType
)
type ErrJSONValueTypeValueIncompatible struct {
value []byte
valueType JSONValueType
}
func (e ErrJSONValueTypeValueIncompatible) Error() string {
return fmt.Sprintf("JSONValueType.writeValue: cannot write %s as %s", unsafebytes.BytesToString(e.value), e.valueType)
}
func (i JSONValueType) writeValue(value, escapeBuf []byte, out io.Writer) (n int, err error) {
if len(value) == 0 || bytes.Equal(value, literal.NULL) {
return i.write(n, err, out, literal.NULL)
}
switch i {
case StringValueType:
return i.write(n, err, out, []byte(strconv.Quote(string(value))))
case IntegerValueType:
if !unsafebytes.BytesIsValidInt64(value) {
return n, ErrJSONValueTypeValueIncompatible{
value: value,
valueType: i,
}
}
return i.write(n, err, out, value)
case FloatValueType:
if !unsafebytes.BytesIsValidFloat32(value) {
return n, ErrJSONValueTypeValueIncompatible{
value: value,
valueType: i,
}
}
return i.write(n, err, out, value)
case BooleanValueType:
if !unsafebytes.BytesIsValidBool(value) {
return n, ErrJSONValueTypeValueIncompatible{
value: value,
valueType: i,
}
}
if unsafebytes.BytesToBool(value) {
return i.write(n, err, out, literal.TRUE)
} else {
return i.write(n, err, out, literal.FALSE)
}
default:
return n, ErrJSONValueTypeValueIncompatible{
value: value,
valueType: i,
}
}
}
func (i JSONValueType) write(n int, err error, out io.Writer, value []byte) (int, error) {
if err != nil {
return n, err
}
written, err := out.Write(value)
return n + written, err
}