>  기사  >  백엔드 개발  >  Golang WaitGroup과 코루틴 풀의 효율적인 조합

Golang WaitGroup과 코루틴 풀의 효율적인 조합

王林
王林원래의
2023-09-28 17:22:491214검색

Golang WaitGroup和协程池的高效结合

Golang WaitGroup과 코루틴 풀의 효율적인 조합에는 특정 코드 예제가 필요합니다

소개:
Go 언어는 동시 프로그래밍을 강조하고 코루틴(고루틴)을 통해 효율적인 동시 실행을 달성하는 언어입니다. 여러 작업을 동시에 실행해야 하는 일부 시나리오에서는 WaitGroup과 코루틴 풀을 결합하면 프로그램 실행 효율성과 리소스 활용도를 효과적으로 향상시킬 수 있습니다. 이 기사에서는 효율적인 동시 프로그래밍을 달성하기 위해 Golang에서 WaitGroup 및 코루틴 풀을 사용하는 방법을 소개하고 구체적인 코드 예제를 제공합니다.

1. WaitGroup 소개
WaitGroup은 코루틴 그룹의 실행이 완료되기를 기다리는 데 사용되는 Go 언어 도구입니다. 소스 코드는 다음과 같이 정의됩니다.

type WaitGroup struct {
    noCopy noCopy

    // 64位的值:高32位存储计数器,低32位存储等待计数器
    // 这个变量可以被原子操作加载和存储。
    // 在64位同步原语中,它必须在64位边界对齐。
    // 是一个强制的要求。
    state1 [3]uint32
}

WaitGroup은 일반적으로 메인 고루틴에서 생성되며, 메인 고루틴의 각 하위 고루틴은 Add 메소드를 호출하여 카운터를 늘리고, 실행이 완료된 후 Done 메소드는 다음과 같습니다. 카운터를 감소시키는 데 사용됩니다. 기본 고루틴은 Wait 메서드를 통해 카운터가 0으로 돌아갈 때까지 기다릴 수 있습니다. 구체적인 샘플 코드는 다음과 같습니다.

package main

import (
    "fmt"
    "sync"
)

func main() {
    var wg sync.WaitGroup
    wg.Add(3)

    go func() {
        defer wg.Done()
        fmt.Println("Task 1 executing")
    }()

    go func() {
        defer wg.Done()
        fmt.Println("Task 2 executing")
    }()

    go func() {
        defer wg.Done()
        fmt.Println("Task 3 executing")
    }()

    wg.Wait()
    fmt.Println("All tasks completed")
}

위의 예에서는 WaitGroup 개체를 만든 다음 Add 메서드를 호출하여 카운터를 증가시킵니다. 다음으로 세 개의 하위 고루틴을 생성했으며, 각 고루틴이 실행된 후 Done 메서드를 통해 카운터가 감소합니다. 마지막으로, 메인 고루틴은 Wait 메서드를 호출하여 카운터가 0으로 돌아갈 때까지 기다립니다. 모든 작업이 완료되면 프로그램은 "모든 작업 완료"를 출력합니다.

2. 코루틴 풀 소개
동시 프로그래밍에서 고루틴 풀은 일반적으로 사용되는 모드입니다. 고정된 수의 고루틴을 생성하고 작업을 균등하게 분배함으로써 지속적으로 고루틴을 생성하고 파괴하는 오버헤드를 피할 수 있습니다. Go 언어에서는 채널을 사용하여 코루틴 풀을 구현할 수 있습니다. 구체적인 샘플 코드는 다음과 같습니다.

package main

import (
    "fmt"
    "sync"
)

func worker(id int, jobs <-chan int, results chan<- int) {
    for j := range jobs {
        fmt.Println("Worker", id, "started job", j)
        fib := fibonacci(j)
        fmt.Println("Worker", id, "finished job", j)
        results <- fib
    }
}

func fibonacci(n int) int {
    if n <= 1 {
        return n
    }

    return fibonacci(n-1) + fibonacci(n-2)
}

const numJobs = 5
const numWorkers = 3

func main() {
    jobs := make(chan int, numJobs)
    results := make(chan int, numJobs)

    var wg sync.WaitGroup
    wg.Add(numWorkers)

    for w := 1; w <= numWorkers; w++ {
        go func(id int) {
            defer wg.Done()
            worker(id, jobs, results)
        }(w)
    }

    for j := 1; j <= numJobs; j++ {
        jobs <- j
    }
    close(jobs)

    wg.Wait()

    for r := 1; r <= numJobs; r++ {
        fmt.Println(<-results)
    }
}

