54 lines
1.1 KiB
V
Executable File
54 lines
1.1 KiB
V
Executable File
// Channel Benchmark
|
|
//
|
|
// `nobj` integers are sent thru a channel with queue length`buflen`
|
|
// using `nsend` sender threads and `nrec` receiver threads.
|
|
//
|
|
// The receive threads add all received numbers and send them to the
|
|
// main thread where the total sum is compare to the expected value.
|
|
|
|
const (
|
|
nsend = 2
|
|
nrec = 2
|
|
buflen = 100
|
|
nobj = 10000
|
|
objs_per_thread = 5000
|
|
)
|
|
|
|
fn do_rec(ch chan int, resch chan i64, n int) {
|
|
mut sum := i64(0)
|
|
for _ in 0 .. n {
|
|
mut r := 0
|
|
for ch.try_pop(r) != .success {}
|
|
sum += r
|
|
}
|
|
println(sum)
|
|
resch <- sum
|
|
}
|
|
|
|
fn do_send(ch chan int, start, end int) {
|
|
for i in start .. end {
|
|
for ch.try_push(i) != .success {}
|
|
}
|
|
}
|
|
|
|
fn test_channel_polling() {
|
|
ch := chan int{cap: buflen}
|
|
resch := chan i64{}
|
|
for _ in 0 .. nrec {
|
|
go do_rec(ch, resch, objs_per_thread)
|
|
}
|
|
mut n := nobj
|
|
for _ in 0 .. nsend {
|
|
end := n
|
|
n -= objs_per_thread
|
|
go do_send(ch, n, end)
|
|
}
|
|
mut sum := i64(0)
|
|
for _ in 0 .. nrec {
|
|
sum += <-resch
|
|
}
|
|
// use sum formula by Gauß to calculate the expected result
|
|
expected_sum := i64(nobj)*(nobj-1)/2
|
|
assert sum == expected_sum
|
|
}
|