Go Sync 包:并发的 6 个关键概念

简介: Go Sync 包:并发的 6 个关键概念

1.sync.Mutex和sync.RWMutex


要知道,mutex(互斥)对于我们 gopher 来说就像一个老伙计。在处理 goroutine 时,确保它们不会同时访问资源是非常重要的,而 mutex 可以帮助我们做到这一点。


sync.Mutex


看看这个简单的例子,我没有使用互斥锁来保护我们的变量 a


var a = 0
func Add() {
  a++
}
func main() {
  for i := 0; i < 500; i++ {
    go Add()
  }
  time.Sleep(5 * time.Second)
  fmt.Println(a)
}


此代码的结果是不可预测的。如果幸运的话,您可能会得到 500,但通常结果会小于 500。现在,让我们使用互斥体增强我们的 Add 函数:


var mtx = sync.Mutex{}
func Add() {
  mtx.Lock()
  defer mtx.Unlock()
  a++
}

现在,代码提供了预期的结果。但是使用 sync.RWMutex 呢?


为什么使用 sync.RWMutex?


想象一下,您正在检查 a 变量,但其他 goroutines 也在调整它。您可能会得到过时的信息。那么,解决这个问题的方法是什么?


让我们退后一步,使用我们的旧方法,将 sync.Mutex 添加到我们的 Get() 函数中:


func Add() {
  mtx.Lock()
  defer mtx.Unlock()
  a++
}
func Get() int {
  mtx.Lock()
  defer mtx.Unlock()
  return a
}

但这里的问题是,如果您的服务或程序调用 Get() 数百万次而只调用 Add() 几次,那么我们实际上是在浪费资源,因为我们大部分时间甚至都没有修改它而将所有内容都锁定了。


这就是 sync.RWMutex 突然出现来拯救我们的一天,这个聪明的小工具旨在帮助我们处理同时读取和写入的情况。


var mtx = sync.RWMutex{}
func Add() {
  mtx.Lock()
  defer mtx.Unlock()
  
  a++
}
func Look() {
  mtx.RLock()
  defer mtx.RUnlock()
  
  fmt.Println(a)
}

那么,RWMutex 有什么了不起的呢?好吧,它允许数百万次并发读取,同时确保一次只能进行一次写入。让我澄清一下它是如何工作的:


  • 写入时,读取被锁定。
  • 读取时,写入被锁定。
  • 多次读取不会相互锁定。


sync.Locker


哦对了,Mutex和RWMutex都实现了sync.Locker接口{},签名是这样的:


// A Locker represents an object that can be locked and unlocked.
type Locker interface {
  Lock()
  Unlock()
}

如果你想创建一个接受 Locker 的函数,你可以将这个函数与你的自定义 locker 或同步互斥锁一起使用:


func Add(mtx sync.Locker) {
  mtx.Lock()
  defer mtx.Unlock()
  
  a++
}


2. sync.WaitGroup


您可能已经注意到我使用了 time.Sleep(5 * time.Second) 来等待所有 goroutine 完成,但老实说,这是一个非常丑陋的解决方案。


这就是 sync.WaitGroup 出现的地方:


func main() {
  wg := sync.WaitGroup{}
  for i := 0; i < 500; i++ {
    wg.Add(1)
    go func() {
      defer wg.Done()
      Add()
    }()
  }
  
  wg.Wait()
  fmt.Println(a)
}

sync.WaitGroup 有 3 个主要方法:Add、Done 和 Wait。


首先是 Add(delta int):此方法将 WaitGroup 计数器增加 delta 的值。你通常会在生成 goroutine 之前调用它,表示有一个额外的任务需要完成。


如果我们将 WaitGroup 放在 go func() {} 中,您认为会发生什么?


go func() {
  wg.Add(1)
  defer wg.Done()
  Add()
}()


我的编译器喊道,“应该在启动 goroutine 之前调用 wg.Add(1) 以避免竞争”,我的运行时出现恐慌,“panic: sync: WaitGroup is reused before previous Wait has returned”。


