[Go] golang原子函数锁住共享资源

唯一Chat

发布于 2019-09-10 11:14:33

5550

发布于 2019-09-10 11:14:33

1.atomic包里的几个函数以及sync包里的mutex类型,提供了解决方案 2.原子函数能够以很底层的加锁机制来同步访问整型变量和指针 3.atomic.AddInt64(&counter, 1)的原理是强制同一时刻只能有一个goroutine运行并完成这个加法操作

package main

import (
	"fmt"
	"runtime"
	"sync"
	"sync/atomic"
)

var (
	//全局变量
	counter int64
	//计数信号量
	wg sync.WaitGroup
)

func main() {
	fmt.Println("hello")
	//计数加2,等待两个goroutine
	wg.Add(2)
	go incCounter()
	go incCounter()
	//主goroutine等待子goroutine结束
	wg.Wait()
	fmt.Println("最终counter值:", counter)
}

//增加counter的值函数
func incCounter() {
	//函数结束,减小信号量
	defer wg.Done()
	for count := 0; count < 2; count++ {
		//安全的对counter加1
		atomic.AddInt64(&counter, 1)
		//强制调度器必须切换
		runtime.Gosched()
	}
}

本文参与腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2019-02-15 ，如有侵权请联系 cloudcommunity@tencent.com 删除

atomic