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
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容