go/json-iterator
1
0
Fork 0
mirror of https://github.com/json-iterator/go.git synced 2025-01-23 18:54:21 +02:00
Code Issues Releases Activity

#36 handle anonymous

Browse Source
This commit is contained in:
Tao Wen 2017-06-02 16:52:20 +08:00
parent 5cb0d35610
commit 59e71bacc8
3 changed files with 38 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
feature_reflect_object.go
View File

@ -11,12 +11,12 @@ import (
func encoderOfStruct(typ reflect.Type) (Encoder, error) {
structEncoder_ := &structEncoder{}
for i := 0; i < typ.NumField(); i++ {
field := typ.Field(i)
fields := map[string]*structFieldEncoder{}
for _, field := range listStructFields(typ) {
fieldEncoderKey := fmt.Sprintf("%s/%s", typ.String(), field.Name)
var extensionProvidedFieldNames []string
for _, extension := range extensions {
alternativeFieldNames, fun, _ := extension(typ, &field)
alternativeFieldNames, fun, _ := extension(typ, field)
if alternativeFieldNames != nil {
extensionProvidedFieldNames = alternativeFieldNames
}
@ -46,16 +46,31 @@ func encoderOfStruct(typ reflect.Type) (Encoder, error) {
}
}
for _, fieldName := range fieldNames {
structEncoder_.fields = append(structEncoder_.fields,
&structFieldEncoder{&field, fieldName, encoder, omitempty})
}
fields[fieldName] = &structFieldEncoder{field, fieldName, encoder, omitempty}
}
}
if len(structEncoder_.fields) == 0 {
if len(fields) == 0 {
return &emptyStructEncoder{}, nil
}
for _, field := range fields {
structEncoder_.fields = append(structEncoder_.fields, field)
}
return structEncoder_, nil
}
func listStructFields(typ reflect.Type) []*reflect.StructField {
fields := []*reflect.StructField{}
for i := 0; i < typ.NumField(); i++ {
field := typ.Field(i)
if field.Anonymous {
fields = append(fields, listStructFields(field.Type)...)
} else {
fields = append(fields, &field)
}
}
return fields
}
func decoderOfStruct(typ reflect.Type) (Decoder, error) {
fields := map[string]*structFieldDecoder{}
for i := 0; i < typ.NumField(); i++ {

2
jsoniter_demo_test.go
View File

@ -21,4 +21,4 @@ func Test_iterator_api_demo(t *testing.T) {
total += iter.ReadInt()
}
fmt.Println(total)
}
}

15
jsoniter_reflect_struct_test.go
View File

@ -223,3 +223,18 @@ func Test_one_field_struct(t *testing.T) {
should.Nil(err)
should.Equal(`{"Me":{"Field":{"Field":{"Field":"abc"}}}}`, str)
}
func Test_anonymous_struct_marshal(t *testing.T) {
should := require.New(t)
type TestObject struct {
Field string
}
str, err := MarshalToString(struct{
TestObject
Field int
}{
Field: 100,
})
should.Nil(err)
should.Equal(`{"Field":100}`, str)
}