如何让 golang mysql 驱动程序在 2 秒内超时 ping？

Question

我在弄清楚如何在 Go 中正确设置数据库连接尝试超时时遇到了一些麻烦。我正在使用这个优秀资源中的一些示例作为基础。我相信我已正确设置所有内容，但我的 ping 只是在 2 秒后拒绝超时。我已经将有问题的代码提取到示例程序中，如下所示。请注意，在 172.1.2.3 上没有运行数据库。package mainimport (    "context"    "database/sql"    _ "github.com/go-sql-driver/mysql" //MySQL driver    "log"    "time")func main() {    log.Print("Trying to ping database!")    //Prepare a "context" to execute queries in, this will allow us to use timeouts    var bgCtx = context.Background()    var ctx2SecondTimeout, cancelFunc2SecondTimeout = context.WithTimeout(bgCtx, time.Second*2)    defer cancelFunc2SecondTimeout()    //Open  database connection    db, err := sql.Open("mysql", "root:@tcp(172.1.2.3)/testdb?parseTime=true")    if err != nil {        log.Printf("Unable to open db, %s", err.Error())        return    }    log.Print("Successfully called open()")    //Ping database connection with 2 second timeout    err = db.PingContext(ctx2SecondTimeout)    if err != nil {        log.Printf("Can't ping database server in order to use storage, %s", err.Error())        return    }    log.Print("Successfully pinged database!")}运行该程序最多需要 2 秒，但实际需要 2 分钟以上：$ go run lambdatry.go2018/09/03 16:33:33 Trying to ping database!2018/09/03 16:33:33 Successfully called open()2018/09/03 16:35:43 Can't ping database server in order to use storage, dial tcp 172.1.2.3:3306: connect: connection timed out如果我更改“数据库”的 IP（我只是选择了一个随机 IP，因此该地址没有数据库）数据库有时会立即超时，有时需要很长时间才能超时。我在 ubuntu 18.04 上运行 go 1.10.1。

Answer 1

看来这不是 的错PingContext。tldr 跳到底部。

PingContext支持带有超时的上下文，但在调试时，似乎原因的根源是调用sql.Open()，或者其他一些调用，然后sql.Conn()是对 after 的调用PingContext。前两个获取db.mu.lock，在它PingContext有机会使用您的超时上下文之前最终等待： https://github.com/golang/go/blob/master/src/database/sql/sql.go#L1394

conn.PingContext当使用instead of 时变得更加清晰db.PingContext，例如对于阻塞的端口，超时是这样发生的：

db, err := sql.Open(driver, connStr)

if err != nil {

    // This error will not happen

    log.Fatalln(err)

}

// Create a 1 second timeout

ctxBG := context.Background()

ctxConnTimeout, cancel := context.WithTimeout(ctxBG, 1*time.Second)

defer cancel()

conn, err := db.Conn(ctxConnTimeout) // Hangs here for 60 seconds, never uses my timeout

if err != nil {

        // This error fires

    log.Fatalln(err)

}

// Never got here

ctxPingTimeout, cancel := context.WithTimeout(ctxBG, 1*time.Second)

defer cancel()

err = conn.PingContext(ctxPingTimeout)

if err != nil {

    log.Fatalln(err)

}

那么如何将超时传递给sql.Openorsql.Conn呢？似乎这是唯一的方法：

sql.Open("mysql", "user:password@/dbname?timeout=5s")

sql.Open("postgres", "user=user dbname=dbname connect_timeout=5")

sql.Open("sqlserver", "sqlserver://username:password@host/instance?dial+timeout=5")

Answer 2

我使用以下 MySQL DSN 格式在 2 秒内超时，它工作得很好。

user:password@tcp(localhost:80)/dbname?timeout=2s