在Golang开发过程中,尤其是在处理涉及长时间运行、循环迭代或者并发任务时,为函数中的for循环设定合理的时间限制至关重要。 这不仅可以确保程序在预设时间内完成特定任务,避免资源浪费,还能提升系统的稳定性,健壮性以及可控性。

基本的方法就是使用 time.Timer 与 select

利用time包中的time.After函数创建一个定时器(Timer),它会在指定的延迟时间后发送一个信号到其关联的通道。 在for循环中,通过select语句监听这个通道和循环条件,一旦定时器触发,select会选择接收定时器通道的数据,从而跳出循环。

例如我有一个执行 cli 上传文件的 shell 任务,这个任务最大运行时间不能超过3分钟,超过3分钟则自动结束掉:

func qiniuUploadWithTimeout() bool {
	timeout := time.After(3 * time.Minute)
	for {
		select {
		case <-timeout:
		   //上报超时重试队列
			log.Fatal("cmd.Run() Timed out\n")
			return false
		default:
			cmd := "/data/bin/qshell rput my-buckets xxx.mp4 /data/wwwroot/storage/xxx.mp4"
			cmdContainer := exec.Command("bash", "-c", cmd)
			cmdContainer.Env = os.Environ()
			cmdContainer.Stdout = os.Stdout
			cmdContainer.Stderr = os.Stderr
			err := cmdContainer.Run()
			if err != nil {
				log.Fatalf("cmd.Run() failed with '%s'\n", err)
				return false
			}

			return true
		}
	}
}

上述代码中,time.After(3 * time.Second)会在3秒后向通道发送一个信号。 select语句会持续监控这个通道和默认分支(即执行循环体)。当3秒过去,通道接收到信号时,select选择相应case,打印“Timed out”并退出。

总结

通过使用time.After和select语句,我们可以为Go语言中的无限循环添加时间限制,从而避免程序在意外情况下无限制地运行下去。这不仅保证了程序的效率,也提高了其可维护性和稳定性。同时,注意goroutine泄露和panic处理,可以进一步提高程序的健壮性。