terraformDummyRepo2/vendor/github.com/vmihailenco/msgpack/v5/encode_map.go

180 lines
3.4 KiB
Go
Raw Normal View History

2022-04-03 04:07:16 +00:00
package msgpack
import (
"math"
2022-04-03 04:07:16 +00:00
"reflect"
"sort"
"github.com/vmihailenco/msgpack/v5/msgpcode"
2022-04-03 04:07:16 +00:00
)
func encodeMapValue(e *Encoder, v reflect.Value) error {
if v.IsNil() {
return e.EncodeNil()
}
if err := e.EncodeMapLen(v.Len()); err != nil {
return err
}
iter := v.MapRange()
for iter.Next() {
if err := e.EncodeValue(iter.Key()); err != nil {
2022-04-03 04:07:16 +00:00
return err
}
if err := e.EncodeValue(iter.Value()); err != nil {
2022-04-03 04:07:16 +00:00
return err
}
}
return nil
}
func encodeMapStringStringValue(e *Encoder, v reflect.Value) error {
if v.IsNil() {
return e.EncodeNil()
}
if err := e.EncodeMapLen(v.Len()); err != nil {
return err
}
m := v.Convert(mapStringStringType).Interface().(map[string]string)
if e.flags&sortMapKeysFlag != 0 {
return e.encodeSortedMapStringString(m)
}
for mk, mv := range m {
if err := e.EncodeString(mk); err != nil {
return err
}
if err := e.EncodeString(mv); err != nil {
return err
}
}
return nil
}
func encodeMapStringInterfaceValue(e *Encoder, v reflect.Value) error {
if v.IsNil() {
return e.EncodeNil()
}
m := v.Convert(mapStringInterfaceType).Interface().(map[string]interface{})
if e.flags&sortMapKeysFlag != 0 {
return e.EncodeMapSorted(m)
2022-04-03 04:07:16 +00:00
}
return e.EncodeMap(m)
}
2022-04-03 04:07:16 +00:00
func (e *Encoder) EncodeMap(m map[string]interface{}) error {
if m == nil {
return e.EncodeNil()
}
if err := e.EncodeMapLen(len(m)); err != nil {
return err
}
2022-04-03 04:07:16 +00:00
for mk, mv := range m {
if err := e.EncodeString(mk); err != nil {
return err
}
if err := e.Encode(mv); err != nil {
return err
}
}
return nil
}
func (e *Encoder) EncodeMapSorted(m map[string]interface{}) error {
if m == nil {
return e.EncodeNil()
}
if err := e.EncodeMapLen(len(m)); err != nil {
return err
}
2022-04-03 04:07:16 +00:00
keys := make([]string, 0, len(m))
2022-04-03 04:07:16 +00:00
for k := range m {
keys = append(keys, k)
}
2022-04-03 04:07:16 +00:00
sort.Strings(keys)
for _, k := range keys {
if err := e.EncodeString(k); err != nil {
2022-04-03 04:07:16 +00:00
return err
}
if err := e.Encode(m[k]); err != nil {
2022-04-03 04:07:16 +00:00
return err
}
}
return nil
}
func (e *Encoder) encodeSortedMapStringString(m map[string]string) error {
2022-04-03 04:07:16 +00:00
keys := make([]string, 0, len(m))
for k := range m {
keys = append(keys, k)
}
sort.Strings(keys)
for _, k := range keys {
err := e.EncodeString(k)
if err != nil {
return err
}
if err = e.EncodeString(m[k]); err != nil {
2022-04-03 04:07:16 +00:00
return err
}
}
return nil
}
func (e *Encoder) EncodeMapLen(l int) error {
if l < 16 {
return e.writeCode(msgpcode.FixedMapLow | byte(l))
2022-04-03 04:07:16 +00:00
}
if l <= math.MaxUint16 {
return e.write2(msgpcode.Map16, uint16(l))
2022-04-03 04:07:16 +00:00
}
return e.write4(msgpcode.Map32, uint32(l))
2022-04-03 04:07:16 +00:00
}
func encodeStructValue(e *Encoder, strct reflect.Value) error {
structFields := structs.Fields(strct.Type(), e.structTag)
if e.flags&arrayEncodedStructsFlag != 0 || structFields.AsArray {
2022-04-03 04:07:16 +00:00
return encodeStructValueAsArray(e, strct, structFields.List)
}
fields := structFields.OmitEmpty(strct, e.flags&omitEmptyFlag != 0)
2022-04-03 04:07:16 +00:00
if err := e.EncodeMapLen(len(fields)); err != nil {
return err
}
for _, f := range fields {
if err := e.EncodeString(f.name); err != nil {
return err
}
if err := f.EncodeValue(e, strct); err != nil {
return err
}
}
return nil
}
func encodeStructValueAsArray(e *Encoder, strct reflect.Value, fields []*field) error {
if err := e.EncodeArrayLen(len(fields)); err != nil {
return err
}
for _, f := range fields {
if err := f.EncodeValue(e, strct); err != nil {
return err
}
}
return nil
}