本文介绍golang 中连接MySQL时,如何设置最大连接数和最大空闲连接数。

database/sql
github.com/go-sql-driver/mysql

设置最大连接数的接口是

func (db *DB) SetMaxOpenConns(n int) 

设置连接MySQL可以打开的最大连接数。

如果n <= 0,表示打开的连接数没有限制。

默认为0,也就是不限制连接数。

MaxIdleConns
MaxIdleConnsMaxOpenConnsMaxIdleConnsMaxOpenConns

设置最大空闲连接数的接口是:

func (db *DB) SetMaxIdleConns(n int)

如果n<=0,表示不使用空闲连接池,即一个连接如果不使用,不会放入空闲连接池。因此,这种方式不会复用连接,每次执行SQL语句,都会重新建立新的连接。

const defaultMaxIdleConns = 2

关于打开的连接和空闲的连接之间的关系,补充一下:

打开的连接 = 正在使用的连接(inuse) + 处于空闲状态的连接(idle)

下面对最大连接数和最大空闲连接数做下测试和验证。

首先设置最大打开的连接数为1,接着开启20个goroutine,每个goroutine执行sql语句,打印执行sql使用的连接的connection id。并执行耗时的sql语句占用连接,观察其他需要执行SQL的goroutine的执行情况。

例子代码如下:

package main

import (
        "database/sql"
        "log"

        _ "github.com/go-sql-driver/mysql"
)

var DB *sql.DB
var dataBase = "root:Aa123456@tcp(127.0.0.1:3306)/?loc=Local&parseTime=true"

func Init() {
        var err error
        DB, err = sql.Open("mysql", dataBase)
        if err != nil {
                log.Fatalln("open db fail:", err)
        }

        DB.SetMaxOpenConns(1)

        err = DB.Ping()
        if err != nil {
                log.Fatalln("ping db fail:", err)
        }
}

func main() {
        Init()
        
        //开启20个goroutine
        for i:=0; i < 20; i++ {
                go one_worker(i)
        }
        
        select {
        }

}

func one_worker(i int) {
        var connection_id int
        err := DB.QueryRow("select CONNECTION_ID()").Scan(&connection_id)
        if err != nil {
                log.Println("query connection id failed:", err)
                return
        }

        log.Println("worker:", i, ", connection id:", connection_id)

        var result int
        err = DB.QueryRow("select sleep(10)").Scan(&result)
        if err != nil {
                log.Println("query sleep connection id faild:", err)
                return
        }

}

output

2019/10/02 18:14:25 worker: 2 , connection id: 55
2019/10/02 18:14:25 worker: 17 , connection id: 55
2019/10/02 18:14:25 worker: 11 , connection id: 55
2019/10/02 18:14:35 worker: 3 , connection id: 55
2019/10/02 18:14:45 worker: 0 , connection id: 55
2019/10/02 18:14:45 worker: 4 , connection id: 55
2019/10/02 18:14:45 worker: 5 , connection id: 55
2019/10/02 18:15:05 worker: 7 , connection id: 55
2019/10/02 18:15:25 worker: 15 , connection id: 55
2019/10/02 18:15:25 worker: 6 , connection id: 55
2019/10/02 18:15:35 worker: 13 , connection id: 55
2019/10/02 18:15:45 worker: 19 , connection id: 55
2019/10/02 18:15:45 worker: 10 , connection id: 55
2019/10/02 18:15:45 worker: 12 , connection id: 55
2019/10/02 18:15:55 worker: 14 , connection id: 55
2019/10/02 18:16:15 worker: 8 , connection id: 55
2019/10/02 18:16:35 worker: 18 , connection id: 55
2019/10/02 18:16:35 worker: 1 , connection id: 55
2019/10/02 18:17:05 worker: 16 , connection id: 55
2019/10/02 18:17:35 worker: 9 , connection id: 55

使用show processlist查看连接

mysql> show processlist;
+----+------+-----------------+------+---------+------+------------+------------------+
| Id | User | Host            | db   | Command | Time | State      | Info             |
+----+------+-----------------+------+---------+------+------------+------------------+
| 20 | root | localhost       | NULL | Query   |    0 | starting   | show processlist |
| 55 | root | localhost:59518 | NULL | Query   |    5 | User sleep | select sleep(10) |
+----+------+-----------------+------+---------+------+------------+------------------+
2 rows in set (0.00 sec)

使用netstat 查看连接

netstat -an | grep 3306
tcp4       0      0  127.0.0.1.3306         127.0.0.1.59518        ESTABLISHED
tcp4       0      0  127.0.0.1.59518        127.0.0.1.3306         ESTABLISHED
tcp46      0      0  *.3306                 *.*                    LISTEN

从结果可以看到,20个goroutine轮流使用同一个连接(connection id 为55)执行sql语句。

当连接被占用时,其他尝试使用连接的goroutine会被阻塞。直到连接使用完后,其他goroutine才可以使用连接。

即使多个goroutine在执行SQL,也没有创建多个连接。

因此,最大连接数设置生效。

有些读者可能会问,没有看到设置最大空闲连接数,此时最大空间连接数是多少?

前面已经提到,默认的最大空闲连接数是2.

下面再来测试下最大空间连接数。

2.最大空闲连接数测试

下面例子中,设置最大连接数为1,最大空闲连接数为0.

并且每隔3s执行一条SQL语句。

代码如下:

package main

import (
        "database/sql"
        "log"
        "time"

        _ "github.com/go-sql-driver/mysql"

)

var DB *sql.DB
var dataBase = "root:Aa123456@tcp(127.0.0.1:3306)/?loc=Local&parseTime=true"

func mysqlInit() {
        var err error
        DB, err = sql.Open("mysql", dataBase)
        if err != nil {
                log.Fatalln("open db fail:", err)
        }

        DB.SetMaxOpenConns(1)
        DB.SetMaxIdleConns(0)

        err = DB.Ping()
        if err != nil {
                log.Fatalln("ping db fail:", err)
        }
}

func main() {
        mysqlInit()

        for {
                execSql()
                time.Sleep(3*time.Second)
        }
}


func execSql() {
        var connection_id int
        err := DB.QueryRow("select CONNECTION_ID()").Scan(&connection_id)
        if err != nil {
                log.Println("query connection id failed:", err)
                return
        }

        log.Println("connection id:", connection_id)
}

output:

2019/10/13 23:06:00 connection id: 26
2019/10/13 23:06:03 connection id: 27
2019/10/13 23:06:06 connection id: 28
2019/10/13 23:06:09 connection id: 29
2019/10/13 23:06:12 connection id: 30
2019/10/13 23:06:15 connection id: 31
2019/10/13 23:06:18 connection id: 32
2019/10/13 23:06:21 connection id: 33
2019/10/13 23:06:24 connection id: 34
2019/10/13 23:06:27 connection id: 35
2019/10/13 23:06:30 connection id: 36
2019/10/13 23:06:33 connection id: 37
2019/10/13 23:06:36 connection id: 38

从结果中可以看出,每次执行SQL使用的连接connection id都不同。

设置最大空闲连接数为0,每次执行SQL后,连接不会放入空闲连接池,而是会被关闭,下次执行SQL时,会重新建立新的连接。

3.参考