slice_map_search_cmp_test.go

package main

import "testing"

var allowList = []string{
	"create",
	"cancel",
	"update",
}

var allowMap = map[string]struct{}{
	"create": {},
	"cancel": {},
	"update": {},
}

type Searcher interface {
	Search(method string) bool
}

type SliceSearcher struct{}

func (ss *SliceSearcher) Search(method string) bool {
	for _, m := range allowList {
		if m == method {
			return true
		}
	}
	return false
}

type MapSearcher struct{}

func (ms *MapSearcher) Search(method string) bool {
	if _, ok := allowMap[method]; ok {
		return true
	}
	return false
}

func DoSearch(s Searcher) (count int) {
	if s.Search("create") {
		count++
	}
	if s.Search("cancel") {
		count++
	}
	if s.Search("update") {
		count++
	}
	for i := 0; i < 20; i++ {
		if s.Search("heartbeat") {
			count++
		}
	}
	return
}

func BenchmarkSliceSearch(b *testing.B) {
	count := 0
	s := &SliceSearcher{}
	for n := 0; n < b.N; n++ {
		count += DoSearch(s)
	}
}

func BenchmarkMapSearch(b *testing.B) {
	count := 0
	s := &MapSearcher{}
	for n := 0; n < b.N; n++ {
		count += DoSearch(s)
	}
}