1
0
mirror of https://github.com/DataDog/go-profiler-notes.git synced 2025-07-12 23:50:13 +02:00
Files
go-profiler-notes/bench/workload_chan.go
Felix Geisendörfer 1332b004e0 split workloads
2021-02-04 14:30:29 +01:00

34 lines
578 B
Go

package main
import (
"fmt"
"sync"
)
func chanWorkload(goroutines, ops, depth, bufsize 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{}, bufsize)
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
}