时间戳

首页 > 解决方案 > 继续在 Golang 中重试一个函数

go - 继续在 Golang 中重试一个函数

问题描述

我正在尝试制作一个可以按以下方式工作的功能：

  1. 一旦调用服务函数，它就使用 Fetch 函数从服务中获取记录（以字节数组的形式出现），JSON 解组字节数组，填充结构，然后将结构发送到 DB 函数保存到数据库。
  2. 现在，由于这需要是一个连续的工作，我添加了两个 if 条件，如果接收到的记录长度为 0，那么我们使用 retry 函数重试提取记录，否则我们只是写入数据库。

我一直在尝试调试重试功能一段时间，但它只是不起作用，并且在第一次重试后基本上停止（即使我将尝试指定为 100）。我能做些什么来确保它不断重试提取记录？

代码如下：

// RETRY FUNCTION
func retry(attempts int, sleep time.Duration, f func() error) (err error) {
for i := 0; ; i++ {
    err = f()
    if err == nil {
        return
    }

    if i >= (attempts - 1) {
        break
    }

    time.Sleep(sleep)
    sleep *= 2

    log.Println("retrying after error:", err)
}
return fmt.Errorf("after %d attempts, last error: %s", attempts, err) }


//Save Data function 

type Records struct {
Messages [][]byte
}

func (s *Service) SaveData(records Records, lastSentPlace uint) error {

//lastSentPlace is sent as 0 to begin with.
for i := lastSentPlace; i <= records.Place-1; i++ {

    var msg Records
    msg.Unmarshal(records.Messages[i])

    order := MyStruct{
        Fruit:    msg.Fruit,
        Burger:   msg.Burger,
        Fries:    msg.Fries,
     }

    err := s.db.UpdateOrder(context.TODO(), nil , order)
    if err != nil {
        logging.Error("Error occured...")
    }
    
}return nil}



//Service function (This runs as a batch, which is why we need retrying)

func (s *Service) MyServiceFunction(ctx context.Context, place uint, length uint) (err error) {

var lastSentPlace = place

records, err := s.Poll(context.Background(), place, length)
if err != nil {
    logging.Info(err)
}

// if no records found then retry.
if len(records.Messages) == 0 {
    
    err = retry(100, 2*time.Minute, func() (err error) {
        records, err := s.Poll(context.Background(), place, length)
        
        // if data received, write to DB
        if len(records.Messages) != 0 {
            err = s.SaveData(records, lastSentPlace)
        }
        return
    })
    // if data is not received, or if err is not null, retry
    if err != nil || len(records.Messages) == 0 {
        log.Println(err)
        return
    }
// if data received on first try, then no need to retry, write to db 
} else if len(records.Messages) >0 {
    err = s.SaveData(records, lastSentPlace)
    if err != nil {
        return err
    }
}

return nil }

我认为，问题出在我尝试实现重试功能的方式上，我一直在尝试调试它一段时间，但是对于语言来说是新手，我真的被卡住了。我想做的是，如果没有找到记录，则实施退避。任何帮助是极大的赞赏。

谢谢 ！！！

标签： goretry-logicexponential-backoff

解决方案

我做了一个更简单的重试。

  • 使用更简单的 for 循环逻辑来确保正确性。
  • 我们在执行重试之前休眠，因此i > 0用作休眠的条件。

这是代码：

func retry(attempts int, sleep time.Duration, f func() error) (err error) {
    for i := 0; i < attempts; i++ {
        if i > 0 {
            log.Println("retrying after error:", err)
            time.Sleep(sleep)
            sleep *= 2
        }
        err = f()
        if err == nil {
            return nil
        }
    }
    return fmt.Errorf("after %d attempts, last error: %s", attempts, err)
}

推荐阅读