在 Go 中为 SQL 连接设置 TCP 超时

Question

当我使用 VPN 连接到数据库（使用标准 go sql 库）并且 VPN 接口关闭时，无论接口是否同时启动，当我尝试执行 SQL 查询时都会超时 75 秒。我想将此超时减少到某个合理的时间，因此在这种情况下，我的应用程序不会冻结 75 秒。db, err := sql.Open(driverName, dataSourceName)是否可以通过db变量以某种方式设置它？

Answer 1

该database/sql包没有提供一种通用的方法来超时调用database/sql.Open. 但是，各个驱动程序通过 DSN (dataSourceName) 连接字符串提供此功能。

https://github.com/lib/pq

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

https://github.com/go-sql-driver/mysql

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

https://github.com/denisenkom/go-mssqldb

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

ETC ...

Answer 2

从 Go 1.8 开始，sql.DB抽象现在接受context.Context，可用于更快地超时连接。

func (c *Client) DoLookup(ctx context.Context, id int) (string, error) {

  var name string

  // create a child context with a timeout

  newCtx, cancel := context.WithTimeout(ctx, time.Second)

  // release resources used in `newCtx` if

  // the DB operation finishes faster than the timeout

  defer cancel()

  row := c.db.QueryRowContext(newCtx, "SELECT name FROM items WHERE id = ?", id)

  err := row.Scan(&name)

  if err != nil {

    return "", err

  }

  return name, nil

}

如果你的DoLookup函数还没有使用context.Context（它真的应该！）你可以通过调用来创建一个父函数context.TODO()。