最有效的超时设置方法

在需要等待特定时间才能上线的异步部署服务中，设置超时机制至关重要。在 Go 语言中，使用 context.WithTimeout 可创建带有取消功能的轮询等待。将较大的超时设置在输入上下文中，并在服务等待函数中使用 select 语句在指定轮询间隔内等待服务上线。若超过指定时间仍无法找到服务，则返回错误。

我有一个异步部署的服务，我需要等待指定的时间才能使其上线。如果指定的时间过去了，我们仍然无法找到该服务，那么我们就会出错。在 go 中编写这个的最佳方式是什么？我正在考虑使用 context.withtimeout 但不确定这到底是如何工作的。感谢您的帮助！

func (c *Client) WaitForServiceToComeAlive(ctx context.Context, name string, timeout time.Duration) error {
    
    var mysvc *Service
    var err error

    endtime := time.Now().Add(timeout)

    for time.Now().Before(endtime) {
        mysvc, err = c.GetService(ctx, name)
        if err != nil {
            return err
        }

        if mysvc != nil {
            break
        }

        time.Sleep(time.Second * 10)
    }

    if mysvc == nil {
        return fmt.Errorf("svc %s did not register", name)
    }

    return nil
}

正确答案

切勿使用 time.sleep，尤其是在较长时期内 - 因为它是不间断的。为什么这很重要？如果它在一个 goroutine 中并且该任务不会完成（即上下文被取消），那么您宁愿立即中止。

因此，要创建一个带有取消的轮询等待：

select {
case <-ctx.done():               // cancel early if context is canceled
    return ctx.err()
case <-time.after(pollinterval): // wait for pollinterval duration
}

将较大的超时设置在输入上下文中：

ctx := context.todo() // <- outer request context goes here or context.background()

// wrap context with a timeout
ctx, cancel := context.withtimeout(ctx, 1 * time.minute)
defer cancel() // avoid leaks

err := c.waitforservicetocomealive(ctx, "job", 10*time.second /* poll interval */)

那么你的服务等待函数就简化为：

func (c *Client) WaitForServiceToComeAlive(ctx context.Context, name string, pollInterval time.Duration) error {

    var mysvc *Service
    var err error

    for {
        mysvc, err = c.GetService(name) // <- this should take a ctx too - if possible for early cancelation
        if err != nil {
            return err
        }
        if mysvc != nil {
            return nil
        }

        select {
        case <-ctx.Done():
            return ctx.Err()
        case <-time.After(pollInterval):
        }
    }

}

https://play.golang.org/p/JwH5CMyY0I2

理论要掌握，实操不能落！以上关于《最有效的超时设置方法》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！