其他两种方法非常简单:


  • 当一个 goroutine 结束它的任务时, Done 被调用。
  • Wait 会阻塞调用者,直到 WaitGroup 计数器归零,这意味着所有派生的 goroutine 都已完成它们的任务。


3. sync.Once


假设您在一个包中有一个 CreateInstance() 函数,但您需要确保它在使用前已初始化。所以你在不同的地方多次调用它,你的实现看起来像这样:


var i = 0
var _isInitialized = false
func CreateInstance() {
  if _isInitialized {
    return
  }
  
  i = GetISomewhere()
  _isInitialized = true
}

但是如果有多个 goroutine 调用这个方法呢? i = GetISomeWhere 行会运行多次,即使您为了稳定性只希望它执行一次。

您可以使用我们之前讨论过的互斥锁,但同步包提供了一种更方便的方法:sync.Once

var i = 0
var once = &sync.Once{}
func CreateInstance() {
  once.Do(func() {
    i = GetISomewhere()
  })
}


使用 sync.Once,你可以确保一个函数只执行一次,不管它被调用了多少次或者有多少 goroutines 同时调用它。


4. sync.Pool


想象一下,你有一个池,里面有一堆你想反复使用的对象。这可以减轻垃圾收集器的一些压力,尤其是在创建和销毁这些资源的成本很高的情况下。

所以,无论何时你需要一个对象,你都可以从池中取出它。当您使用完它时,您可以将它放回池中以备日后重复使用。


var pool = sync.Pool{
  New: func() interface{} {
    return 0
  },
}
func main() {
  pool.Put(1)
  pool.Put(2)
  pool.Put(3)
  
  a := pool.Get().(int)
  b := pool.Get().(int)
  c := pool.Get().(int)
  
  fmt.Println(a, b, c) // Output: 1, 3, 2 (order may vary)
}


请记住,将对象放入池中的顺序不一定是它们出来的顺序,即使多次运行上述代码时顺序也是随机。

让我分享一些使用 sync.Pool 的技巧:


  • 它非常适合长期存在并且有多个实例需要管理的对象,例如数据库连接(1000 个连接?)、worker goroutine,甚至缓冲区。
  • 在将对象返回池之前始终重置对象的状态。这样,您可以避免任何无意的数据泄漏或奇怪的行为。
  • 不要指望池中已经存在的对象,因为它们可能会意外释放。


5. sync.Map


当您同时使用 map 时,有点像使用 RWMutex。您可以同时进行多次读取,但不能进行多次读写或写入。如果存在冲突,您的服务将崩溃而不是覆盖数据或导致意外行为。


