blob: 189aaf6be249ef8426e9b96e162ed3684d89de41 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
import time
fn do_rec_i64(ch chan i64) {
mut sum := i64(0)
for _ in 0 .. 300 {
sum += <-ch
}
assert sum == 300 * (300 - 1) / 2
}
fn do_send_int(ch chan int) {
for i in 0 .. 300 {
ch <- i
}
}
fn do_send_byte(ch chan byte) {
for i in 0 .. 300 {
ch <- byte(i)
}
}
fn do_send_i64(ch chan i64) {
for i in 0 .. 300 {
ch <- i
}
}
fn test_select() {
chi := chan int{}
chl := chan i64{cap: 1}
chb := chan byte{cap: 10}
recch := chan i64{cap: 0}
go do_rec_i64(recch)
go do_send_int(chi)
go do_send_byte(chb)
go do_send_i64(chl)
mut sum := i64(0)
mut rl := i64(0)
mut sl := i64(0)
for _ in 0 .. 1200 {
select {
ri := <-chi {
sum += ri
}
recch <- sl {
sl++
}
rl = <-chl {
sum += rl
}
rb := <-chb {
sum += rb
}
}
}
// Use Gauß' formula for the first 2 contributions
// the 3rd contribution is `byte` and must be seen modulo 256
expected_sum := 2 * (300 * (300 - 1) / 2) + 256 * (256 - 1) / 2 + 44 * (44 - 1) / 2
assert sum == expected_sum
time.sleep(20 * time.millisecond) // to give assert in coroutine enough time
}
|
