Skip to content

Commit

Permalink
Merge pull request #371 from nikhita/byte-base64-encode
Browse files Browse the repository at this point in the history
Don't marshal empty byte or uint8 slice as null
  • Loading branch information
thockin authored Jun 3, 2019
2 parents 08047c1 + fb5614a commit 0039f4a
Show file tree
Hide file tree
Showing 2 changed files with 29 additions and 6 deletions.
21 changes: 21 additions & 0 deletions misc_tests/jsoniter_array_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -158,6 +158,27 @@ func Test_encode_byte_array(t *testing.T) {
should.Equal(`"AQID"`, string(bytes))
}

func Test_encode_empty_byte_array(t *testing.T) {
should := require.New(t)
bytes, err := json.Marshal([]byte{})
should.Nil(err)
should.Equal(`""`, string(bytes))
bytes, err = jsoniter.Marshal([]byte{})
should.Nil(err)
should.Equal(`""`, string(bytes))
}

func Test_encode_nil_byte_array(t *testing.T) {
should := require.New(t)
var nilSlice []byte
bytes, err := json.Marshal(nilSlice)
should.Nil(err)
should.Equal(`null`, string(bytes))
bytes, err = jsoniter.Marshal(nilSlice)
should.Nil(err)
should.Equal(`null`, string(bytes))
}

func Test_decode_byte_array_from_base64(t *testing.T) {
should := require.New(t)
data := []byte{}
Expand Down
14 changes: 8 additions & 6 deletions reflect_native.go
Original file line number Diff line number Diff line change
Expand Up @@ -432,17 +432,19 @@ func (codec *base64Codec) Decode(ptr unsafe.Pointer, iter *Iterator) {
}

func (codec *base64Codec) Encode(ptr unsafe.Pointer, stream *Stream) {
src := *((*[]byte)(ptr))
if len(src) == 0 {
if codec.sliceType.UnsafeIsNil(ptr) {
stream.WriteNil()
return
}
src := *((*[]byte)(ptr))
encoding := base64.StdEncoding
stream.writeByte('"')
size := encoding.EncodedLen(len(src))
buf := make([]byte, size)
encoding.Encode(buf, src)
stream.buf = append(stream.buf, buf...)
if len(src) != 0 {
size := encoding.EncodedLen(len(src))
buf := make([]byte, size)
encoding.Encode(buf, src)
stream.buf = append(stream.buf, buf...)
}
stream.writeByte('"')
}

Expand Down

0 comments on commit 0039f4a

Please sign in to comment.