위의 예에서는 작업 채널에서 보류 중인 작업을 읽은 다음 작업을 실행하고 결과를 결과 채널로 보내는 작업자 함수를 정의했습니다. 작업을 분산하고 결과를 얻어 코루틴 풀의 기능을 구현하기 위해 작업 채널과 결과 채널을 만들었습니다.

주 함수에서는 WaitGroup을 사용하여 모든 작업자(고루틴)가 작업 실행을 완료할 때까지 기다립니다. 그런 다음 실행할 작업을 작업 채널로 보내고 실행 후 채널을 닫습니다. 마지막으로 결과 채널에서 계산 결과를 가져와서 출력합니다.

3. WaitGroup과 코루틴 풀의 효율적인 결합 사례
다음으로, 위의 두 개념을 결합하여 동시 프로그래밍을 구현하기 위해 WaitGroup과 코루틴 풀을 효율적으로 사용하는 방법을 소개하겠습니다. 구체적인 샘플 코드는 다음과 같습니다.

package main

import (
    "fmt"
    "sync"
)

func worker(id int, jobs <-chan int, results chan<- int) {
    for j := range jobs {
        fmt.Println("Worker", id, "started job", j)
        fib := fibonacci(j)
        fmt.Println("Worker", id, "finished job", j)
        results <- fib
    }
}

func fibonacci(n int) int {
    if n <= 1 {
        return n
    }

    return fibonacci(n-1) + fibonacci(n-2)
}

const numJobs = 5
const numWorkers = 3

func main() {
    var wg sync.WaitGroup
    wg.Add(numWorkers)

    jobs := make(chan int, numJobs)
    results := make(chan int, numJobs)

    for w := 1; w <= numWorkers; w++ {
        go func(id int) {
            defer wg.Done()
            worker(id, jobs, results)
        }(w)
    }

    for j := 1; j <= numJobs; j++ {
        jobs <- j
    }
    close(jobs)

    go func() {
        wg.Wait()
        close(results)
    }()

    for r := range results {
        fmt.Println(r)
    }
}

위의 예에서는 WaitGroup 개체를 생성하고 Add 메서드를 호출하여 카운터를 증가시켰습니다. 그런 다음 작업을 분배하고 결과를 얻기 위해 작업 채널과 결과 채널을 만들었습니다. 고정된 수의 작업자(고루틴)를 생성하고 Wait 메서드를 사용하여 작업자가 작업을 완료할 때까지 기다립니다.

기본 기능에서는 실행할 작업을 작업 채널로 보내고 실행 후 채널을 닫습니다. 그런 다음 모든 작업자가 작업을 완료할 때까지 기다리고 완료되면 결과 채널을 닫는 코루틴을 시작합니다. 마지막으로 결과 채널에서 계산 결과를 가져와서 출력합니다.

결론:
WaitGroup과 코루틴 풀을 결합하면 동시 프로그래밍을 효율적으로 달성할 수 있습니다. WaitGroup을 사용하여 고루틴 그룹의 실행이 완료될 때까지 기다리면 모든 작업이 완료된 후에도 기본 고루틴이 계속 실행되도록 할 수 있습니다. 코루틴 풀을 사용하면 고루틴을 자주 생성하고 파괴하는 오버헤드를 방지하고 프로그램의 실행 효율성과 리소스 활용도를 향상시킬 수 있습니다.

코드 예제의 피보나치 수열 계산은 단지 데모 예제일 뿐이며 실제 응용 프로그램의 특정 요구에 따라 다른 작업으로 대체될 수 있습니다. WaitGroup과 코루틴 풀을 사용하면 동시에 실행되는 작업 수를 더 효과적으로 제어하고 컴퓨팅 리소스를 효과적으로 활용할 수 있습니다.

Go 언어는 풍부한 동시 프로그래밍 도구와 기능을 제공하지만, 이를 사용할 때는 여전히 주의가 필요합니다. WaitGroup과 코루틴 풀을 적절하게 사용하면 고루틴을 더 잘 관리하고 예약하며 효율적인 동시 프로그래밍을 달성하는 데 도움이 될 수 있습니다.

위 내용은 Golang WaitGroup과 코루틴 풀의 효율적인 조합의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

성명:
본 글의 내용은 네티즌들의 자발적인 기여로 작성되었으며, 저작권은 원저작자에게 있습니다. 본 사이트는 이에 상응하는 법적 책임을 지지 않습니다. 표절이나 침해가 의심되는 콘텐츠를 발견한 경우 admin@php.cn으로 문의하세요.