serialize/xml.go

// Package serialize ...
//
// Description : serialize ...
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 2024-10-23 17:51
package serialize

import (
	"bytes"
	"encoding/xml"
	"io"
)

var (
	Xml = &ownXml{}
)

type ownXml struct{}

func (o *ownXml) UnmarshalWithNumber(byteData []byte, receiver any) error {
	return xml.NewDecoder(bytes.NewReader(byteData)).Decode(receiver)
}

func (o *ownXml) UnmarshalWithNumberIgnoreError(byteData []byte, receiver any) {
	_ = o.UnmarshalWithNumber(byteData, receiver)
	return
}

func (o *ownXml) UnmarshalWithNumberForIOReader(ioReader io.ReadCloser, receiver any) error {
	return xml.NewDecoder(ioReader).Decode(receiver)
}

func (o *ownXml) UnmarshalWithNumberForIOReaderIgnoreError(ioReader io.ReadCloser, receiver any) {
	_ = o.UnmarshalWithNumberForIOReader(ioReader, receiver)
	return
}

func (o *ownXml) UnmarshalWithNumberForString(input string, receiver any) error {
	return o.UnmarshalWithNumber([]byte(input), receiver)
}

func (o *ownXml) UnmarshalWithNumberForStringIgnoreError(input string, receiver any) {
	_ = o.UnmarshalWithNumberForString(input, receiver)
	return
}

func (o *ownXml) MarshalForByte(input any) ([]byte, error) {
	return xml.Marshal(input)
}

func (o *ownXml) MarshalForByteIgnoreError(input any) []byte {
	byteData, _ := o.MarshalForByte(input)
	return byteData
}

func (o *ownXml) MarshalForString(input any) (string, error) {
	byteData, err := o.MarshalForByte(input)
	if nil != err {
		return "", err
	}
	return string(byteData), nil
}

func (o *ownXml) MarshalForStringIgnoreError(input any) string {
	str, _ := o.MarshalForString(input)
	return str
}

type AnyMap map[string]any

type xmlMapEntry struct {
	XMLName xml.Name
	Value   any `xml:",chardata"`
}

func (m *AnyMap) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
	if nil == m {
		return nil
	}
	if len(*m) == 0 {
		return nil
	}

	err := e.EncodeToken(start)
	if err != nil {
		return err
	}

	for k, v := range *m {
		_ = e.Encode(xmlMapEntry{XMLName: xml.Name{Local: k}, Value: v})
	}

	return e.EncodeToken(start.End())
}

func (m *AnyMap) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
	*m = AnyMap{}
	for {
		var e xmlMapEntry

		err := d.Decode(&e)
		if err == io.EOF {
			break
		} else if err != nil {
			return err
		}

		(*m)[e.XMLName.Local] = e.Value
	}
	return nil
}