为啥两个线程同时执行 ++i，可能会导致其中一个线程的自增操作被覆盖

易先讯 / 2024-07-09 / 原文

这是一个典型的并发编程问题，涉及到线程的同步和共享资源的访问问题。让我们详细来看一下为什么会这样：

自增操作的细节

++i 看起来是一个简单的操作，但实际上它包含了多个步骤：

读取变量 i 的当前值。
将读取到的值加 1。
将结果写回变量 i。

在单线程环境中，这些步骤会顺序执行，不会有问题。

多线程环境中的问题

当两个线程同时执行 ++i 操作时，以下情况可能发生：

线程A 读取 i 的当前值，比如 i=5。
线程B 也读取 i 的当前值，仍然是 i=5。
线程A 将 i 加 1，结果是 6，并写回 i。
线程B 也将读取到的 i 加 1，结果是 6，并写回 i。

最终，虽然两个线程都执行了自增操作，但结果却只增加了 1，而不是预期的 2，这就是所谓的“覆盖”问题。

解决方法

要解决这个问题，可以使用同步机制来确保每次只有一个线程执行 ++i 操作。以下是一些常见的方法：

使用互斥锁（Mutex）：在执行 ++i 操作时，加锁保证只有一个线程能访问这个操作。

java
 
Copy

   synchronized (this) {
       ++i;
   }

使用原子操作（Atomic Operations）：例如在Java中可以使用 AtomicInteger，它提供了一些线程安全的操作。

java
 
Copy

   AtomicInteger atomicInteger = new AtomicInteger();
   atomicInteger.incrementAndGet();

使用其他高层次的并发控制机制：如信号量（Semaphore）、读写锁（ReadWriteLock）等。

通过这些方法，可以确保自增操作在多线程环境下的正确性，避免覆盖问题。

在 Go 语言中，也存在类似的问题，当多个 Goroutine 同时访问和修改共享变量时，会出现竞态条件（Race Condition）。为了避免这种情况，可以使用以下几种方法：

方法一：使用互斥锁（Mutex）

使用 sync 包中的 Mutex 来保护共享变量的访问。

go
 
Copy

package main

import (
    "fmt"
    "sync"
)

func main() {
    var i int
    var mu sync.Mutex
    var wg sync.WaitGroup

    for j := 0; j < 1000; j++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            mu.Lock()
            i++
            mu.Unlock()
        }()
    }

    wg.Wait()
    fmt.Println("Final value of i:", i)
}

在这个例子中，mu.Lock() 和 mu.Unlock() 确保了每次只有一个 Goroutine 能执行 i++ 操作。

方法二：使用原子操作（Atomic Operations）

使用 sync/atomic 包提供的原子操作，可以在不使用锁的情况下实现线程安全。

go
 
Copy

package main

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

func main() {
    var i int32
    var wg sync.WaitGroup

    for j := 0; j < 1000; j++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            atomic.AddInt32(&i, 1)
        }()
    }

    wg.Wait()
    fmt.Println("Final value of i:", i)
}

在这个例子中，atomic.AddInt32(&i, 1) 确保了 i 的自增操作是原子的，不会被其他 Goroutine 干扰。

方法三：使用通道（Channels）

Go 提供了通道（Channels），可以用于在 Goroutine 之间传递数据，避免共享内存的直接修改。

go
 
Copy

package main

import (
    "fmt"
    "sync"
)

func main() {
    var i int
    var wg sync.WaitGroup
    ch := make(chan bool)

    for j := 0; j < 1000; j++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            ch <- true
        }()
    }

    go func() {
        for range ch {
            i++
        }
    }()

    wg.Wait()
    close(ch) // 关闭通道，确保所有数据都被处理
    fmt.Println("Final value of i:", i)
}