summaryrefslogtreecommitdiffstats
path: root/vendor/github.com/vmihailenco/msgpack/v5/encode_slice.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/github.com/vmihailenco/msgpack/v5/encode_slice.go')
-rw-r--r--vendor/github.com/vmihailenco/msgpack/v5/encode_slice.go139
1 files changed, 139 insertions, 0 deletions
diff --git a/vendor/github.com/vmihailenco/msgpack/v5/encode_slice.go b/vendor/github.com/vmihailenco/msgpack/v5/encode_slice.go
new file mode 100644
index 00000000..ca46eada
--- /dev/null
+++ b/vendor/github.com/vmihailenco/msgpack/v5/encode_slice.go
@@ -0,0 +1,139 @@
+package msgpack
+
+import (
+ "math"
+ "reflect"
+
+ "github.com/vmihailenco/msgpack/v5/msgpcode"
+)
+
+var stringSliceType = reflect.TypeOf(([]string)(nil))
+
+func encodeStringValue(e *Encoder, v reflect.Value) error {
+ return e.EncodeString(v.String())
+}
+
+func encodeByteSliceValue(e *Encoder, v reflect.Value) error {
+ return e.EncodeBytes(v.Bytes())
+}
+
+func encodeByteArrayValue(e *Encoder, v reflect.Value) error {
+ if err := e.EncodeBytesLen(v.Len()); err != nil {
+ return err
+ }
+
+ if v.CanAddr() {
+ b := v.Slice(0, v.Len()).Bytes()
+ return e.write(b)
+ }
+
+ e.buf = grow(e.buf, v.Len())
+ reflect.Copy(reflect.ValueOf(e.buf), v)
+ return e.write(e.buf)
+}
+
+func grow(b []byte, n int) []byte {
+ if cap(b) >= n {
+ return b[:n]
+ }
+ b = b[:cap(b)]
+ b = append(b, make([]byte, n-len(b))...)
+ return b
+}
+
+func (e *Encoder) EncodeBytesLen(l int) error {
+ if l < 256 {
+ return e.write1(msgpcode.Bin8, uint8(l))
+ }
+ if l <= math.MaxUint16 {
+ return e.write2(msgpcode.Bin16, uint16(l))
+ }
+ return e.write4(msgpcode.Bin32, uint32(l))
+}
+
+func (e *Encoder) encodeStringLen(l int) error {
+ if l < 32 {
+ return e.writeCode(msgpcode.FixedStrLow | byte(l))
+ }
+ if l < 256 {
+ return e.write1(msgpcode.Str8, uint8(l))
+ }
+ if l <= math.MaxUint16 {
+ return e.write2(msgpcode.Str16, uint16(l))
+ }
+ return e.write4(msgpcode.Str32, uint32(l))
+}
+
+func (e *Encoder) EncodeString(v string) error {
+ if intern := e.flags&useInternedStringsFlag != 0; intern || len(e.dict) > 0 {
+ return e.encodeInternedString(v, intern)
+ }
+ return e.encodeNormalString(v)
+}
+
+func (e *Encoder) encodeNormalString(v string) error {
+ if err := e.encodeStringLen(len(v)); err != nil {
+ return err
+ }
+ return e.writeString(v)
+}
+
+func (e *Encoder) EncodeBytes(v []byte) error {
+ if v == nil {
+ return e.EncodeNil()
+ }
+ if err := e.EncodeBytesLen(len(v)); err != nil {
+ return err
+ }
+ return e.write(v)
+}
+
+func (e *Encoder) EncodeArrayLen(l int) error {
+ if l < 16 {
+ return e.writeCode(msgpcode.FixedArrayLow | byte(l))
+ }
+ if l <= math.MaxUint16 {
+ return e.write2(msgpcode.Array16, uint16(l))
+ }
+ return e.write4(msgpcode.Array32, uint32(l))
+}
+
+func encodeStringSliceValue(e *Encoder, v reflect.Value) error {
+ ss := v.Convert(stringSliceType).Interface().([]string)
+ return e.encodeStringSlice(ss)
+}
+
+func (e *Encoder) encodeStringSlice(s []string) error {
+ if s == nil {
+ return e.EncodeNil()
+ }
+ if err := e.EncodeArrayLen(len(s)); err != nil {
+ return err
+ }
+ for _, v := range s {
+ if err := e.EncodeString(v); err != nil {
+ return err
+ }
+ }
+ return nil
+}
+
+func encodeSliceValue(e *Encoder, v reflect.Value) error {
+ if v.IsNil() {
+ return e.EncodeNil()
+ }
+ return encodeArrayValue(e, v)
+}
+
+func encodeArrayValue(e *Encoder, v reflect.Value) error {
+ l := v.Len()
+ if err := e.EncodeArrayLen(l); err != nil {
+ return err
+ }
+ for i := 0; i < l; i++ {
+ if err := e.EncodeValue(v.Index(i)); err != nil {
+ return err
+ }
+ }
+ return nil
+}