bench_test.go

package main

import (
	"sync"
	"testing"
	"time"
)

const concurrency = 4

func BenchmarkLimitByWorkers(b *testing.B) {
	ch := make(chan struct{}, 10000)
	wg := sync.WaitGroup{}

	wg.Add(concurrency)
	for i := 0; i < concurrency; i++ {
		go func() {
			for range ch {
				// 何かしら処理の代わり
				// ベンチマークそのものに支配的にならないようにマイクロ秒だけsleep
				time.Sleep(time.Microsecond)
			}
			wg.Done()
		}()
	}

	for i := 0; i < b.N; i++ {
		ch <- struct{}{}
	}
	close(ch)
	wg.Wait()
}

func BenchmarkLimitBySemaphore(b *testing.B) {
	ch := make(chan struct{}, concurrency)
	wg := sync.WaitGroup{}

	wg.Add(b.N)
	for i := 0; i < b.N; i++ {
		go func() {
			ch <- struct{}{}
			// 何かしら処理の代わり
			// ベンチマークそのものに支配的にならないようにマイクロ秒だけsleep
			time.Sleep(time.Microsecond)
			<-ch
			wg.Done()
		}()
	}
	wg.Wait()
}