Go语言并发Goroutine的互斥控制：实现关键代码段的独占执行

package main

import (
    "fmt"
    "math/rand"
    "sync"
    "time"
)

var (
    // 定义一个全局共享的互斥锁，用于保护所有Goroutine的临界区
    sharedMutex sync.Mutex
    // WaitGroup 用于等待所有Goroutine完成
    wg sync.WaitGroup
)

// 模拟一些非关键操作
func doSomething(routineID int, stage string) {
    fmt.Printf("Routine %d: %s - 执行非关键操作...\n", routineID, stage)
    time.Sleep(time.Duration(rand.Intn(50)) * time.Millisecond) // 模拟耗时
}

// 模拟关键操作序列（临界区）
func criticalSection(routineID int) {
    fmt.Printf("Routine %d: 进入关键操作区。\n", routineID)
    // 模拟发送和打印事件，这里用多个打印语句代替
    for i := 0; i < 3; i++ {
        fmt.Printf("Routine %d: 关键事件 %d\n", routineID, i+1)
        time.Sleep(time.Duration(rand.Intn(30)) * time.Millisecond) // 模拟耗时
    }
    fmt.Printf("Routine %d: 退出关键操作区。\n", routineID)
}

// 每个Goroutine的执行逻辑
func runRoutine(id int) {
    defer wg.Done() // Goroutine完成时通知WaitGroup

    doSomething(id, "前置操作")

    // 在进入关键操作区前获取锁
    sharedMutex.Lock()
    criticalSection(id)
    // 在退出关键操作区后释放锁
    sharedMutex.Unlock()

    doSomething(id, "后置操作")
}

func main() {
    rand.Seed(time.Now().UnixNano()) // 初始化随机数种子

    wg.Add(3) // 告知WaitGroup将启动3个Goroutine

    go runRoutine(1)
    go runRoutine(2)
    go runRoutine(3)

    wg.Wait() // 等待所有Goroutine完成
    fmt.Println("所有Goroutine已完成。")
}