blob: cba8abfc962150e38e29d5ad4d150c89bbc8f61f (
plain) (
blame)
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
package main
import (
"fmt"
"math/rand"
"sync"
"sync/atomic"
)
type readMsg struct {
key int
sender chan int
}
type writeMsg struct {
key int
value int
sender chan bool
}
func main() {
fmt.Println("Hello, go.")
var readOps uint64
var writeOps uint64
var wg sync.WaitGroup
reads := make(chan readMsg)
writes := make(chan writeMsg)
go func() {
state := make(map[int]int)
for {
select {
case msg := <-reads:
msg.sender <- state[msg.key]
case msg := <-writes:
state[msg.key] = msg.value
msg.sender <- true
}
}
}()
// Reads
for i := 0; i < 100; i += 1 {
go func() {
wg.Add(1)
defer wg.Done()
for j := 0; j < 100; j += 1 {
msg := readMsg{
key: rand.Intn(5),
sender: make(chan int)}
reads <- msg
val := <-msg.sender
fmt.Printf("Received %d.\n", val)
atomic.AddUint64(&readOps, 1)
}
}()
}
// Writes
for i := 0; i < 100; i += 1 {
go func() {
wg.Add(1)
defer wg.Done()
for j := 0; j < 100; j += 1 {
msg := writeMsg{
key: rand.Intn(5),
value: rand.Intn(10),
sender: make(chan bool)}
writes <- msg
<-msg.sender
fmt.Printf("Set %d as %d in state\n", msg.key, msg.value)
atomic.AddUint64(&writeOps, 1)
}
}()
}
wg.Wait()
fmt.Printf("Read ops: %d\tWrite ops: %d\n", atomic.LoadUint64(&readOps), atomic.LoadUint64(&writeOps))
}
|
