Go에서 WaitGroup.Wait()에 시간 초과를 추가하는 방법은 무엇입니까?

영원히 기다리기 지치셨나요? WaitGroup.Wait()에 시간 제한 추가

WaitGroup.Wait()를 사용하여 고루틴을 동기화하고 실행이 완료되었는지 확인할 때 때때로 특정 기간 후에 대기를 종료하는 기능이 필요할 수 있습니다. 이렇게 하면 시스템이 자신의 작업을 완료하지 못할 수도 있는 잘못된 작업자에게 끝없이 의존하는 것을 방지할 수 있습니다.

시간 제한 대기 그룹을 위한 실용적인 솔루션

시간 초과를 구현하는 한 가지 접근 방식은 고루틴, 채널, 시간 패키지의 조합입니다. 아이디어는 모든 고루틴이 완료되면 신호를 수신하는 채널을 만드는 것입니다. 동시에 타이머는 원하는 시간 초과 기간으로 초기화됩니다. select 문은 채널과 타이머를 모두 모니터링하는 데 사용됩니다.

import (
    "sync"
    "time"
)

func task(wg *sync.WaitGroup) {
    // Perform some task...
    time.Sleep(time.Second * 2)
    wg.Done()
}

func main() {
    var wg sync.WaitGroup
    wg.Add(2)
    go task(&wg)
    go task(&wg)

    ch := make(chan struct{})
    defer close(ch)

    go func() {
        defer wg.Done()
        wg.Wait()
        close(ch)
    }()

    timeout := 5*time.Second

    select {
    case <-ch:
        fmt.Println("All jobs completed within timeout.")
    case <-time.After(timeout):
        fmt.Println("Timeout reached, not all jobs completed.")
    }
}

도우미 기능으로 시간 초과 단순화

보다 편리하고 재사용 가능한 접근 방식을 제공하려면, 이 기능을 캡슐화하는 도우미 함수 생성을 고려해보세요.

func waitTimeout(wg *sync.WaitGroup, timeout time.Duration) bool {
    c := make(chan struct{})
    go func() {
        defer close(c)
        wg.Wait()
    }()
    select {
    case <-c:
        return false
    case <-time.After(timeout):
        return true
    }
}

이 도우미 함수는 다음을 허용합니다. WaitGroup 및 원하는 제한 시간을 지정하고 대기가 지정된 제한을 초과했는지 여부를 반환합니다.

도우미 기능 사용

도우미 기능 활용은 간단합니다.

if waitTimeout(&wg, 5*time.Second) {
    fmt.Println("Timeout reached waiting for wait group.")
}

이 예에서 waitTimeout 함수는 대기 그룹이 완료되는 데 5초보다 오래 걸리면 true를 반환합니다. 시간 초과가 발생했습니다.

위 내용은 Go에서 WaitGroup.Wait()에 시간 초과를 추가하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!