memoize_with_cache_err_bench_test.go

package memoize

import (
	"fmt"
	"math"
	"testing"

	"github.com/emad-elsaid/memoize/cache"
)

const maxKeySpace = 6

var returns int

func BenchmarkMemoizerWithCacheErr(b *testing.B) {
	for i := 1; i < maxKeySpace; i++ {
		keysN := math.Pow(10, float64(i))
		name := fmt.Sprintf("Keys:%.f", keysN)

		b.Run(name, func(b *testing.B) {

			mem := NewWithCacheErr(
				cache.New[string, Pair[int]](),
				func(k string) (int, error) { return len(k), nil })

			keys := []string{}
			for i := range int64(keysN) {
				keys = append(keys, string(fmt.Sprintf("Key%d", i)))
			}

			b.ResetTimer()
			for i := range b.N {
				k := keys[i%len(keys)]
				r, _ := mem(k)
				returns = r
			}

		})
	}
}

func BenchmarkMemoizerWithCacheErrParallel(b *testing.B) {
	for i := 1; i < maxKeySpace; i++ {
		keysN := math.Pow(10, float64(i))
		name := fmt.Sprintf("Keys:%.f", keysN)

		b.Run(name, func(b *testing.B) {

			mem := NewWithCacheErr(
				cache.New[string, Pair[int]](),
				func(k string) (int, error) { return len(k), nil })

			keys := []string{}
			for i := range int64(keysN) {
				keys = append(keys, string(fmt.Sprintf("Key%d", i)))
			}

			b.ResetTimer()
			b.RunParallel(func(b *testing.PB) {
				i := 0
				for b.Next() {
					k := keys[i%len(keys)]
					r, _ := mem(k)
					returns = r
					i++
				}
			})

		})
	}
}