golang定时器实现

拼搏现实的明天。 2023-06-27 12:59 34阅读 0赞

golang定时器，需求是能够在指定时间后停止。或某种条件出发停止。

> [https://golang.google.cn/pkg/time/\#example\_Tick][https_golang.google.cn_pkg_time_example_Tick]

好，下面上货。

package main
    
    import (
    	"context"
    	"fmt"
    	"sync"
    	"time"
    )
    
    func main() {
    	fmt.Println(123)
    	go goClock()
    	go goClock2()
    	stopTheWorld()
    }
    
    func stopTheWorld() {
    	var wg sync.WaitGroup
    	wg.Add(1)
    	wg.Wait()
    }
    
    func goClock() {
    	ticker := time.NewTicker(time.Second)
    	defer ticker.Stop()
    	done := make(chan bool)
    	go func() {
    		time.Sleep(10 * time.Second)
    		done <- true
    	}()
    	for {
    		select {
    		case <-done:
    			fmt.Println("Done!")
    			return
    		case t := <-ticker.C:
    			fmt.Println("clock Current time: ", t)
    		}
    	}
    }
    
    func goClock2() {
    	ticker := time.NewTicker(time.Second)
    	defer ticker.Stop()
    	rootCtx := context.Background()
    	ctx, cancelFunc := context.WithTimeout(rootCtx, 10*time.Second)
    	for {
    		select {
    		case <-ctx.Done():
    			fmt.Println("Done!")
    			cancelFunc()
    			return
    		case t := <-ticker.C:
    			//可以在这里调用停止函数cancelFunc,能够实现指定条件停止定时器的目的
    			fmt.Println("clock2 Current time: ", t)
    		}
    	}
    }

运行结果：

123
    clock2 Current time:  2020-01-07 10:46:19.122178 +0800 CST m=+1.003391336
    clock Current time:  2020-01-07 10:46:19.122301 +0800 CST m=+1.003513931
    clock2 Current time:  2020-01-07 10:46:20.121693 +0800 CST m=+2.002832908
    clock Current time:  2020-01-07 10:46:20.121726 +0800 CST m=+2.002866446
    clock2 Current time:  2020-01-07 10:46:21.1248 +0800 CST m=+3.005867116
    clock Current time:  2020-01-07 10:46:21.1248 +0800 CST m=+3.005866846
    clock2 Current time:  2020-01-07 10:46:22.1235 +0800 CST m=+4.004567391
    clock Current time:  2020-01-07 10:46:22.123503 +0800 CST m=+4.004569986
    clock2 Current time:  2020-01-07 10:46:23.119886 +0800 CST m=+5.000879581
    clock Current time:  2020-01-07 10:46:23.119885 +0800 CST m=+5.000878736
    clock2 Current time:  2020-01-07 10:46:24.119906 +0800 CST m=+6.000899833
    clock Current time:  2020-01-07 10:46:24.119945 +0800 CST m=+6.000938638
    clock Current time:  2020-01-07 10:46:25.121888 +0800 CST m=+7.002808593
    clock2 Current time:  2020-01-07 10:46:25.121927 +0800 CST m=+7.002847958
    clock Current time:  2020-01-07 10:46:26.119995 +0800 CST m=+8.000842811
    clock2 Current time:  2020-01-07 10:46:26.119992 +0800 CST m=+8.000840493
    clock2 Current time:  2020-01-07 10:46:27.120127 +0800 CST m=+9.000975136
    clock Current time:  2020-01-07 10:46:27.120108 +0800 CST m=+9.000956121
    clock Current time:  2020-01-07 10:46:28.122912 +0800 CST m=+10.003686127
    Done!
    clock2 Current time:  2020-01-07 10:46:28.122912 +0800 CST m=+10.003686386
    Done!

简单总结一下，个人推荐使用context，因为能够更加方便控制定时器的停止时间，同时还可以在每次执行定时器业务逻辑的时候进行判断是否达到定时器的停止条件，从而停止计时器。

[https_golang.google.cn_pkg_time_example_Tick]: https://golang.google.cn/pkg/time/#example_Tick