mutexes.go

package main

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

// In the previous example we saw how to manage simple counter state using atomic operations.
// For more complex state we can use a mutex to safely access data across multiple goroutines.

// Mutex func to illustrate mutexes in go.
func Mutex() {
	// For our example the state will be a map.
	var state = make(map[int]int)

	// This mutex will synchronize access to state.
	var mutex = &sync.Mutex{}

	// We’ll keep track of how many read and write operations we do.
	var readOps uint64
	var writeOps uint64

	// Here we start 100 goroutines to execute repeated reads against the state,
	// once per millisecond in each goroutine.
	for r := 0; r < 100; r++ {
		go func() {
			total := 0
			for {
				// For each read we pick a key to access,
				// Lock() the mutex to ensure exclusive access to the state,
				// read the value at the chosen key,
				// Unlock() the mutex,
				// and increment the readOps count.
				key := rand.Intn(5)
				mutex.Lock()
				total += state[key]
				mutex.Unlock()
				atomic.AddUint64(&readOps, 1)

				// Wait a bit between reads.
				time.Sleep(time.Millisecond)
			}
		}()
	}

	// We’ll also start 10 goroutines to simulate writes,
	// using the same pattern we did for reads.

	for w := 0; w < 10; w++ {
		go func() {
			for {
				key := rand.Intn(5)
				val := rand.Intn(100)
				mutex.Lock()
				state[key] = val
				mutex.Unlock()
				atomic.AddUint64(&writeOps, 1)

				// Wait a bit between writes.
				time.Sleep(time.Millisecond)

			}
		}()
	}
	// Let the 10 goroutines work on the state and mutex for a second.
	time.Sleep(time.Second)

	// Take and report final operation counts.
	readOpsFinal := atomic.LoadUint64(&readOps)
	fmt.Println("readOps:", readOpsFinal)
	writeOpsFinal := atomic.LoadUint64(&writeOps)
	fmt.Println("WriteOps:", writeOpsFinal)

	// With a final lock of state, show how it ended up.
	mutex.Lock()
	fmt.Println("state:", state)
	mutex.Unlock()

	// Running the program shows that we executed about 90,000 total operations against our mutex-synchronized state.

}