Golang中TCP I/O超时及无法建立TCP连接的问题排查

在您的场景中，TCP I/O超时和连接建立问题通常与数据库连接池配置、网络环境或数据库服务器负载有关。以下是一些具体的排查步骤和代码示例：

1. 优化数据库连接池配置

func initDB() (*sql.DB, error) {
    connString := "server=10.0.222.8;port=1433;database=yourdb;user id=user;password=pass"
    db, err := sql.Open("sqlserver", connString)
    if err != nil {
        return nil, err
    }
    
    // 根据您的负载调整这些参数
    db.SetMaxOpenConns(100)           // 最大打开连接数
    db.SetMaxIdleConns(20)            // 最大空闲连接数
    db.SetConnMaxLifetime(30 * time.Minute) // 连接最大生命周期
    db.SetConnMaxIdleTime(10 * time.Minute) // 连接最大空闲时间
    
    // 验证连接
    if err := db.Ping(); err != nil {
        return nil, err
    }
    
    return db, nil
}

2. 实现带超时的上下文控制

func DgsDoLineChange(ctx context.Context, params ...interface{}) error {
    // 设置查询超时
    queryCtx, cancel := context.WithTimeout(ctx, 30*time.Second)
    defer cancel()
    
    err := DoLineChangePrep.QueryRowContext(queryCtx, params...).Scan(&result)
    if err != nil {
        if errors.Is(err, context.DeadlineExceeded) {
            log.Printf("Query timeout: %v", err)
        }
        return err
    }
    return nil
}

3. 添加连接健康检查

func checkDBHealth(db *sql.DB) bool {
    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
    defer cancel()
    
    if err := db.PingContext(ctx); err != nil {
        log.Printf("Database health check failed: %v", err)
        return false
    }
    return true
}

// 定期执行健康检查
func startHealthCheck(db *sql.DB) {
    ticker := time.NewTicker(1 * time.Minute)
    defer ticker.Stop()
    
    for range ticker.C {
        if !checkDBHealth(db) {
            log.Println("Database connection unhealthy")
        }
    }
}

4. 实现重试机制

func executeWithRetry(ctx context.Context, maxRetries int, fn func() error) error {
    var lastErr error
    
    for i := 0; i < maxRetries; i++ {
        select {
        case <-ctx.Done():
            return ctx.Err()
        default:
        }
        
        if err := fn(); err != nil {
            lastErr = err
            if isRetryableError(err) {
                time.Sleep(time.Duration(i+1) * time.Second) // 指数退避
                continue
            }
            return err
        }
        return nil
    }
    return lastErr
}

func isRetryableError(err error) bool {
    // 检查是否为可重试的错误类型
    if err == nil {
        return false
    }
    
    errorStr := err.Error()
    return strings.Contains(errorStr, "i/o timeout") ||
           strings.Contains(errorStr, "connection reset") ||
           strings.Contains(errorStr, "broken pipe")
}

5. 监控连接池状态

func monitorConnectionPool(db *sql.DB) {
    stats := db.Stats()
    log.Printf("Connection Pool Stats - Open: %d, InUse: %d, Idle: %d, WaitCount: %d, WaitDuration: %v",
        stats.OpenConnections,
        stats.InUse,
        stats.Idle,
        stats.WaitCount,
        stats.WaitDuration)
}

// 定期记录连接池状态
go func() {
    ticker := time.NewTicker(30 * time.Second)
    defer ticker.Stop()
    for range ticker.C {
        monitorConnectionPool(dgsData)
    }
}()

6. 网络层面排查

检查网络配置和防火墙规则：

func testNetworkConnectivity() {
    timeout := 5 * time.Second
    conn, err := net.DialTimeout("tcp", "10.0.222.8:1433", timeout)
    if err != nil {
        log.Printf("Network connectivity test failed: %v", err)
        return
    }
    defer conn.Close()
    log.Println("Network connectivity test passed")
}

这些措施应该能帮助您识别和解决TCP连接问题。重点关注连接池配置、超时设置和重试逻辑，这些通常是导致I/O超时的主要原因。