Go에서 WaitGroup.Wait()에 대한 시간 초과 대기를 구현하는 방법은 무엇입니까?

WaitGroup.Wait()에 대한 Timed Wait()

Go에서 WaitGroup 유형은 모두가 작업을 완료할 때까지 여러 고루틴을 동기화하는 데 사용됩니다. . WaitGroup.Wait() 메서드는 그룹의 모든 구성원이 완료 신호를 보내거나 컨텍스트가 취소될 때까지 차단됩니다.

그러나 때로는 스케줄러가 중단되는 것을 방지하기 위해 WaitGroup 대기 작업에 시간 초과를 적용하는 것이 바람직할 때도 있습니다. 잘못된 작업자를 무기한 기다리지 않고. 이는 이러한 오류가 있는 경우 시스템의 신뢰성과 복구 가능성에 대한 철학적 질문을 제기하지만 이 논의는 이 질문의 범위를 벗어납니다.

해결책

제공된 솔루션은 채널과 select 문을 활용하여 시간 초과 기능을 구현합니다.

c := make(chan struct{})
go func() {
    defer wg.Done()
    // Do the work
    c <- struct{}{}
}()
select {
case <-c:
    // Work completed
case <-time.After(timeout):
    // Timed out
}

이 솔루션은 효과적이지만 지나치게 복잡해 보일 수도 있습니다. 다음 팁을 사용하면 이를 개선할 수 있습니다.

• 채널에 값을 전송하는 대신 신호 완료를 위해 채널 폐쇄를 사용하는 것이 좋습니다. 닫힌 채널에서 읽기는 항상 즉시 진행됩니다.
• 완료 신호를 보내려면 defer 문을 사용하세요. 함수가 갑자기 종료되어도 실행됩니다.
• 대기할 작업이 하나만 있으면 WaitGroup을 생략하고 채널을 직접 사용하세요.
• 시간 초과의 경우 기간을 직접 지정하세요. 예: timeout := time.Second(또는 2초의 경우 2 * time.Second) timeout).

도우미 함수

프로세스를 단순화하기 위해 시간 초과 논리를 캡슐화하는 도우미 함수를 만들 수 있습니다.

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 // Completed normally
    case <-time.After(timeout):
        return true // Timed out
    }
}

이 기능은 시간 초과 기능을 래핑하여 쉽게 사용:

if waitTimeout(&wg, time.Second) {
    fmt.Println("Timed out waiting for wait group")
} else {
    fmt.Println("Wait group finished")
}

이 접근 방식은 WaitGroup.Wait() 작업의 시간 초과를 처리하는 더 깔끔하고 편리한 방법을 제공합니다.

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