5000 字带你精通golang sync包

golang是一门支持并发编程的语言,它提供了goroutine和channel等机制来实现多个任务的并行执行。但是,并发编程也会带来一些挑战,比如数据竞争、死锁、内存泄漏等。为了解决这些问题,golang提供了一个标准库sync,它包含了一些高性能的同步原语,可以帮助我们更好地管理并发状态和资源。

一、sync.Mutex

相信大多数同学都有线上抢购东西的经历,在开始抢购的一瞬间,有大量的用户都发起了请求,形成了不同的线程,对同一个商品进行抢购。现在我们来模拟一下这个场景,假设待抢购的商品是一款网红电视机,库存为1000台,在开始抢购的一瞬间,有刚好1000人点击了购买按钮,按照预期,抢购完成后,库存为0,代码如下:

func main() {
	stock := 1000
	group := sync.WaitGroup{}
	group.Add(1000)
	for i := 0; i < 1000; i++ {
		go func() {
			stock -= 1
			group.Done()
		}()
	}
	group.Wait()
	fmt.Println(stock)
}

输出如下:

76

可能不熟悉并发编程的同学可能会想:咦?为啥不是0呢?归根到底,-=1这个操作并不是原子性的,为了解决这个问题,go引入了sync.Mutex{},这是一个互斥锁,它可以保证在任意时刻,只有一个goroutine可以访问某个共享变量或临界区。我们可以使用Lock()和Unlock()方法来加锁和解锁。我们对上面的代码做如下的改造例如:

func main() {
	stock := 1000
	mutex := sync.Mutex{} //1.声明互斥锁
	group := sync.WaitGroup{} 
	group.Add(1000)
	for i := 0; i < 1000; i++ {
		go func() {
			mutex.Lock() //2.加锁
			stock -= 1
			mutex.Unlock() //3.解锁
			group.Done()
		}()
	}
	group.Wait()
	fmt.Println(stock)
}

输出如下:

0

为了保证stock的正确性,我们使用了sync.Mutex{}来加锁和解锁,这样同时只会有一个协程对stock变量进行操作,这样就可以避免数据竞争的问题,最终输出结果也符合我们最终的预期。

二、sync.RWMutex

sync.Mutext解决了并发问题,但是在实际使用场景中,有很多时候读的次数是远大于写的次数的,读取数据并不会对数据造成影响,只需要限制其他协程不能对数据同时进行修改即可,不需要限制其他的协程对该数据的读取操作。sync.RWMutex{}是一个读写锁,它可以保证在任意时刻,只有一个goroutine可以对某个共享变量或临界区进行写操作,但是可以有多个goroutine同时进行读操作。我们可以使用RLock()和RUnlock()方法来加读锁和解读锁,以及Lock()和Unlock()方法来加写锁和解写锁。例如:

var data int
var rwmu sync.RWMutex

func readData() int {
    rwmu.RLock() // 加读锁
    defer rwmu.RUnlock() // 延迟解读锁
    return data // 读取共享变量
}

func writeData(n int) {
    rwmu.Lock() // 加写锁
    defer rwmu.Unlock() // 延迟解写锁
    data = n // 写入共享变量
}

func main() {
    var wg sync.WaitGroup
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go func(i int) {
            defer wg.Done()
            if i%2 == 0 {
                writeData(i) // 写操作
            } else {
                fmt.Println(readData()) // 读操作
            }
        }(i)
    }
    wg.Wait()
}

上面的代码中,我们定义了一个全局变量data,用来存储一个整数。我们启动了10个goroutine,其中偶数序号的goroutine都调用writeData()函数来对data进行写操作,奇数序号的goroutine都调用readData()函数来对data进行读操作。为了保证data的正确性和并发性能,我们在writeData()函数中使用了sync.RWMutex{}来加写锁和解写锁,在readData()函数中使用了sync.RWMutex{}来加读锁和解读锁。这样就可以允许多个goroutine同时读取data,但是只有一个goroutine可以修改data。

三、sync.Once{}

sync.Once{}是一个只执行一次的对象,它可以保证在多个goroutine中,某个函数或代码块只被执行一次。我们可以使用Do()方法来传入要执行的函数。例如:

var once sync.Once

func initConfig() {
    fmt.Println("init config")
}

func main() {
    var wg sync.WaitGroup
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            once.Do(initConfig) // 只执行一次
        }()
    }
    wg.Wait()
}

上面的代码中,我们定义了一个全局变量once,用来控制initConfig()函数只被执行一次。我们启动了10个goroutine,每个goroutine都调用once.Do(initConfig)来尝试执行initConfig()函数。但是由于once的保证,只有第一个goroutine能够成功执行initConfig()函数,后面的goroutine都会被忽略。这样就可以避免重复初始化或资源浪费的问题。

