Golang 超时控制代码模版

func main(){
    ctx, cancelFunc := context.WithTimeout(context.Background(), 10*time.Second)
    defer cancelFunc()
    result := make(chan int, 1)
    errChan := make(chan error, 1)
    fmt.Println("start: ", time.Now().Unix())
    select {
    case <-ctx.Done():
        fmt.Println("timeout")
    case result <- process(errChan):
        if len(errChan) > 0 {
            fmt.Println("stop err: ", <-errChan)
            return
        }
        fmt.Println("normal, result: ", <-result)
    case err := <-errChan:
        fmt.Println("err: ", err.Error())
    }
    fmt.Println("end: ", time.Now().Unix())
}
func process(errChan chan error) int {
    // 此处模拟业务逻辑，一定概率报错返回，一定概率正常返回
    rand.Seed(time.Now().UnixNano())
    n := rand.Intn(10)
    fmt.Println(n)
    if n < 5 {
        // 正常返回
        // 此处模拟业务逻辑代码耗费时间
        time.Sleep(3 * time.Second)
    } else {
        errChan <- errors.New("process error")
    }
    return n
}

有疑问加站长微信联系（非本文作者）