>  기사  >  백엔드 개발  >  동시 프로그래밍 기술: Go WaitGroup의 고급 사용

동시 프로그래밍 기술: Go WaitGroup의 고급 사용

王林
王林원래의
2023-09-28 21:52:45887검색

并发编程技巧:Go WaitGroup的高级用法

동시 프로그래밍 팁: Go WaitGroup의 고급 사용법

동시 프로그래밍에서는 여러 동시 작업의 실행을 조정하고 관리하는 것이 중요한 작업입니다. Go 언어는 동시성 제어를 우아하게 구현하는 데 도움이 되는 매우 실용적인 동시성 기본 요소인 WaitGroup을 제공합니다. 이 기사에서는 독자가 이를 더 잘 이해하고 적용할 수 있도록 특정 코드 예제를 사용하여 WaitGroup의 기본 사용법을 소개하고 고급 사용법에 중점을 둘 것입니다.

WaitGroup은 Go 언어에 내장된 동시성 기본 요소로, 동시 작업이 완료될 때까지 기다리는 데 도움이 됩니다. Add, Done, Wait의 세 가지 방법을 제공합니다. Add 메서드는 대기 작업 수를 설정하는 데 사용되고, Done 메서드는 대기 작업 수를 줄이는 데 사용되며, Wait 메서드는 모든 대기 작업이 완료될 때까지 현재 코루틴을 차단하는 데 사용됩니다.

다음은 WaitGroup의 기본 사용법을 보여주는 간단한 예입니다.

package main

import (
    "fmt"
    "sync"
    "time"
)

func main() {
    var wg sync.WaitGroup

    for i := 0; i < 5; i++ {
        wg.Add(1)
        go func(num int) {
            defer wg.Done()

            time.Sleep(time.Second)
            fmt.Println("Task", num, "done")
        }(i)
    }

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

위 코드에서는 WaitGroup 개체 wg를 만들고 루프를 통해 5개의 동시 작업을 만듭니다. 각 작업을 실행하는 동안에는 Add 메서드를 사용하여 대기 중인 작업 수를 늘리고, 작업이 끝나면 Done 메서드를 사용하여 대기 중인 작업 수를 줄입니다. 마지막으로 대기 중인 모든 작업이 완료될 때까지 기본 코루틴을 차단하기 위해 Wait 메서드를 호출합니다.

기본 사용법 외에도 WaitGroup은 동시 작업 실행을 보다 유연하게 제어할 수 있는 몇 가지 고급 사용법도 제공합니다. 아래에서는 일반적으로 사용되는 몇 가지 고급 사용법을 자세히 소개합니다.

  1. 일련의 작업을 실행하고 최대 동시성 수 설정

일련의 작업을 동시에 실행해야 하지만 최대 동시성 수를 제한하고 싶다면 버퍼링된 채널을 다음과 결합하여 사용할 수 있습니다. 이를 달성하려면 WaitGroup을 사용하세요. 아래 코드는 일련의 작업을 동시에 실행하는 방법을 보여 주지만 최대 3개의 작업만 동시에 실행할 수 있도록 허용합니다.

package main

import (
    "fmt"
    "sync"
    "time"
)

func main() {
    var wg sync.WaitGroup
    maxConcurrency := 3
    tasks := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}

    sem := make(chan struct{}, maxConcurrency)

    for _, task := range tasks {
        wg.Add(1)
        sem <- struct{}{} // 获取令牌,控制最大并发数

        go func(num int) {
            defer wg.Done()

            time.Sleep(time.Second)
            fmt.Println("Task", num, "done")

            <-sem // 释放令牌,允许新的任务执行
        }(task)
    }

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

위 코드에서는 버퍼링된 채널 sem을 생성하고 크기를 최대 개수로 설정합니다. 동시성의 . 각 작업이 시작되기 전에 sem <- struct{}{} 문을 통해 토큰을 얻습니다. 작업이 완료되면 <-sem 문을 사용하여 토큰을 해제합니다. 토큰 획득 및 출시를 제어함으로써 최대 동시성 수를 제한할 수 있습니다.

  1. Timeout은 동시 작업의 실행을 제어합니다.

때로는 동시 작업의 실행 시간을 제어하고 시간 초과 시 작업 실행을 종료하고 싶을 때가 있습니다. 버퍼링된 채널과 타이머를 사용하면 이 기능을 쉽게 구현할 수 있습니다. 다음 코드는 동시 작업의 시간 초과를 3초로 설정하는 방법을 보여줍니다.

package main

import (
    "fmt"
    "sync"
    "time"
)

func main() {
    var wg sync.WaitGroup
    tasks := []int{1, 2, 3, 4, 5, 6, 7}

    timeout := 3 * time.Second
    done := make(chan struct{})

    for _, task := range tasks {
        wg.Add(1)

        go func(num int) {
            defer wg.Done()

            // 模拟任务执行时间不定
            time.Sleep(time.Duration(num) * time.Second)
            fmt.Println("Task", num, "done")

            // 判断任务是否超时
            select {
            case <-done:
                // 任务在超时前完成,正常退出
                return
            default:
                // 任务超时,向通道发送信号
                close(done)
            }
        }(task)
    }



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

위 코드에서는 done 채널을 생성하고 작업 실행 중에 채널이 닫혔는지 확인하여 작업 시간 초과 여부를 확인합니다. 작업이 완료되면 close(done) 문을 사용하여 완료 채널에 신호를 보내 작업 시간이 초과되었음을 나타냅니다. 다양한 상황을 처리하려면 select 문을 통해 다양한 분기를 선택하세요.

위의 샘플 코드를 통해 WaitGroup의 고급 사용법이 실제 동시 프로그래밍에서 매우 실용적이라는 것을 알 수 있습니다. 이러한 기술을 익히면 동시 작업의 실행을 더 잘 제어하고 코드의 성능과 유지 관리성을 향상시킬 수 있습니다. 독자들이 본 글의 소개와 샘플 코드를 통해 WaitGroup의 사용법을 심도 있게 이해하고 실제 프로젝트에 적용할 수 있기를 바랍니다.

위 내용은 동시 프로그래밍 기술: Go WaitGroup의 고급 사용의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

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