go-profiler-notes/bench/workloads.go

package main

import (
	"fmt"
	"sync"
)

func chanWorkload(goroutines, ops, depth int) error {
	if goroutines%2 != 0 {
		return fmt.Errorf("bad goroutines: %d: must be a multiple of 2", goroutines)
	}

	wg := &sync.WaitGroup{}
	for j := 0; j < goroutines/2; j++ {
		ch := make(chan struct{})
		wg.Add(1)
		go atStackDepth(depth, func() {
			defer wg.Done()
			for i := 0; i < ops; i++ {
				ch <- struct{}{}
			}
		})
		wg.Add(1)
		go atStackDepth(depth, func() {
			defer wg.Done()
			for i := 0; i < ops; i++ {
				<-ch
			}
		})
	}
	wg.Wait()
	return nil
}

func mutexWorkload(goroutines, ops, depth int) error {
	if goroutines%2 != 0 {
		return fmt.Errorf("bad goroutines: %d: must be a multiple of 2", goroutines)
	}

	wg := &sync.WaitGroup{}
	for j := 0; j < goroutines/2; j++ {
		m := &sync.Mutex{}
		wg.Add(1)
		go atStackDepth(depth, func() {
			defer wg.Done()
			for i := 0; i < ops; i++ {
				m.Lock()
				m.Unlock()
			}
		})
		wg.Add(1)
		go atStackDepth(depth, func() {
			defer wg.Done()
			for i := 0; i < ops; i++ {
				m.Lock()
				m.Unlock()
			}
		})
	}
	wg.Wait()
	return nil
}