如果连接中断，如何让Go中的net.Dial重新连接？

import (
    "fmt"
    "net"
    "time"
)

func connect() (net.Conn, error) {
    conn, err := net.Dial("tcp", "服务器地址:端口号")
    if err != nil {
        return nil, err
    }
    return conn, nil
}

func main() {
    for {
        conn, err := connect()
        if err != nil {
            fmt.Println("连接失败:", err)
            time.Sleep(5 * time.Second) // 等待5秒后重新连接
            continue
        }
        defer conn.Close()

        // 连接成功后的操作
        // ...
    }
}