diff options
Diffstat (limited to 'users/wpcarro/go/channels.go')
-rw-r--r-- | users/wpcarro/go/channels.go | 81 |
1 files changed, 81 insertions, 0 deletions
diff --git a/users/wpcarro/go/channels.go b/users/wpcarro/go/channels.go new file mode 100644 index 000000000000..cba8abfc9621 --- /dev/null +++ b/users/wpcarro/go/channels.go @@ -0,0 +1,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)) +} |