74 lines
1.3 KiB
V
74 lines
1.3 KiB
V
import sync
|
|
|
|
fn do_rec(ch chan int, resch chan i64) {
|
|
mut sum := i64(0)
|
|
for {
|
|
a := <-ch or {
|
|
break
|
|
}
|
|
sum += a
|
|
}
|
|
assert ch.closed == true
|
|
println(sum)
|
|
resch <- sum
|
|
}
|
|
|
|
fn do_send(ch chan int) {
|
|
for i in 0 .. 8000 {
|
|
ch <- i
|
|
}
|
|
assert ch.closed == false
|
|
ch.close()
|
|
assert ch.closed == true
|
|
}
|
|
|
|
fn test_channel_close_buffered_multi() {
|
|
ch := chan int{cap: 10}
|
|
resch := chan i64{}
|
|
go do_rec(ch, resch)
|
|
go do_rec(ch, resch)
|
|
go do_rec(ch, resch)
|
|
go do_rec(ch, resch)
|
|
go do_send(ch)
|
|
mut sum := i64(0)
|
|
for _ in 0 .. 4 {
|
|
sum += <- resch
|
|
}
|
|
assert sum == i64(8000) * (8000 - 1) / 2
|
|
}
|
|
|
|
fn test_channel_close_unbuffered_multi() {
|
|
ch := chan int{}
|
|
resch := chan i64{}
|
|
go do_rec(ch, resch)
|
|
go do_rec(ch, resch)
|
|
go do_rec(ch, resch)
|
|
go do_rec(ch, resch)
|
|
go do_send(ch)
|
|
mut sum := i64(0)
|
|
for _ in 0 .. 4 {
|
|
sum += <-resch
|
|
}
|
|
assert sum == i64(8000) * (8000 - 1) / 2
|
|
}
|
|
|
|
fn test_channel_close_buffered() {
|
|
ch := chan int{cap: 100}
|
|
resch := chan i64{}
|
|
go do_rec(ch, resch)
|
|
go do_send(ch)
|
|
mut sum := i64(0)
|
|
sum += <-resch
|
|
assert sum == i64(8000) * (8000 - 1) / 2
|
|
}
|
|
|
|
fn test_channel_close_unbuffered() {
|
|
ch := chan int{}
|
|
resch := chan i64{cap: 100}
|
|
go do_rec(ch, resch)
|
|
go do_send(ch)
|
|
mut sum := i64(0)
|
|
sum += <-resch
|
|
assert sum == i64(8000) * (8000 - 1) / 2
|
|
}
|