Go语言：调度间隔用哪个好？time.Sleep v.s. time.After

Go编程中循环调度任务的执行间隔我们通常采用 time.Sleep或time.After来实现。 写法1: ``` go func(ctx context.Context) { for { select { case <-ctx.Done(): fmt.Println("Cancelled", time.Now()) wg.Done() return default: time.Sleep(time.Second * 10) fmt.Println("Invoked", time.Now()) } } }(ctx) ``` 写法2: ``` go func(ctx context.Context) { for { select { case <-ctx.Done(): fmt.Println("Cancelled", time.Now()) wg.Done() return case <-time.After(time.Second * 10): fmt.Println("Invoked", time.Now()) } } }(ctx) ``` 这两种方式一样吗？那种方式更好呢？ 以上两种写法，都可以满足需求。那么是否这两种写法是等效的呢？ 当然不是，为了更好的说明它们的差别，请运行下面的这两段测试代码： ``` var wg sync.WaitGroup func cancelTaskAfter(interval time.Duration, cancel context.CancelFunc) { go func(cancel context.CancelFunc) { time.Sleep(interval) fmt.Println("Cancell task", time.Now()) cancel() wg.Done() }(cancel) } func TestTimerTask1(t *testing.T) { ctx, cancel := context.WithCancel(context.Background()) wg.Add(1) cancelTaskAfter(time.Second*25, cancel) wg.Add(1) go func(ctx context.Context) { for { select { case <-ctx.Done(): fmt.Println("Cancelled", time.Now()) wg.Done() return default: time.Sleep(time.Second * 10) fmt.Println("Invoked", time.Now()) } } }(ctx) wg.Wait() } func TestTimerTask2(t *testing.T) { ctx, cancel := context.WithCancel(context.Background()) wg.Add(1) cancelTaskAfter(time.Second*25, cancel) wg.Add(1) go func(ctx context.Context) { for { select { case <-ctx.Done(): fmt.Println("Cancelled", time.Now()) wg.Done() return case <-time.After(time.Second * 10): fmt.Println("Invoked", time.Now()) } } }(ctx) wg.Wait() } ``` 也许看到这，你大概已经知道这两种写法的不同了。 以下是输出结果 ``` === RUN TestTimerTask1 Invoked 2019-09-22 17:12:07.055392 +0800 CST m=+35.007752073 Invoked 2019-09-22 17:12:17.057073 +0800 CST m=+45.009375450 Cancell task 2019-09-22 17:12:22.051311 +0800 CST m=+50.003583680 Invoked 2019-09-22 17:12:27.060766 +0800 CST m=+55.013010382 Cancelled 2019-09-22 17:12:27.060804 +0800 CST m=+55.013048342 --- PASS: TestTimerTask1 (30.01s) ``` ``` === RUN TestTimerTask2 Invoked 2019-09-22 17:11:42.049262 +0800 CST m=+10.001766181 Invoked 2019-09-22 17:11:52.053054 +0800 CST m=+20.005500478 Cancell task 2019-09-22 17:11:57.050145 +0800 CST m=+25.002563035 Cancelled 2019-09-22 17:11:57.05041 +0800 CST m=+25.002827648 --- PASS: TestTimerTask2 (25.00s) ``` 采用Sleep来实现间隔的时候，如果cancel调用后任务协程正好处于sleep过程中，这时任务是无法被及时取消的。 小结 如果你需要任务能够被及时的取消，那你应该采用time.After来控制调用的间隔。 更多知识，欢迎订阅学习 《Go 语言入门到实战》