benchmark_test.go

package xmltokenizer_test

import (
	"bytes"
	"encoding/xml"
	"fmt"
	"io"
	"io/fs"
	"os"
	"path/filepath"
	"strings"
	"testing"

	"github.com/muktihari/xmltokenizer"
	"github.com/muktihari/xmltokenizer/internal/gpx"
	"github.com/muktihari/xmltokenizer/internal/xlsx"
)

func BenchmarkToken(b *testing.B) {
	filepath.Walk("testdata", func(path string, info fs.FileInfo, _ error) error {
		if info.IsDir() {
			return nil
		}
		name := strings.TrimPrefix(path, "testdata/")
		data, err := os.ReadFile(path)
		if err != nil {
			b.Logf("%v: %v", path, err)
			return nil
		}

		b.Run(fmt.Sprintf("stdlib.xml:%q", name), func(b *testing.B) {
			var err error
			for i := 0; i < b.N; i++ {
				if err = unmarshalWithStdlibXML(bytes.NewReader(data)); err != nil {
					b.Skipf("could not unmarshal: %v", err)
				}
			}
		})
		b.Run(fmt.Sprintf("xmltokenizer:%q", name), func(b *testing.B) {
			var err error
			for i := 0; i < b.N; i++ {
				if err = unmarshalWithXMLTokenizer(bytes.NewReader(data)); err != nil {
					b.Skipf("could not unmarshal: %v", err)
				}
			}
		})
		return nil
	})
}

func unmarshalWithXMLTokenizer(r io.Reader) error {
	tok := xmltokenizer.New(r)
	for {
		token, err := tok.Token()
		if err == io.EOF {
			break
		}
		if err != nil {
			return err
		}
		_ = token
	}
	return nil
}

func unmarshalWithStdlibXML(r io.Reader) error {
	dec := xml.NewDecoder(r)
	for {
		token, err := dec.Token()
		if err == io.EOF {
			break
		}
		if err != nil {
			return err
		}
		_ = token
	}
	return nil
}

func BenchmarkUnmarshalGPX(b *testing.B) {
	filepath.Walk("testdata", func(path string, info fs.FileInfo, _ error) error {
		if info.IsDir() {
			return nil
		}
		if strings.ToLower(filepath.Ext(path)) != ".gpx" {
			return nil
		}

		name := strings.TrimPrefix(path, "testdata/")

		data, err := os.ReadFile(path)
		if err != nil {
			panic(err)
		}

		b.Run(fmt.Sprintf("stdlib.xml:%q", name), func(b *testing.B) {
			for i := 0; i < b.N; i++ {
				_, _ = gpx.UnmarshalWithStdlibXML(bytes.NewReader(data))
			}
		})
		b.Run(fmt.Sprintf("xmltokenizer:%q", name), func(b *testing.B) {
			for i := 0; i < b.N; i++ {
				_, _ = gpx.UnmarshalWithXMLTokenizer(bytes.NewReader(data))
			}
		})

		return nil
	})
}

func BenchmarkUnmarshalXLSX(b *testing.B) {
	path := filepath.Join("testdata", "xlsx_sheet1.xml")
	name := strings.TrimPrefix(path, "testdata/")

	data, err := os.ReadFile(path)
	if err != nil {
		panic(err)
	}

	b.Run(fmt.Sprintf("stdlib.xml:%q", name), func(b *testing.B) {
		for i := 0; i < b.N; i++ {
			_, _ = xlsx.UnmarshalWithStdlibXML(bytes.NewReader(data))
		}
	})
	b.Run(fmt.Sprintf("xmltokenizer:%q", name), func(b *testing.B) {
		for i := 0; i < b.N; i++ {
			_, _ = xlsx.UnmarshalWithXMLTokenizer(bytes.NewReader(data))
		}
	})
}