41.基于共享内存的携程通信

package main
import (
    "fmt"
    "runtime"
    "sync"
    "time"
)
var counter int = 0
var lock *sync.Mutex
func add(a, b int) {
    c := a + b
    lock.Lock()
    counter++
    lock.Unlock()
    fmt.Printf("%d: %d + %d = %d\n", counter, a, b, c)
}
func main() {
    startTime := time.Now()
    lock = &sync.Mutex{}
    for i := 0; i < 10; i++ {
        go add(1, i)
    }
    for {
        lock.Lock()
        c := counter
        lock.Unlock()
        runtime.Gosched()
        if c >= 10 {
            break
        }
    }
    endTime := time.Now()
    consume := endTime.Sub(startTime).Milliseconds()
    fmt.Println("程序执行耗时(s)：", consume)
}

代码分析

程序启动后使用go关键词启动了十个携程，携程执行方法add。在方法add中我们对变量counter进行一次自增，为了保证counter++的并发安全使用了sync.Mutex对操作进行了加锁确保原子性，保证了并发时间内只有一个携程来更新或者读取它。我们在主协程中通过一个死循环来判断 counter的值，只有当它大于等于 10 时，才退出循环，进而退出整个程序。
代码也因此变得更复杂，更难以维护，这还只是个简单的加法运算实现，就要写这么多代码，要引入共享变量，还要引入互斥锁来保证操作的原子性，对于更加复杂的业务代码，如果到处都要加锁、解锁，显然对开发者和维护者来说都是噩梦，Go 语言既然以并发编程作为语言的核心优势，当然不至于将这样的问题用这么繁琐的方式来解决。