go :如何优雅地超时退出

使用wait.Poll优雅控制程序执行

最新推荐文章于 2025-08-14 12:11:56 发布

原创最新推荐文章于 2025-08-14 12:11:56 发布 · 731 阅读

0 ·

CC 4.0 BY-SA版权

go语言编程专栏收录该内容

55 篇文章

订阅专栏

本文介绍了一种在业务代码中利用wait.Poll函数，实现程序在指定时间段内运行并优雅退出的方法。通过一个具体示例，展示了如何设置程序在5秒后超时并检查条件退出，适用于需要定时任务或限制执行时间的场景。

业务代码中，我们怎么让一段程序，在某个时间段内执行，然后优雅退出？
本文给出一种办法：使用wait.poll

package main

import (
	"fmt"
	"time"

	"k8s.io/apimachinery/pkg/util/wait"
	"k8s.io/klog"
)

func main() {
	for {
		a := wait.Poll(time.Second, time.Second*5, func() (done bool, err error) {
			fmt.Println(time.Now())
			return false, nil
		})
		if a != nil {
			klog.Error(a)
			break
		}
	}

}