这就是 sync.Map 派上用场的地方,因为它可以帮助我们避免这个问题。让我们仔细看看 sync.Map 给我们提供什么:


  • CompareAndDelete (go 1.20):如果值匹配则删除键的条目;如果不存在值或旧值为 nil,则返回 false。
  • CompareAndSwap(go 1.20):如果新旧值匹配,则交换一个键,只要确保旧值是可比较的。
  • Swap (go 1.20):交换键的值并返回旧值(如果存在)。
  • LoadOrStore:获取当前键值或保存并返回提供的值(如果不存在)
  • Range (f func(key, value any):遍历映射,将函数 f 应用于每个键值对。如果 f 说返回 false,它会停止。
  • Store
  • Delete
  • Load
  • LoadAndDelete

Q: 我们为什么不使用带有 Mutex 的常规 map 呢?


我通常选择带有 RWMutex 的 map,但在某些情况下认识到 sync.Map 的强大功能很重要。那么,它真正发光的地方在哪里呢?


如果您有许多 goroutines 访问 map 中的单独键,则具有单个互斥锁的常规 map 可能会导致争用,因为它仅针对单个写操作锁定整个 map。

另一方面,sync.Map 使用更完善的锁定机制,有助于最大限度地减少此类场景中的争用。

6. sync.Cond


将 sync.Cond 视为支持多个 goroutine 等待和相互交互的条件变量。为了更好地理解,让我们看看如何使用它。


首先,我们需要创建带有 Locker 的 sync.Cond:


var mtx sync.Mutex
var cond = sync.NewCond(&mtx)


goroutine 调用 cond.Wait 并等待来自其他地方的信号以继续执行:


func dummyGoroutine(id int) {
  cond.L.Lock()
  defer cond.L.Unlock()
  fmt.Printf("Goroutine %d is waiting...\n", id)
  cond.Wait()
  fmt.Printf("Goroutine %d received the signal.\n", id)
}


然后,另一个 goroutine(就像主 goroutine)调用 cond.Signal(),让我们等待的 goroutine 继续:


func main() {
  go dummyGoroutine(1)
  
  time.Sleep(1 * time.Second)
  fmt.Println("Sending signal...")
  cond.Signal()
  time.Sleep(1 * time.Second)
}


结果如下所示:


Goroutine 1 is waiting...
Sending signal...
Goroutine 1 received the signal.


如果有多个 goroutines 在等待我们的信号怎么办?这就是我们可以使用广播的时候:


func main() {
  go dummyGoroutine(1)
  go dummyGoroutine(2)
  
  time.Sleep(1 * time.Second)
  cond.Broadcast() // broadcast to all goroutines
  time.Sleep(1 * time.Second)
}


结果如下所示:


Goroutine 1 is waiting...
Goroutine 2 is waiting...
Goroutine 2 received the signal.
Goroutine 1 received the signal.
相关文章
|
7天前
|
Go 调度 开发者
探索Go语言中的并发模式:goroutine与channel
在本文中,我们将深入探讨Go语言中的核心并发特性——goroutine和channel。不同于传统的并发模型,Go语言的并发机制以其简洁性和高效性著称。本文将通过实际代码示例,展示如何利用goroutine实现轻量级的并发执行,以及如何通过channel安全地在goroutine之间传递数据。摘要部分将概述这些概念,并提示读者本文将提供哪些具体的技术洞见。
|
11天前
|
编译器 Go 开发者
go语言中导入相关包
【11月更文挑战第1天】
23 3
|
18天前
|
并行计算 安全 Go
Go语言的并发特性
【10月更文挑战第26天】Go语言的并发特性
8 1
|
23天前
|
Java 大数据 Go
Go语言:高效并发的编程新星
【10月更文挑战第21】Go语言:高效并发的编程新星
46 7
|
28天前
|
安全 Go 调度
探索Go语言的并发模式:协程与通道的协同作用
Go语言以其并发能力闻名于世,而协程(goroutine)和通道(channel)是实现并发的两大利器。本文将深入了解Go语言中协程的轻量级特性,探讨如何利用通道进行协程间的安全通信,并通过实际案例演示如何将这两者结合起来,构建高效且可靠的并发系统。
|
28天前
|
安全 Go 开发者
破译Go语言中的并发模式:从入门到精通
在这篇技术性文章中,我们将跳过常规的摘要模式,直接带你进入Go语言的并发世界。你将不会看到枯燥的介绍,而是一段代码的旅程,从Go的并发基础构建块(goroutine和channel)开始,到高级模式的实践应用,我们共同探索如何高效地使用Go来处理并发任务。准备好,让Go带你飞。
|
26天前
|
安全 程序员 Go
深入浅出Go语言的并发之道
在本文中,我们将探索Go语言如何优雅地处理并发编程。通过对比传统多线程模型,我们将揭示Go语言独特的goroutine和channel机制是如何简化并发编程,并提高程序的效率和稳定性。本文不涉及复杂的技术术语,而是用通俗易懂的语言,结合生动的比喻,让读者能够轻松理解Go语言并发编程的核心概念。
|
JavaScript Go
|
7天前
|
存储 JSON 监控
Viper,一个Go语言配置管理神器!
Viper 是一个功能强大的 Go 语言配置管理库,支持从多种来源读取配置,包括文件、环境变量、远程配置中心等。本文详细介绍了 Viper 的核心特性和使用方法,包括从本地 YAML 文件和 Consul 远程配置中心读取配置的示例。Viper 的多来源配置、动态配置和轻松集成特性使其成为管理复杂应用配置的理想选择。
24 2
|
5天前
|
Go 索引
go语言中的循环语句
【11月更文挑战第4天】
13 2