/
json.go
103 lines (85 loc) · 2.14 KB
/
json.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
94
95
96
97
98
99
100
101
102
103
package scalars
import (
"database/sql/driver"
"encoding/json"
"errors"
"fmt"
"github.com/deref/exo/internal/util/jsonutil"
)
// JSON scalar for values known to be map-like objects. Marshals to and from
// the database as an encoded JSON string, but marshals to and from GraphQL as
// plain JSON-values, avoiding double-encoding.
type JSONObject map[string]any
// JSON scalar with json.RawMessage style unmarshaling behavior.
type RawJSON []byte
var _ Scalar = &JSONObject{}
var _ interface {
// Is it worth implementing GraphQLScalar?
json.Marshaler
json.Unmarshaler
DatabaseScalar
} = &RawJSON{}
func (_ JSONObject) ImplementsGraphQLType(name string) bool {
return name == "JSONObject"
}
func (obj *JSONObject) UnmarshalGraphQL(input any) (err error) {
m, ok := input.(map[string]any)
if !ok {
return errors.New("expected JSON object")
}
*obj = m
return nil
}
func (obj *JSONObject) Scan(src any) error {
switch s := src.(type) {
case nil:
*obj = nil
return nil
case string:
return json.Unmarshal([]byte(s), (*map[string]any)(obj))
default:
return fmt.Errorf("expected string, got %T", src)
}
}
func (raw *RawJSON) Scan(src any) error {
switch s := src.(type) {
case nil:
*raw = nil
return nil
case string:
*raw = []byte(s)
return nil
default:
return fmt.Errorf("expected string, got %T", src)
}
}
func (obj JSONObject) Value() (driver.Value, error) {
return jsonutil.MarshalString((map[string]any)(obj))
}
func (raw RawJSON) Value() (driver.Value, error) {
if len(raw) == 0 {
return nil, nil
}
return string(raw), nil
}
func (obj *JSONObject) UnmarshalJSON(bs []byte) (err error) {
return json.Unmarshal(bs, (*map[string]any)(obj))
}
// Same as json.RawMessage.MarshalJSON.
func (raw RawJSON) MarshalJSON() ([]byte, error) {
if raw == nil {
return []byte("null"), nil
}
return raw, nil
}
func (obj JSONObject) MarshalJSON() ([]byte, error) {
return json.Marshal((map[string]any)(obj))
}
// Same as json.RawMessage.UnmarshalJSON.
func (raw *RawJSON) UnmarshalJSON(data []byte) error {
if raw == nil {
return errors.New("RawJSON: UnmarshalJSON on nil pointer")
}
*raw = append((*raw)[0:0], data...)
return nil
}