golang数组加锁

2024-10-21 19:16:47 编程来源：ZONE.CI 全球网 0 阅读模式

使用golang数组加锁实现线程安全

Golang是一门以并发为核心设计的编程语言，可以通过使用锁来保护对共享资源的并发访问。在Golang中，数组是一种常见的数据结构，可以在处理大量数据时提供高效的性能。然而，在多线程环境下，对数组的并发访问可能会导致竞争条件的出现。为了解决这个问题，我们可以使用锁来保护数组，确保在任何给定的时间只有一个线程可以访问它。

首先，我们需要导入sync包来使用锁：


import "sync"

使用互斥锁保护数组

互斥锁(Mutex)是Golang提供的一种基本锁机制，它可以用来保护临界区，确保同一时间只有一个goroutine可以执行该临界区的代码。以下是一个示例代码，演示了如何使用互斥锁保护一个数组：


var mu sync.Mutex
var arr []int

func addToArr(num int) {
    mu.Lock()
    arr = append(arr, num)
    mu.Unlock()
}

在上面的代码中，我们首先定义了一个互斥锁mu和一个数组arr。然后，我们定义了一个addToArr函数，该函数用来将一个整数添加到数组中。

在addToArr函数中，我们首先调用mu.Lock()来获取互斥锁，这将阻塞其他goroutine对arr的访问。然后，我们使用append函数将整数添加到数组中。最后，我们调用mu.Unlock()释放互斥锁，允许其他goroutine对arr进行访问。

使用读写锁保护数组

有时候，多个goroutine可能只是读取数组而不修改它。为了提高性能，我们可以使用读写锁(RWMutex)来保护数组。读写锁允许多个goroutine同时读取共享资源，但在有写操作时会阻塞所有的读和写。


var rwMu sync.RWMutex
var arr []int

func readFromArr() []int {
    rwMu.RLock()
    defer rwMu.RUnlock()
    
    return arr
}

func addToArr(num int) {
    rwMu.Lock()
    defer rwMu.Unlock()
    
    arr = append(arr, num)
}

在上面的代码中，我们定义了一个读写锁rwMu和一个数组arr。然后，我们定义了一个readFromArr函数用于读取数组，并使用RWMutex的RLock方法来锁定读取操作。另外，我们还定义了一个addToArr函数用于在数组中添加元素，并使用RWMutex的Lock方法来锁定写入操作。