Go语言Token定时轮询控制台应用问题:到期不刷新+Ctrl+C退出
解决Token定时轮询与刷新的问题
你的核心问题出在每次select循环都会重新创建time.After()定时器,导致原本应该等待到期的定时器被不断重置,永远触发不了刷新逻辑。另外,done通道在这里确实是多余的,我们可以简化逻辑,同时正确监听定时器和中断信号。
现有代码的关键问题分析
time.After()的误用:在select的每个case里使用time.After(),每次循环都会生成新的定时器。比如你设置的<-time.After(exp),每次循环都会创建一个新的、等待exp时长的通道,而之前的定时器会被GC回收,永远等不到它触发。- 倒计时逻辑的并发问题:用goroutine打印剩余时间虽然没问题,但结合上面的定时器问题,会导致打印逻辑频繁触发,而且和到期逻辑冲突。
- 信号监听的位置错误:把信号监听放在
poller函数里,每次调用poller都会重新注册信号,虽然不会有大问题,但更合理的是在main里只注册一次。
修正后的实现方案
我们可以用time.NewTicker来实现定期打印倒计时,用time.NewTimer来等待Token到期,同时在main里统一监听中断信号,这样逻辑更清晰,也能避免定时器被重复创建的问题。
package main import ( "fmt" "math" "os" "os/signal" "syscall" "time" ) // 模拟获取Token的HTTP请求,返回下次刷新时间 func refreshToken() (time.Time, error) { fmt.Println("\n正在刷新Token...") time.Sleep(2 * time.Second) // 模拟网络请求耗时 nextRefresh := time.Now().Add(10 * time.Second) fmt.Printf("Token刷新完成,下次刷新时间:%v\n", nextRefresh.Format(time.RFC3339)) return nextRefresh, nil } // 计算距离到期的剩余时间(向下取整到秒) func timeLeft(d time.Time) time.Duration { exactLeft := d.Sub(time.Now()) if exactLeft <= 0 { return 0 } floorSeconds := math.Floor(exactLeft.Seconds()) return time.Duration(floorSeconds) * time.Second } // 轮询逻辑:定期打印剩余时间,直到Token到期 func poller(expDate time.Time) { // 创建倒计时定时器:到点触发刷新 expireTimer := time.NewTimer(timeLeft(expDate)) defer expireTimer.Stop() // 创建定期打印的Ticker:每3秒更新一次剩余时间 printTicker := time.NewTicker(3 * time.Second) defer printTicker.Stop() for { select { // 定期打印剩余时间 case <-printTicker.C: left := timeLeft(expDate) fmt.Printf("\r下次Token刷新剩余时间:%v", left) // 如果剩余时间为0,手动触发定时器(避免因为时间计算误差导致延迟) if left <= 0 { if !expireTimer.Stop() { <-expireTimer.C } expireTimer.Reset(0) } // Token到期,退出轮询回到main刷新 case <-expireTimer.C: fmt.Println("\nToken已到期,准备刷新...") return } } } func main() { // 监听中断信号(Ctrl+C) sigChan := make(chan os.Signal, 1) signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM) go func() { <-sigChan fmt.Println("\n用户终止程序,正在退出...") os.Exit(0) }() var expiration time.Time var err error // 首次获取Token expiration, err = refreshToken() if err != nil { panic(fmt.Errorf("首次获取Token失败:%v", err)) } // 持续循环:轮询等待到期 -> 刷新Token -> 再次轮询 for { poller(expiration) expiration, err = refreshToken() if err != nil { panic(fmt.Errorf("刷新Token失败:%v", err)) } } }
关键改进点说明
- 使用
time.NewTimer和time.NewTicker:Timer用于单次触发的到期事件,只会触发一次,避免重复创建。Ticker用于定期重复的打印逻辑,稳定触发,不会像time.After那样每次循环生成新定时器。
- 统一信号监听:在main函数里只注册一次中断信号监听,用goroutine处理,避免每次调用
poller都重复注册。 - 移除多余的
done通道:poller函数在到期时直接return,回到main循环执行刷新逻辑,逻辑更简洁。 - 时间误差处理:在打印剩余时间时,如果发现剩余时间为0,手动重置定时器,避免因为时间计算的微小误差导致刷新延迟。
- 输出优化:用
\r实现同一行刷新剩余时间,让控制台输出更整洁。
这样修改后,程序就能稳定地定期显示剩余时间,到期自动触发Token刷新,同时响应Ctrl+C正常退出。
内容的提问来源于stack exchange,提问作者silverfighter
相关产品推荐
相关产品推荐

