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
|
// Copyright 2019-present Facebook Inc. All rights reserved.
// This source code is licensed under the Apache 2.0 license found
// in the LICENSE file in the root directory of this source tree.
package graphson
import (
"io"
"reflect"
jsoniter "github.com/json-iterator/go"
"github.com/modern-go/reflect2"
)
type encodeExtension struct {
jsoniter.DummyExtension
}
// Marshal returns the graphson encoding of v.
func Marshal(v interface{}) ([]byte, error) {
return config.Marshal(v)
}
// MarshalToString returns the graphson encoding of v as string.
func MarshalToString(v interface{}) (string, error) {
return config.MarshalToString(v)
}
// Encoder defines a graphson encoder.
type Encoder interface {
Encode(interface{}) error
}
// NewEncoder create a graphson encoder.
func NewEncoder(w io.Writer) Encoder {
return config.NewEncoder(w)
}
// Marshaler is the interface implemented by types that
// can marshal themselves as graphson.
type Marshaler interface {
MarshalGraphson() ([]byte, error)
}
// UpdateStructDescriptor decorates struct field encoders for graphson tags.
func (ext encodeExtension) UpdateStructDescriptor(desc *jsoniter.StructDescriptor) {
for _, binding := range desc.Fields {
if tag, ok := binding.Field.Tag().Lookup("graphson"); ok && tag != "-" {
if enc := ext.DecoratorOfStructField(binding.Encoder, tag); enc != nil {
binding.Encoder = enc
}
}
}
}
// CreateEncoder returns a value encoder for type.
func (ext encodeExtension) CreateEncoder(typ reflect2.Type) jsoniter.ValEncoder {
if enc := ext.EncoderOfRegistered(typ); enc != nil {
return enc
}
if enc := ext.EncoderOfNative(typ); enc != nil {
return enc
}
switch typ.Kind() {
case reflect.Map:
return ext.EncoderOfMap(typ)
default:
return nil
}
}
// DecorateEncoder decorates an passed in value encoder for type.
func (ext encodeExtension) DecorateEncoder(typ reflect2.Type, enc jsoniter.ValEncoder) jsoniter.ValEncoder {
if enc := ext.DecoratorOfRegistered(enc); enc != nil {
return enc
}
if enc := ext.DecoratorOfMarshaler(typ, enc); enc != nil {
return enc
}
if enc := ext.DecoratorOfTyper(typ, enc); enc != nil {
return enc
}
if enc := ext.DecoratorOfNative(typ, enc); enc != nil {
return enc
}
switch typ.Kind() {
case reflect.Ptr, reflect.Interface, reflect.Struct:
return enc
case reflect.Array:
return ext.DecoratorOfArray(enc)
case reflect.Slice:
return ext.DecoratorOfSlice(typ, enc)
case reflect.Map:
return ext.DecoratorOfMap(enc)
default:
return ext.EncoderOfError("graphson: unsupported type: " + typ.String())
}
}
|