四、sync.Pool{}

sync.Pool{}是一个临时对象池,它可以缓存一些可重用的对象,以减少内存分配和垃圾回收的开销。我们可以使用New字段来指定对象的创建方式,以及Get()和Put()方法来获取和归还对象。例如:

type Data struct {
    Content string
}

var pool = sync.Pool{
    New: func() interface{} {
        return &Data{} // 创建对象的方式
    },
}

func main() {
    var wg sync.WaitGroup
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go func(i int) {
            defer wg.Done()
            data := pool.Get().(*Data) // 获取对象
            defer pool.Put(data) // 归还对象
            data.Content = fmt.Sprintf("data %d", i) // 修改对象
            fmt.Println(data.Content) // 输出对象
        }(i)
    }
    wg.Wait()
}

上面的代码中,我们定义了一个全局变量pool,用来缓存Data类型的对象。我们启动了10个goroutine,每个goroutine都从pool中获取一个Data对象,修改它的Content字段,然后输出它的内容,最后归还给pool。这样就可以避免每次都创建和销毁Data对象,提高性能和内存利用率。

五、sync.Map{}

sync.Map{}是一个并发安全的映射,它可以在多个goroutine中存储和读取键值对。我们可以使用Store()、Load()、Delete()、Range()等方法来操作映射。例如:

var m sync.Map

func main() {
    var wg sync.WaitGroup
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go func(i int) {
            defer wg.Done()
            m.Store(i, i*i) // 存储键值对
        }(i)
    }
    wg.Wait()
    m.Range(func(key, value interface{}) bool {
        fmt.Println(key, value) // 遍历映射
        return true
    })
}

上面的代码中,我们定义了一个全局变量m,用来存储整数和其平方的映射。我们启动了10个goroutine,每个goroutine都向m中存储一个键值对。然后我们使用m.Range()方法来遍历m中的所有键值对,并输出它们。这样就可以避免使用普通的map时需要加锁的问题,提高并发性能。

六、sync.Cond{}

sync.Cond{}是一个条件变量,它可以让一组goroutine在满足某个条件时被唤醒。我们可以使用NewCond()函数来创建一个条件变量,并传入一个互斥锁作为参数。我们还可以使用Wait()、Signal()、Broadcast()等方法来等待、单发通知、广播通知等。例如:

var queue []int
var mu sync.Mutex
var cond = sync.NewCond(&mu)

func produce(i int) {
    mu.Lock() // 加锁
    defer mu.Unlock() // 延迟解锁
    queue = append(queue, i) // 生产数据
    fmt.Println("produce:", i)
    cond.Signal() // 通知一个消费者
}

func consume(i int) {
    mu.Lock() // 加锁
    for len(queue) == 0 { // 如果队列为空
        cond.Wait() // 等待生产者通知
    }
    data := queue[0] // 消费数据
    queue = queue[1:]
    mu.Unlock() // 解锁
    fmt.Println("consume:", i, data)
}

func main() {
    var wg sync.WaitGroup
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go func(i int) {
            defer wg.Done()
            produce(i) // 生产数据
        }(i)
    }
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go func(i int) {
            defer wg.Done()
            consume(i) // 消费数据
        }(i)
    }
    wg.Wait()
}

上面的代码中,我们定义了一个全局变量queue,用来存储一些整数。我们还定义了一个全局变量cond,用来创建一个条件变量,并传入一个互斥锁mu作为参数。我们启动了10个生产者goroutine和10个消费者goroutine,每个生产者goroutine都调用produce()函数来向queue中添加一个整数,并使用cond.Signal()方法来通知一个消费者goroutine;每个消费者goroutine都调用consume()函数来从queue中取出一个整数,并使用cond.Wait()方法来等待生产者goroutine的通知。这样就可以实现一个简单的生产者-消费者模型,避免队列为空或满时的阻塞问题。

七、总结

golang并发编程中sync包提供了一些高性能的同步原语,可以帮助我们更好地管理并发状态和资源。在平时的编码过程中,要注意sync不同的使用场景,不可为了使用而使用,当然,装逼除外!

#golang##golang项目##go项目##go#
全部评论

相关推荐

