如何在goworker中实现Redis连接超时前指定次数重试？

由于github.com/benmanns/goworker本身没有提供连接重试的配置参数，你可以通过自定义Redis连接池的Dial函数来实现指定次数的连接重试逻辑，具体步骤如下：

1. 编写带重试的Dial函数 #

基于redigo（goworker默认依赖的Redis客户端），封装一个支持重试的Dial方法，在连接失败时重复尝试指定次数：

import (
    "fmt"
    "time"

    "github.com/gomodule/redigo/redis"
)

// dialWithRetry 返回一个带重试逻辑的Dial函数
func dialWithRetry(network, addr string, retryCount int) func() (redis.Conn, error) {
    return func() (redis.Conn, error) {
        var conn redis.Conn
        var err error

        for i := 0; i < retryCount; i++ {
            // 尝试建立Redis连接
            conn, err = redis.Dial(network, addr)
            if err == nil {
                return conn, nil
            }
            // 重试前短暂休眠，避免高频重试
            time.Sleep(100 * time.Millisecond)
        }
        // 所有重试失败后返回错误
        return nil, fmt.Errorf("redis connect failed after %d retries: %w", retryCount, err)
    }
}

2. 为goworker配置自定义连接池 #

创建包含重试逻辑的Redis连接池，并将其设置为goworker的默认连接池：

import (
    "log"
    "time"

    "github.com/benmanns/goworker"
    "github.com/gomodule/redigo/redis"
)

func main() {
    // 设置重试次数，根据实际需求调整
    retryTimes := 3

    // 初始化带重试的Redis连接池
    redisPool := &redis.Pool{
        MaxIdle:     5,          // 最大空闲连接数
        MaxActive:   20,         // 最大活跃连接数
        IdleTimeout: 180 * time.Second, // 空闲连接超时时间
        Dial:        dialWithRetry("tcp", "your-redis-addr:6379", retryTimes),
        // 可选：从池取连接时验证有效性，避免拿到失效连接
        TestOnBorrow: func(c redis.Conn, t time.Time) error {
            _, err := c.Do("PING")
            return err
        },
    }

    // 让goworker使用这个自定义连接池
    goworker.SetRedisPool(redisPool)

    // 注册你的任务处理函数
    goworker.Register("your-job-name", yourJobHandler)

    // 启动goworker
    if err := goworker.Work(); err != nil {
        log.Fatalf("goworker start failed: %v", err)
    }
}

// 示例任务处理函数
func yourJobHandler(queue string, args ...interface{}) error {
    // 任务逻辑
    return nil
}

额外优化建议 #

• 可以把固定的重试间隔改成指数退避（比如第一次等100ms，第二次200ms，第三次400ms），避免在Redis短暂不可用时持续冲击服务
• 如果你的goworker版本使用的是go-redis客户端，逻辑类似：自定义客户端的RetryBackoff配置来实现指定次数的重试

内容的提问来源于stack exchange，提问作者HarshM