当前位置：首页 > article >正文

Golang 并发机制-4：用Mutex管理共享资源

article 2025/2/4 2:05:20

并发性是Go的强大功能之一，它允许多个线程（并发线程）同时执行。然而，权力越大，责任越大。当多个例程并发地访问和修改共享资源时，可能会导致数据损坏、竞争条件和不可预测的程序行为。为了解决这些问题，Go提供了一个称为互斥的同步原语（互斥的缩写）。在本文中，我们将探讨Mutex在管理共享资源中的作用，以及在并发编程中使用它的必要性。

互斥锁简介

互斥锁是一种同步原语，提供对共享资源或代码关键段的独占访问。它充当看门人，一次只允许一个Goroutine访问和修改受保护的资源。当一个线程持有互斥锁时，所有其他试图获取互斥锁的线程都必须等待轮到它们。
在这里插入图片描述

互斥锁提供了两个基本方法：

Lock() ：该方法获取互斥对象，授予对资源的独占访问权。如果另一个线程已经持有互斥对象，新的线程将阻塞，直到它被释放。
Unlock()：这个方法释放互斥锁，允许其他等待的例程获取互斥锁并访问资源。

互斥锁应用场景

对互斥锁的需求源于这样一个事实，即共享资源在被多个例程并发访问时容易受到数据竞争和不一致的影响。以下是互斥锁必不可少的一些常见场景：

数据竞争

当多个协程并发地访问共享数据，并且其中至少有一个修改共享数据时，就会发生数据竞争。这可能导致不可预测的错误行为，因为无法保证执行顺序。互斥锁通过一次只允许一个协程访问共享资源来防止数据竞争。

package main

import (
    "fmt"
    "sync"
)

var sharedData int
var mu sync.Mutex

func increment() {
    mu.Lock()
    sharedData++
    mu.Unlock()
}

func main() {
    var wg sync.WaitGroup
    for i := 0; i < 100; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            increment()
        }()
    }
    wg.Wait()
    fmt.Println("Shared Data:", sharedData)
}

在这个例子中，多个协程并发地增加sharedData变量，这将导致没有互斥锁的数据竞争。

临界区

临界区是访问共享资源的代码的一部分。当多个go例程试图同时访问同一个临界区时，可能会导致不可预测的行为。互斥锁确保一次只有一个程序进入临界区，保证对共享资源的有序访问。

package main

import (
    "fmt"
    "sync"
)

var (
    sharedResource int
    mu             sync.Mutex
)

func updateSharedResource() {
    mu.Lock()
    // Critical section: Access and modify sharedResource
    sharedResource++
    mu.Unlock()
}

func main() {
    var wg sync.WaitGroup
    for i := 0; i < 100; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            updateSharedResource()
        }()
    }
    wg.Wait()
    fmt.Println("Shared Resource:", sharedResource)
}

在这个例子中，updateSharedResource 函数代表了一个关键区域，其中sharedResource被访问和修改。如果没有互斥锁，对这个临界区的并发访问可能会导致不正确的结果。

互斥锁方法

互斥锁提供了两种基本操作：锁定和解锁。让我们从理解互斥锁开始：

锁定互斥对象：当一个线程想要访问共享资源或临界区时，它会调用互斥对象上的Lock（）方法。如果互斥对象当前处于未锁定状态，它将被锁定，从而允许线程程继续执行。如果互斥锁已经被另一个线程锁住了，调用的线程将被阻塞，直到互斥锁可用。

下面是演示互斥锁的代码示例：

package main

import (
    "fmt"
    "sync"
)

func main() {
    var mu sync.Mutex

    mu.Lock() // Lock the Mutex
    // Critical section: Access and modify shared resource
    fmt.Println("Locked the Mutex")
    mu.Unlock() // Unlock the Mutex
}

解锁互斥锁：当一个线程完成了它的临界区并且不再需要独占访问共享资源时，它调用互斥锁上的Unlock（）方法。这个动作释放互斥锁，允许其他例程获取它。

下面是互斥锁解锁的执行方式：

package main

import (
    "fmt"
    "sync"
)

func main() {
    var mu sync.Mutex

    mu.Lock() // Lock the Mutex
    // Critical section: Access and modify shared resource
    fmt.Println("Locked the Mutex")
    mu.Unlock() // Unlock the Mutex
    fmt.Println("Unlocked the Mutex")
}

在这个例子中，在临界区之后调用mu.Unlock()来释放互斥锁，使其可供其他例程使用。

避免死锁

虽然互斥锁是确保并发安全性的强大工具，但如果使用不当，它们也会引入死锁。当两个或多个线程被卡住，等待对方释放资源时，就会发生死锁。要避免死锁，请遵循以下最佳实践：

Always Unlock：确保互斥锁在锁定后处于解锁状态。如果不这样做，可能会导致死锁。
使用defer：为了保证互斥锁总是被解锁，可以考虑在函数结束时使用defer语句来解锁它们。
避免循环依赖：要小心循环依赖，在循环依赖中，多个例程会等待彼此释放资源。设计代码以避免这种情况。

package main

import (
    "fmt"
    "sync"
)

func main() {
    var mu sync.Mutex

    mu.Lock() // Lock the Mutex
    // Critical section: Access and modify shared resource

    // Oops! Forgot to unlock the Mutex
    // mu.Unlock() // Uncomment this line to avoid deadlock
    fmt.Println("Locked the Mutex")

    // ... Some more code

    // Potential deadlock if mu.Unlock() is not called
}