jsoniter ignore omitempty

jsoniter_ext.go

import jsoniter "github.com/json-iterator/go"

type NotOmitemptyValEncoder struct {
	encoder jsoniter.ValEncoder
}

func (codec *NotOmitemptyValEncoder) Encode(ptr unsafe.Pointer, stream *jsoniter.Stream) {
	codec.encoder.Encode(ptr, stream)
}

func (codec *NotOmitemptyValEncoder) IsEmpty(ptr unsafe.Pointer) bool {
	return false
}

// NotOmitemptyExtension 忽略omitempty的扩展
// func init中注册 jsoniter.RegisterExtension(new(NotOmitemptyExtension)) 即可
type NotOmitemptyExtension struct {
	jsoniter.DummyExtension
}

func (ed NotOmitemptyExtension) DecorateEncoder(typ reflect2.Type, encoder jsoniter.ValEncoder) jsoniter.ValEncoder {
	return &NotOmitemptyValEncoder{encoder: encoder}
}

// func TestJsonMarshaler(t *testing.T) {
// 	type before struct {
// 		Name string `json:"name,omitempty"`
// 		Age  int    `json:"age,omitempty"`
// 	}
// 	type after struct {
// 		Name string `json:"name,omitempty"`
// 		Age  int    `json:"age,omitempty"`
// 	}
// 	data1, _ := jsoniter.Marshal(&before{})
// 	assert.Equal(t, `{}`, string(data1))

// 	jsoniter.RegisterExtension(new(NotOmitemptyExtension))
// 	data2, _ := jsoniter.Marshal(&after{})
// 	assert.Equal(t, `{"name":"","age":0}`, string(data2))
// }