json.go

package ipam

import (
	"encoding/json"
	"fmt"
)

type prefixJSON struct {
	Prefix
	Namespace              string          `json:"Namespace"`
	AvailableChildPrefixes map[string]bool `json:"AvailableChildPrefixes"` // available child prefixes of this prefix
	// TODO remove this in the next release
	ChildPrefixLength int             `json:"ChildPrefixLength"` // the length of the child prefixes. Legacy to migrate existing prefixes stored in the db to set the IsParent on reads.
	IsParent          bool            `json:"IsParent"`          // set to true if there are child prefixes
	IPs               map[string]bool `json:"IPs"`               // The ips contained in this prefix
	Version           int64           `json:"Version"`           // Version is used for optimistic locking
}

func (p prefixJSON) toPrefix() Prefix {
	// Legacy support only on reading from database, convert to isParent.
	// TODO remove this in the next release
	if p.ChildPrefixLength > 0 {
		p.IsParent = true
	}
	return Prefix{
		Cidr:                   p.Cidr,
		ParentCidr:             p.ParentCidr,
		availableChildPrefixes: p.AvailableChildPrefixes,
		childPrefixLength:      p.ChildPrefixLength,
		isParent:               p.IsParent,
		ips:                    p.IPs,
		version:                p.Version,
	}
}

func (p Prefix) toPrefixJSON() prefixJSON {
	return prefixJSON{
		Prefix: Prefix{
			Cidr:       p.Cidr,
			ParentCidr: p.ParentCidr,
		},
		AvailableChildPrefixes: p.availableChildPrefixes,
		IsParent:               p.isParent,
		// TODO remove this in the next release
		ChildPrefixLength: p.childPrefixLength,
		IPs:               p.ips,
		Version:           p.version,
	}
}

func (p Prefix) toJSON() ([]byte, error) {
	pj, err := json.Marshal(p.toPrefixJSON()) // nolint:musttag
	if err != nil {
		return nil, fmt.Errorf("unable to marshal prefix:%w", err)
	}
	return pj, nil
}

func (ps Prefixes) toJSON() ([]byte, error) {
	var pfxjs []prefixJSON
	for _, p := range ps {
		pfxjs = append(pfxjs, p.toPrefixJSON())
	}
	pj, err := json.Marshal(pfxjs)
	if err != nil {
		return nil, fmt.Errorf("unable to marshal prefixes:%w", err)
	}
	return pj, nil
}

func fromJSON(js []byte) (Prefix, error) {
	var pre prefixJSON
	err := json.Unmarshal(js, &pre) // nolint:musttag
	if err != nil {
		return Prefix{}, fmt.Errorf("unable to unmarshal prefix:%w", err)
	}
	return pre.toPrefix(), nil
}

func fromJSONs(js []byte) (Prefixes, error) {
	var pres []prefixJSON
	err := json.Unmarshal(js, &pres)
	if err != nil {
		return Prefixes{}, fmt.Errorf("unable to unmarshal prefixes:%w", err)
	}
	var pfxs Prefixes
	for _, pj := range pres {
		pfxs = append(pfxs, pj.toPrefix())
	}
	return pfxs, nil
}