.go

package test_test

import (
	"context"
	"fmt"
	"math/rand"
	"testing"
)

func BenchmarkMap(b *testing.B) {
	for _, size := range []int{1, 5, 10, 100, 1000} {
		m := make(map[int]int)
		for i := 0; i < size; i++ {
			m[i] = rand.Int()
		}
		b.Run(fmt.Sprintf("%d", size), func(b *testing.B) {
			for i := 0; i < b.N; i++ {
				_ = m[rand.Intn(size)]
			}
		})
	}
}

func BenchmarkContextValue(b *testing.B) {
	for _, size := range []int{1, 5, 10, 100, 1000} {
		ctx := context.Background()
		for i := 0; i < size; i++ {
			ctx = context.WithValue(ctx, i, rand.Int())
		}
		b.Run(fmt.Sprintf("%d", size), func(b *testing.B) {
			for i := 0; i < b.N; i++ {
				_ = ctx.Value(rand.Intn(size))
			}
		})
	}
}