You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

Go语言Token定时轮询控制台应用问题:到期不刷新+Ctrl+C退出

解决Token定时轮询与刷新的问题

你的核心问题出在每次select循环都会重新创建time.After()定时器,导致原本应该等待到期的定时器被不断重置,永远触发不了刷新逻辑。另外,done通道在这里确实是多余的,我们可以简化逻辑,同时正确监听定时器和中断信号。

现有代码的关键问题分析

  1. time.After()的误用:在select的每个case里使用time.After(),每次循环都会生成新的定时器。比如你设置的<-time.After(exp),每次循环都会创建一个新的、等待exp时长的通道,而之前的定时器会被GC回收,永远等不到它触发。
  2. 倒计时逻辑的并发问题:用goroutine打印剩余时间虽然没问题,但结合上面的定时器问题,会导致打印逻辑频繁触发,而且和到期逻辑冲突。
  3. 信号监听的位置错误:把信号监听放在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))
		}
	}
}

关键改进点说明

  1. 使用time.NewTimer和time.NewTicker:
    • Timer用于单次触发的到期事件,只会触发一次,避免重复创建。
    • Ticker用于定期重复的打印逻辑,稳定触发,不会像time.After那样每次循环生成新定时器。
  2. 统一信号监听:在main函数里只注册一次中断信号监听,用goroutine处理,避免每次调用poller都重复注册。
  3. 移除多余的done通道:poller函数在到期时直接return,回到main循环执行刷新逻辑,逻辑更简洁。
  4. 时间误差处理:在打印剩余时间时,如果发现剩余时间为0,手动重置定时器,避免因为时间计算的微小误差导致刷新延迟。
  5. 输出优化:用\r实现同一行刷新剩余时间,让控制台输出更整洁。

这样修改后,程序就能稳定地定期显示剩余时间,到期自动触发Token刷新,同时响应Ctrl+C正常退出。

内容的提问来源于stack exchange,提问作者silverfighter

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.05.29 07:35:58