优雅的关闭ticker

文章目录

1. 1. 优雅的关闭ticker

在ticker和goroutine组合使用中当调用ticker.Stop()关闭ticker之后，相应的goroutine中的ticker.C并没有停掉，下面总结一种优雅的方式关闭ticker.

优雅的关闭ticker

package main
import (
	"fmt"
	"time"
)

func main() {
	Demo()
}

func Demo() {
	done := startTicker(PrintInfo)
	time.Sleep(time.Duration(12) * time.Second)
	close(done)
	time.Sleep(time.Duration(1) * time.Minute)
	fmt.Println("main finished")
}

func startTicker(f func()) chan struct{} {
	done := make(chan struct{}, 1)
	go func() {
		ticker := time.NewTicker(5 * time.Second)
		//	ticker := time.NewTicker(5 * time.Minute)
		defer ticker.Stop()

		for {
			select {
			case <-ticker.C:
				f()
			case <-done:
				return
			}
		}
	}()
	return done
}

func PrintInfo() {
	fmt.Println("hello")
}

上述代码通过startTicker创建一个ticker, 当想要关闭这个ticker并同时退出对应goroutine中的for时，可以直接close()，发送done信号直接返回退出startTicker即可.