• 设为首页
  • 点击收藏
  • 手机版
    手机扫一扫访问
    迪恩网络手机版
  • 关注官方公众号
    微信扫一扫关注
    迪恩网络公众号

Go语言通用重试函数封装

原作者: [db:作者] 来自: [db:来源] 收藏 邀请

以下实现了两个方法,分别是

  1. 设定最大重试次数,对一个方法进行重试
  2. 设定最大重试时间,对一个方法进行重试

//重试,限制次数
func RetryTimes(name string, tryTimes int, sleep time.Duration, callback func() error) (err error) {
	for i := 1; i <= tryTimes; i++ {
		err = callback()
		if err == nil {
			return nil
		}
		fmt.Printf("[%v]失败,第%v次重试, 错误信息:%s \n", name, i, err)
		time.Sleep(sleep)
	}
	err = fmt.Errorf("[%v]失败,共重试%d次, 最近一次错误:%s \n", name, tryTimes, err)
	fmt.Println(err)
	return err

}

//重试,限制时间
func RetryDurations(name string, max time.Duration, sleep time.Duration, callback func() error) (err error) {
	t0 := time.Now()
	i := 0
	for {
		err = callback()
		if err == nil {
			return
		}
		delta := time.Now().Sub(t0)
		if delta > max {
			fmt.Printf("[%v]失败,超过最大时间%s, 共重试%d次,最近一次错误: %s \n", name, max, i, err)
			return err
		}
		time.Sleep(sleep)
		i++
		fmt.Printf("[%v]失败,第%v次重试, 错误信息:%s \n", name, i, err)
	}
}

使用

func main() {
	//示例任务函数,只会失败
	var task = func() error {
		return errors.New("task error")
	}

	fmt.Println("RetryTimes")
	//设定3秒最大重试次数,最大3次,1秒重试一次
	RetryTimes("测试任务", 3, time.Second, task)

	fmt.Println("RetryDurations")
	//设定3秒最大重试期限,最多3秒,1秒重试一次
	RetryDurations("测试任务", 3*time.Second, time.Second, task)
}

运行结果如下

就是这么简单粗暴,light weight baby~


鲜花

握手

雷人

路过

鸡蛋
该文章已有0人参与评论

请发表评论

全部评论

专题导读
上一篇:
go框架macaron的使用发布时间:2022-07-10
下一篇:
Go语言学习笔记(十五)之异常处理发布时间:2022-07-10
热门推荐
热门话题
阅读排行榜

扫描微信二维码

查看手机版网站

随时了解更新最新资讯

139-2527-9053

在线客服(服务时间 9:00~18:00)

在线QQ客服
地址:深圳市南山区西丽大学城创智工业园
电邮:jeky_zhao#qq.com
移动电话:139-2527-9053

Powered by 互联科技 X3.4© 2001-2213 极客世界.|Sitemap