atomic_vs_channel_benchmark_test.go

package main

import (
	"sync/atomic"
	"testing"
)

var id uint64

const chanLength = 100

var idChan = make(chan uint64, chanLength)

func nextIDAtomic() uint64 {
	return atomic.AddUint64(&id, 1)
}

func nextIDChan() uint64 {
	if len(idChan) == 0 {
		for i := 0; i < chanLength; i++ {
			idChan <- id
			id++
		}
	}

	return <-idChan
}

func BenchmarkNextIDAtomic(b *testing.B) {
	for n := 0; n < b.N; n++ {
		nextIDAtomic()
	}
}
func BenchmarkNextIDChan(b *testing.B) {
	for n := 0; n < b.N; n++ {
		nextIDChan()
	}
}