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

Question

我在弄清楚如何在 Go 中正确设置数据库连接尝试超时时遇到了一些麻烦。我正在使用这个优秀资源中的一些示例作为基础。我相信我已正确设置所有内容，但我的 ping 只是在 2 秒后拒绝超时。我已经将有问题的代码提取到示例程序中，如下所示。请注意，在 172.1.2.3 上没有运行数据库。

package main

import (

    "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.go

2018/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#L1394conn.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 timeoutctxBG := 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 timeoutif err != nil {        // This error fires    log.Fatalln(err)}// Never got herectxPingTimeout, 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