bench_test.go

package gqlscan_test

import (
	"testing"

	"github.com/dustin/go-humanize"
	"github.com/graph-guard/gqlscan"
)

func BenchmarkScan(b *testing.B) {
	for _, td := range testdata {
		b.Run(td.decl, func(b *testing.B) {
			in := []byte(td.input)
			b.ResetTimer()
			for i := 0; i < b.N; i++ {
				if err := gqlscan.Scan(in, func(*gqlscan.Iterator) (err bool) {
					return false
				}); err.IsErr() {
					panic(err)
				}
			}
		})
	}
}

func BenchmarkScanAll(b *testing.B) {
	for _, td := range testdata {
		b.Run(td.decl, func(b *testing.B) {
			in := []byte(td.input)
			b.ResetTimer()
			for i := 0; i < b.N; i++ {
				if err := gqlscan.ScanAll(in, func(*gqlscan.Iterator) {
				}); err.IsErr() {
					panic(err)
				}
			}
		})
	}
}

func BenchmarkScanErr(b *testing.B) {
	for _, td := range testdataErr {
		b.Run(td.decl, func(b *testing.B) {
			in := []byte(td.input)
			b.ResetTimer()
			for i := 0; i < b.N; i++ {
				if err := gqlscan.Scan(in, func(*gqlscan.Iterator) (err bool) {
					return false
				}); !err.IsErr() {
					panic("unexpected success")
				}
			}
		})
	}
}

func BenchmarkScanAllErr(b *testing.B) {
	for _, td := range testdataErr {
		b.Run(td.decl, func(b *testing.B) {
			in := []byte(td.input)
			b.ResetTimer()
			for i := 0; i < b.N; i++ {
				if err := gqlscan.ScanAll(in, func(*gqlscan.Iterator) {
				}); !err.IsErr() {
					panic("unexpected success")
				}
			}
		})
	}
}

var InterpretedBuffer []byte

func BenchmarkScanInterpreted(b *testing.B) {
	for _, td := range testdataBlockStrings {
		b.Run(td.Decl, func(b *testing.B) {
			for _, bufSize := range []int{
				1, 1024, 65536,
			} {
				b.Run(humanize.Bytes(uint64(bufSize)), func(b *testing.B) {
					input := []byte(td.Input)
					buffer := make([]byte, bufSize)
					b.ResetTimer()
					for n := 0; n < b.N; n++ {
						c := 0
						if err := gqlscan.ScanAll(
							input,
							func(i *gqlscan.Iterator) {
								c++
								if c != td.TokenIndex {
									return
								}
								i.ScanInterpreted(
									buffer,
									func(buffer []byte) (stop bool) {
										InterpretedBuffer = buffer
										return false
									},
								)
							},
						); err.IsErr() {
							panic(err)
						}
					}

				})
			}
		})
	}
}