golang基础(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
", 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 语言既然以并发编程作为语言的核心优势，当然不至于将这样的问题用这么繁琐的方式来解决。