golanggo-sql-drivemysql连接池的实现

设置最大的连接数，可以避免并发太高导致连接mysql出现too many connections的错误。设置闲置的连接数则当开启的一个连接使用完成后可以放在池里等候下一次使用。

请求方法

上面开启http请求设置了请求/pool地址的执行方法

func pool(w http.ResponseWriter, r *http.Request) {
	rows, err := db.Query("SELECT * FROM user limit 1")
	defer rows.Close()
	checkErr(err)
	columns, _ := rows.Columns()
	scanArgs := make([]interface{}, len(columns))
	values := make([]interface{}, len(columns))
	for j := range values {
	scanArgs[j] = &values[j]
	}
	record := make(map[string]string)
	for rows.Next() {
	//将行数据保存到record字典
	err = rows.Scan(scanArgs...)
	for i, col := range values {
	if col != nil {
	record[columns[i]] = string(col.([]byte))
	}
	}
	}
	fmt.Println(record)
	fmt.Fprintln(w, "finish")
}
func checkErr(err error) {
	if err != nil {
	fmt.Println(err)
	panic(err)
	}
}

pool方法就是从user表中查出一条记录然后存放到map中，最后输出finish。代码到这里就算完了非常简单，下面来测试一下。首先启动http服务，然后使用ab进行并发测试访问：

$ ab -c 100 -n 1000 'http://localhost:9090/pool'

在数据库中通过show processlist查看连接进程：

golang数据库连接池

可以看到有100来个进程。

因为避免了重复创建连接，所以使用连接池可以很明显的提高性能。有兴趣的童靴可以去掉连接池代码自己测试一下。完整代码如下：

//数据库连接池测试
package main
import (
	"database/sql"
	"fmt"
	_ "github.com/go-sql-driver/mysql"
	"log"
	"net/http"
)
var db *sql.DB
func init() {
	db, _ = sql.Open("mysql", "root:@tcp(127.0.0.1:3306)/test?charset=utf8")
	db.SetMaxOpenConns(2000)
	db.SetMaxIdleConns(1000)
	db.Ping()
}
func main() {
	startHttpServer()
}
func startHttpServer() {
	http.HandleFunc("/pool", pool)
	err := http.ListenAndServe(":9090", nil)
	if err != nil {
	log.Fatal("ListenAndServe: ", err)
	}
}
func pool(w http.ResponseWriter, r *http.Request) {
	rows, err := db.Query("SELECT * FROM user limit 1")
	defer rows.Close()
	checkErr(err)
	columns, _ := rows.Columns()
	scanArgs := make([]interface{}, len(columns))
	values := make([]interface{}, len(columns))
	for j := range values {
	scanArgs[j] = &values[j]
	}
	record := make(map[string]string)
	for rows.Next() {
	//将行数据保存到record字典
	err = rows.Scan(scanArgs...)
	for i, col := range values {
	if col != nil {
	record[columns[i]] = string(col.([]byte))
	}
	}
	}
	fmt.Println(record)
	fmt.Fprintln(w, "finish")
}
func checkErr(err error) {
	if err != nil {
	fmt.Println(err)
	panic(err)
	}
}

小结

golang这边实现的连接池只提供了SetMaxOpenConns和SetMaxIdleConns方法进行连接池方面的配置。在使用的过程中有一个问题就是数据库本身对连接有一个超时时间的设置，如果超时时间到了数据库会单方面断掉连接，此时再用连接池内的连接进行访问就会出错。

packets.go:32: unexpected EOF
packets.go:118: write tcp 192.168.3.90:3306: broken pipe

上面都是错误都是go-sql-drive本身的输出，有的时候还会出现bad connection的错误。多请求几次后连接池会重新打开新连接这时候就没有问题了。关于这个问题自己有初步的解决方法，但是感觉不太完美，下次再放上来。

转载请注明：快乐编程 » golang go-sql-drive mysql连接池的实现