map_value.go

package orderedmap

import (
	"encoding/json"
	"fmt"
	"reflect"
)

// Entry represents a map value entry.
type Entry struct {
	index uint64
	Value any
}

// UnmarshalJSON implements the json.Unmarshaler interface.
func (s *Entry) UnmarshalJSON(b []byte) error {
	s.index = nextSequence()

	if err := json.Unmarshal(b, &s.Value); err != nil {
		return fmt.Errorf("unmarshalling entry: %w", err)
	}

	if s.Value != nil {
		valueKind := reflect.TypeOf(s.Value).Kind()
		if valueKind == reflect.Map { // force values of type map to be of ordered map as well
			var m Map
			if err := json.Unmarshal(b, &m); err != nil {
				return fmt.Errorf("unmarshalling json entry into map: %w", err)
			}
			s.Value = m
		}
	}
	return nil
}

// MarshalJSON implements the json.Marshaler interface.
func (s *Entry) MarshalJSON() ([]byte, error) {
	b, err := json.Marshal(s.Value)
	if err != nil {
		return nil, fmt.Errorf("marshalling entry: %w", err)
	}
	return b, nil
}