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

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

Susan Sarandon

풀어 주다： 2024-11-12 17:17:01

원래의

948명이 탐색했습니다.

How to Implement a Timeout for WaitGroup.Wait() in Go?

WaitGroup.Wait()의 시간 초과: 관용적 접근 방식

Go에서는 잘못된 작업자에 대한 무한 대기를 방지하기 위해 WaitGroup.Wait()에 시간 제한을 설정할 수 있습니다. . 다음은 이를 구현하는 관용적 방법과 대체 접근 방식에 대한 자세한 탐색입니다.

권장 솔루션: 타이머 및 채널 사용

다음 접근 방식은 가장 관용적인 솔루션으로 널리 간주됩니다.

import (
    "sync"
    "time"
)

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

로그인 후 복사

대기 그룹이 완료되면 신호를 보내는 채널(c)이 생성됩니다. wait.
wg.Wait()를 호출하고 완료되면 채널을 닫는 goroutine이 생성됩니다.
select 문은 채널이 닫히거나 시간 초과가 발생하는지 수신 대기합니다.
채널이 닫히면 함수는 false를 반환하여 정상 완료를 나타냅니다.
타임아웃에 도달하면 함수는 반환됩니다. true, 시간 초과를 나타냅니다.

권장 솔루션 단순화

더 간단한 구현을 위해 다음을 고려하십시오.

대신 신호 완료를 위해 채널을 닫습니다.
함수가 종료되더라도 채널을 닫으려면 defer를 사용하세요.
하나의 작업만 기다려야 하는 경우 대기 그룹을 제거하세요.
변환 없이 직접 기간을 사용하세요(예: time.Second).

대체 접근 방식: 취소 가능한 컨텍스트 사용

대체 접근 방식은 취소 가능한 컨텍스트를 사용하는 것입니다. context:

import (
    "context"
    "sync/atomic"
    "sync"
)

func waitWithCancel(ctx context.Context, wg *sync.WaitGroup) bool {
    ctxDone := make(chan struct{})
    var exited int32
    go func() {
        defer close(ctxDone)
        wg.Wait()
        atomic.StoreInt32(&exited, 1)
    }()

    select {
    case <-ctx.Done():
        return atomic.LoadInt32(&exited) == 0
    case <-ctxDone:
        return true
    }
}

로그인 후 복사