go 线程同步

代码来源于《Go语言开发教程》

// 线程同步: sync.Mutex

package main

import (
	"errors"
	"fmt"
	"sync"
)

type MyMap struct {
	mp    map[string]int
	mutex *sync.Mutex
}

func (this *MyMap) Get(key string) (int, error) {
	this.mutex.Lock()
	i, ok := this.mp[key]
	this.mutex.Unlock()

	if !ok {
		return i, errors.New("not exist")
	}

	return i, nil
}

func (this *MyMap) Set(key string, v int) {
	this.mutex.Lock()
	defer this.mutex.Unlock()
	this.mp[key] = v
}

func SetValue(m *MyMap) {
	var a rune
	a = 'a'
	for i := 0; i < 10; i++ {
		m.Set(string(a+rune(i)), i)
	}
}

func (this *MyMap) Display() {
	this.mutex.Lock()
	defer this.mutex.Unlock()

	for k, v := range this.mp {
		fmt.Println(k, "=", v)
	}
}

func main() {
	m := &MyMap{mp: make(map[string]int), mutex: new(sync.Mutex)}
	go SetValue(m)
	go m.Display()
	var str string
	fmt.Scan(&str)
}