/
jsonw.go
93 lines (74 loc) · 1.76 KB
/
jsonw.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
85
86
87
88
89
90
91
92
93
package graphql
import (
"context"
"io"
)
var (
nullLit = []byte(`null`)
trueLit = []byte(`true`)
falseLit = []byte(`false`)
openBrace = []byte(`{`)
closeBrace = []byte(`}`)
openBracket = []byte(`[`)
closeBracket = []byte(`]`)
colon = []byte(`:`)
comma = []byte(`,`)
)
var (
Null = &lit{nullLit}
True = &lit{trueLit}
False = &lit{falseLit}
)
type Marshaler interface {
MarshalGQL(w io.Writer)
}
type Unmarshaler interface {
UnmarshalGQL(v interface{}) error
}
type ContextMarshaler interface {
MarshalGQLContext(ctx context.Context, w io.Writer) error
}
type ContextUnmarshaler interface {
UnmarshalGQLContext(ctx context.Context, v interface{}) error
}
type contextMarshalerAdapter struct {
Context context.Context
ContextMarshaler
}
func WrapContextMarshaler(ctx context.Context, m ContextMarshaler) Marshaler {
return contextMarshalerAdapter{Context: ctx, ContextMarshaler: m}
}
func (a contextMarshalerAdapter) MarshalGQL(w io.Writer) {
err := a.MarshalGQLContext(a.Context, w)
if err != nil {
AddError(a.Context, err)
Null.MarshalGQL(w)
}
}
type WriterFunc func(writer io.Writer)
func (f WriterFunc) MarshalGQL(w io.Writer) {
f(w)
}
type ContextWriterFunc func(ctx context.Context, writer io.Writer) error
func (f ContextWriterFunc) MarshalGQLContext(ctx context.Context, w io.Writer) error {
return f(ctx, w)
}
type Array []Marshaler
func (a Array) MarshalGQL(writer io.Writer) {
writer.Write(openBracket)
for i, val := range a {
if i != 0 {
writer.Write(comma)
}
val.MarshalGQL(writer)
}
writer.Write(closeBracket)
}
type lit struct{ b []byte }
func (l lit) MarshalGQL(w io.Writer) {
w.Write(l.b)
}
func (l lit) MarshalGQLContext(ctx context.Context, w io.Writer) error {
w.Write(l.b)
return nil
}