2025-12-27 16:21
已编辑
门头沟学院 Java
bg:中下211本科,java后端,无竞赛,无基础,大一升大二暑假开始学java。五段实习:美团-小红书-腾讯-淘天-字节。面秋招的简历只有美团、小红书、淘天。刚刚发现我的秋招蚂蚁流程挂了,这是我最后一个流程,那么我的秋招就算彻底结束了,总结一下:字节ssp+,职级2-1。美团ssp,+2打了半小时微信电话极力挽留。快手ssp,但报了字节薪资后没有争取的想法了。小红书sp,今年小红书给的很高,但比字节2-1还是差很多。虾皮应该是小sp?对虾皮一点意向都没,纯拿来集邮了。淘天ssp(暑期转正),说不要我的三方,毕业前考虑好了随时可以不签三方选择淘天。挂了的流程:京东二面挂,估计学历被卡了。懂车帝一面挂,和面试官聊不来,不认同我的方案。拼多多hr面挂,问我低于预期还来不来,当时说不考虑了,估计觉得我不忠诚。蚂蚁hr面挂,聊的还行,但估计我不会去给我挂了吧。阿里控股一面挂,没面前就知道是kpi了,因为时间可选的很多,而且都是半小时,我也拿他刷我的kpi了。上面差不多是我的情况,下面是我想说的话。我觉得我不算特别突出优秀的那类人,但我多少也算是靠前的那一批人,即使这样,秋招也不算特别顺利,也有挂了的流程,但你能说是我的问题吗,我觉得大部分情况不是的,如果真的是我的问题,我不可能本科校招拿到2-1,所以很多面试挂了,问题不出在面试者身上,很多是看运气+眼缘+和面试官合不合得来。所以我觉得,学会察言观色,了解面试官的脾性,也是面试很重要的一个点。比如面试官是喜欢听长回答,还是听短回答,他更看重哪些点,每个面试官对这些的侧重都是不一样的,所以作为面试者,要学会察言观色,通过面试官开局的一两个问题以及你回答后他的表现,就要判断出来。像我现在其实面试开局个五分钟,我就基本能判断个七七八八了,然后我后面的回答就会有所变化。这是我想说的第一个点:不要为面试结果焦虑,有时候问题不出在你身上,但你可以学一些面试技巧,尽量提高你的面试通过率,这里说的面试技巧指的不是网上那种烂大街的,一两分钟短视频说什么提高你面试通过率的,而是你要在你自己的面试过程中不断总结经验,吸取教训,旁人教你的终究是有限的。另外想说下选offer的事,上面其实可以看出来,我秋招最后是选了字节的,还没签三方我就来提前实习感受业务了,当我签完三方又过了一个多月,我这些天又在想这个问题,字节真的是我想要的吗,我现在总结了一下字节的好坏,发现当时可能被字节的高薪资影响判断了,如果现在再选一次的话,我应该会选杭州的小红书,会生活的更舒服点。具体种种就不展开说了。然后虽然我现在也可以说去把小红书舔回来,去毁字节,但我觉得没必要这么做,我可以采用其他的措施去不就,比如规划好两年内就跳槽,跳到杭州,跳到更舒适的城市。我觉得大家选offer的时候,真的可以冷静下来多方面考虑,薪资、城市、组内氛围、业务、老板是否看重、组内情况、未来升职机会等等都是可以考虑的因素,虽然有的时候不管选哪个,都不会坏,但最好也别让自己后悔吧,即使真后悔了,我觉得也没必要过度美化没走过的路,想好补救措施即可。这是我想说的第二个点:冷静好好做选择,不管是offer还是其他。但人生容错率很大,即使选错了,也一定有补救措施。最后还想说一些成长上的东西,尤其是现在AI火热的时代。我觉得大家如果想提高自己,或者说在未来社招跳槽有竞争力,肯定是要学AI相关的东西的,不说要会多懂AI,至少也要了解基本概念,而且一定要学会用AI提效。我现在字节的mt和我说,他现在80%代码都是AI写的。而我最近也开始尝试用AI工具,感觉现在AI真的进步很多,挺聪明的了,我现在写需求基本都是先让AI写,我再人工review小改动一下就差不多了。我觉得「AI取代程序员」是个很远的话题,但是「AI取代不会用AI的程序员」,可能真的就是近两年的事了。而怎么去学习这块的内容,其实我也正在探索,我也是刚学AI的起步阶段,我觉得大家也要有自己的信息检索能力,而不是别人喂你什么,你才学什么,自己一个人就不会学了。这是我想说的第三个点:趁年轻,多学习提升自己,拥抱AI,不要原地踏步,原地踏步的程序员最容易被淘汰。大概就是这样吧,今天看蚂蚁流程发现挂了,前几天腾讯约面我也拒了,就想到自己的秋招/校招算彻底结束了,有感而发,随便聊了下。牛客以后应该不会更新,大家不用关注,熟悉我的朋友应该知道我在其他平台有号。我更喜欢以长视频的形式去做分享,感觉会更有体系,而不是网上那种一两分钟的零碎短视频的那种营销号去起号,我也推荐大家多去看高质量的长文章、长视频,我觉得收获的能更多。希望大家能收获满意的offer与未来。
CEXBB:刷到最后才发现原来是优雅✌🏻,我的Java引路人
2025年终总结
点赞 评论 收藏
分享
评论
6
16
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务