Golang 语言如何为for循环设置超时终止操作?

在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处理,可以进一步提高程序的健壮性。

  • 12
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值