item.go

package mc

import (
	"encoding"
	"encoding/json"

	"github.com/bytedance/sonic"
	"github.com/tinylib/msgp/msgp"
	"google.golang.org/protobuf/proto"
)

type (
	Item struct {
		Key        string
		Value      Value
		Flags      uint16
		cr         float32 // compression ratio
		cas        uint64
		hit        bool
		won        bool
		stale      bool
		opaque     int
		lastAccess int64
	}
	Value []byte
)

func (i *Item) Won() bool                 { return i.won }
func (i *Item) Hit() bool                 { return i.hit }
func (i *Item) Stale() bool               { return i.stale }
func (i *Item) LastAccess() int           { return int(i.lastAccess) }
func (i *Item) CompressionRatio() float32 { return i.cr }

func (val *Value) Marshal(v any) error {
	r, err := marshal(v)
	if err != nil {
		return err
	}
	*val = r
	return nil
}

func (val Value) Unmarshal(v any) error {
	return unmarshal(val, v)
}

func marshal(v any) ([]byte, error) {
	switch v := v.(type) {
	case proto.Message:
		return proto.Marshal(v)
	case json.Marshaler:
		return v.MarshalJSON()
	case msgp.Marshaler:
		return v.MarshalMsg(nil)
	case encoding.BinaryMarshaler:
		return v.MarshalBinary()
	}
	return sonic.Marshal(v)
}

func unmarshal(data []byte, v any) error {
	switch v := v.(type) {
	case proto.Message:
		return proto.Unmarshal(data, v)
	case json.Unmarshaler:
		return v.UnmarshalJSON(data)
	case msgp.Unmarshaler:
		if _, err := v.UnmarshalMsg(data); err != nil {
			return err
		}
		return nil
	case encoding.BinaryUnmarshaler:
		return v.UnmarshalBinary(data)
	}
	return sonic.Unmarshal(data, v)
}