blob: 97228fe7392e416c0fc64352c7471d67de1c3d83 (
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
|
fn do_rec_i64(ch chan i64, sumch chan i64) {
mut sum := i64(0)
for _ in 0 .. 10000 {
sum += <-ch
}
sumch <- sum
}
fn do_send_int(ch chan int) {
for i in 0 .. 10000 {
ch <- i
}
}
fn do_send_int2(ch chan int) {
for i in 10000 .. 20000 {
ch <- i
}
}
fn do_send_int3(ch chan int) {
for i in 20000 .. 30000 {
ch <- i
}
}
fn test_select() {
chi := chan int{cap: 10}
recch := chan i64{cap: 10}
chsum := chan i64{}
go do_rec_i64(recch, chsum)
go do_rec_i64(recch, chsum)
go do_rec_i64(recch, chsum)
go do_send_int(chi)
go do_send_int2(chi)
go do_send_int3(chi)
mut sum := i64(0)
mut sl := i64(0)
for _ in 0 .. 60000 + recch.cap {
select {
ri := <-chi {
sum += ri
}
recch <- sl {
sl++
}
}
}
// Use Gauß' formula
expected_sum := i64(30000) * (30000 - 1) / 2
assert sum == expected_sum
mut sumrec := <-chsum
sumrec += <-chsum
sumrec += <-chsum
// Empty receive buffer
for _ in 0 .. recch.cap {
sumrec += <-recch
}
assert sumrec == i64(30000 + recch.cap) * (30000 + recch.cap - 1) / 2
}
|
