about summary refs log tree commit diff
path: root/go/mutex.go
blob: 5cea20754bed51521a54cb910c1af0359df0e5bc (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
package main

import (
	"fmt"
	"math/rand"
	"sync"
	"sync/atomic"
	"time"
)

func main() {
	state := make(map[int]int)
	mux := &sync.Mutex{}

	var readOps uint64
	var writeOps uint64

	// Read from state
	for i := 0; i < 1000; i += 1 {
		for j := 0; j < 100; j += 1 {
			go func() {
				key := rand.Intn(5)
				mux.Lock()
				fmt.Printf("state[%d] = %d\n", key, state[key])
				mux.Unlock()
				atomic.AddUint64(&readOps, 1)
				time.Sleep(time.Millisecond)
			}()
		}
	}

	// Write to state
	for i := 0; i < 10; i += 1 {
		for j := 0; j < 100; j += 1 {
			go func() {
				key := rand.Intn(5)
				mux.Lock()
				state[key] += 1
				mux.Unlock()
				fmt.Printf("Wrote to state[%d].\n", key)
				atomic.AddUint64(&writeOps, 1)
				time.Sleep(time.Millisecond)
			}()
		}
	}

	time.Sleep(time.Millisecond)

	mux.Lock()
	fmt.Printf("State: %v\n", state)
	mux.Unlock()
	fmt.Printf("Reads: %d\tWrites: %d\n", atomic.LoadUint64(&readOps), atomic.LoadUint64(&writeOps))
}