溫馨提示×

golang多線程共享變量的方法是什么

小億
128
2023-11-28 14:17:28
欄目: 編程語言

在Golang中,可以使用goroutine和channel來實現(xiàn)多線程共享變量。

  1. 使用goroutine和channel:

    • 使用goroutine創(chuàng)建多個線程,每個線程可以并發(fā)執(zhí)行。
    • 使用channel來進行線程間的通信和數(shù)據(jù)共享,確保線程安全。
    • 通過channel發(fā)送數(shù)據(jù)到指定的goroutine,并通過channel接收來自其他goroutine的數(shù)據(jù)。

    示例代碼如下:

    package main
    
    import (
        "fmt"
    )
    
    func worker(id int, jobs <-chan int, results chan<- int) {
        for j := range jobs {
            fmt.Println("Worker", id, "started job", j)
            results <- j * 2 // 將處理結(jié)果發(fā)送到results通道中
        }
    }
    
    func main() {
        jobs := make(chan int, 100)
        results := make(chan int, 100)
    
        // 創(chuàng)建3個worker goroutine
        for w := 1; w <= 3; w++ {
            go worker(w, jobs, results)
        }
    
        // 向jobs通道中發(fā)送任務(wù)
        for j := 1; j <= 9; j++ {
            jobs <- j
        }
        close(jobs)
    
        // 從results通道中接收處理結(jié)果
        for a := 1; a <= 9; a++ {
            <-results
        }
    }
    

    上述示例中,使用goroutine創(chuàng)建了3個worker線程,它們從jobs通道中接收任務(wù),并將處理結(jié)果發(fā)送到results通道中。在main函數(shù)中,向jobs通道中發(fā)送9個任務(wù),然后從results通道中接收9個處理結(jié)果。

  2. 使用sync包中的互斥鎖:

    • 使用sync包中的Mutex類型來保護共享變量,確保同一時間只有一個線程可以訪問共享變量。
    • 使用Lock方法鎖定共享變量,在訪問共享變量之前調(diào)用,使用Unlock方法解鎖共享變量,在訪問共享變量之后調(diào)用。

    示例代碼如下:

    package main
    
    import (
        "fmt"
        "sync"
    )
    
    var (
        counter int
        wg      sync.WaitGroup
        mutex   sync.Mutex
    )
    
    func worker() {
        defer wg.Done()
        for i := 0; i < 1000; i++ {
            mutex.Lock()
            counter++
            mutex.Unlock()
        }
    }
    
    func main() {
        wg.Add(2)
        go worker()
        go worker()
        wg.Wait()
        fmt.Println("Counter:", counter)
    }
    

    上述示例中,使用sync包中的Mutex類型來保護共享變量counter,確保同一時間只有一個線程可以訪問counter。在每個worker函數(shù)中,使用Lock方法鎖定counter,在訪問counter之前調(diào)用,使用Unlock方法解鎖counter,在訪問counter之后調(diào)用。最后輸出counter的值。

無論是使用goroutine和channel,還是使用互斥鎖,都可以實現(xiàn)多線程共享變量。具體使用哪種方式取決于具體的業(yè)務(wù)場景和需求。

0