switch_vs_map_bench_test.go

package main_test

import (
	"runtime"
	"testing"
)

//go:noinline
func switchImplementation() int {
	switch runtime.GOARCH {
	case "amd64":
		return 0
	case "arm64":
		return 1
	case "arm":
		return 2
	default:
		return 0
	}
}

func BenchmarkSwitchImpl(b *testing.B) {
	for i := 0; i < b.N; i++ {
		switchImplementation()
	}
}

var m = map[string]int{
	"amd64": 0,
	"arm":   1,
	"arm64": 2,
}

func mapImplementation() int {
	return m[runtime.GOARCH]
}

func BenchmarkMapImpl(b *testing.B) {
	for i := 0; i < b.N; i++ {
		mapImplementation()
	}
}