文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

golang mysql的连接池如何使用

2023-07-05 05:00

关注

本文小编为大家详细介绍“golang mysql的连接池如何使用”,内容详细,步骤清晰,细节处理妥当,希望这篇“golang mysql的连接池如何使用”文章能帮助大家解决疑惑,下面跟着小编的思路慢慢深入,一起来学习新知识吧。

1.mysql-通过sql建立连接池

连接池用sql.Open函数创建连接池,可是此时只是初始化了连接池,并没有创建任何连接。连接创建都是惰性的,只有当你真正使用到连接的时候,连接池才会创建连接。连接池很重要,它直接影响着你的程序行为。

连接池的工作原来却相当简单。当你的函数(例如Exec,Query)调用需要访问底层数据库的时候,函数首先会向连接池请求一个连接。如果连接池有空闲的连接,则返回给函数。否则连接池将会创建一个新的连接给函数。一旦连接给了函数,连接则归属于函数。函数执行完毕后,要不把连接所属权归还给连接池,要么传递给下一个需要连接的(Rows)对象,最后使用完连接的对象也会把连接释放回到连接池。

请求一个连接的函数有好几种,执行完毕处理连接的方式稍有差别,大致如下:

因为每一个连接都是惰性创建的,如何验证sql.Open调用之后,sql.DB对象可用呢?通常使用db.Ping()方法初始化,调用了Ping之后,连接池一定会初始化一个数据库连接。

连接失败关于连接池另外一个知识点就是你不必检查或者尝试处理连接失败的情况。当你进行数据库操作的时候,如果连接失败了,database/sql会帮你处理。实际上,当从连接池取出的连接断开的时候,database/sql会自动尝试重连10次。仍然无法重连的情况下会自动从连接池再获取一个或者新建另外一个。

连接池配置配置连接池有两个的方法:

对于连接池的使用依赖于你是如何配置连接池,如果使用不当会导致下面问题:

数据库标准接口里面有3个方法用于设置连接池的属性: SetConnMaxLifetime, SetMaxIdleConns, SetMaxOpenConns

项目结构

.+--- go.mod+--- go.sum+--- main.go+--- pool|   +--- sql-pool.go

代码pool

package poolimport ("database/sql""fmt"_ "github.com/go-sql-driver/mysql""log""time")var DB *sql.DBfunc init()  {DB, _ = sql.Open("mysql", "root:root@tcp(localhost:3306)/test?charset=utf8&parseTime=True&loc=Local") // 使用本地时间,即东八区,北京时间 // set pool paramsDB.SetMaxOpenConns(2000)DB.SetMaxIdleConns(1000)DB.SetConnMaxLifetime(time.Minute * 60) // mysql default conn timeout=8h, should < mysql_timeout err := DB.Ping() if err != nil { log.Fatalf("database init failed, err: ", err) }log.Println("mysql conn pool has initiated.")}func checkErr(err error)  {if err != nil {log.Println(err)panic(err)}}func createTable() {db := DBtable := `CREATE TABLE IF NOT EXISTS test.user ( user_id INT(11) UNSIGNED NOT NULL AUTO_INCREMENT COMMENT '用户编号', user_name VARCHAR(45) NOT NULL COMMENT '用户名称', user_age TINYINT(3) UNSIGNED NOT NULL DEFAULT 0 COMMENT '用户年龄', user_sex TINYINT(3) UNSIGNED NOT NULL DEFAULT 0 COMMENT '用户性别', PRIMARY KEY (user_id)) ENGINE = InnoDB AUTO_INCREMENT = 1 DEFAULT CHARACTER SET = utf8 COLLATE = utf8_general_ci COMMENT = '用户表'`if _, err := db.Exec(table); err != nil {checkErr(err)}}func insert() {db := DBstmt, err := db.Prepare(`INSERT user (user, age) values (?, ?)`)checkErr(err)res, err := stmt.Exec("Elvis", 26)checkErr(err)id, err := res.LastInsertId()checkErr(err)log.Println(id)}func query() {db := DBrows, err := db.Query("SELECT * FROM user")checkErr(err)for rows.Next() {var userId intvar userName stringvar userAge intvar userSex introws.Columns()err = rows.Scan(&userId, &userName, &userAge, &userSex)checkErr(err)fmt.Println(userId)fmt.Println(userName)fmt.Println(userAge)fmt.Println(userSex)}}func queryToMap() {db := DBrows, err := db.Query("SELECT * FROM user")checkErr(err)//字典类型//构造scanArgs、values两个数组,scanArgs的每个值指向values相应值的地址columns, _ := rows.Columns()scanArgs := make([]interface{}, len(columns))values := make([]interface{}, len(columns))for i := range values {scanArgs[i] = &values[i]}for rows.Next() {//将行数据保存到record字典err = rows.Scan(scanArgs...)record := make(map[string]string)for i, col := range values {if col != nil {record[columns[i]] = string(col.([]byte))}}fmt.Println(record)}}func update() {db := DBstmt, err := db.Prepare(`UPDATE user SET user_age=?,user_sex=? WHERE user_id=?`)checkErr(err)res, err := stmt.Exec(21, 2, 1)checkErr(err)num, err := res.RowsAffected()checkErr(err)fmt.Println(num)}func remove() {db := DBstmt, err := db.Prepare(`DELETE FROM user WHERE user_id=?`)checkErr(err)res, err := stmt.Exec(1)checkErr(err)num, err := res.RowsAffected()checkErr(err)fmt.Println(num)}

main

package mainimport ("fmt""log""net/http". "go-mysql-pool-v1/pool")func main()  {http.HandleFunc("/pool", pool)log.Println("server is up now...")http.ListenAndServe(":8080", nil)}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() {err = rows.Scan(scanArgs...)for i, col := range values {if col != nil {record[columns[i]] = string(col.([]byte))}}}log.Println(record)fmt.Fprintf(w, "finish")}func checkErr(err error)  {if err != nil {log.Println(err)panic(err)}}

可以通过工具ab测试连接池性能,用法见最下方注。

2.mysql-gorm 建立连接池

其实gorm的连接池设置,底层还是用的database/sql的设置连接池的方法,无非就是加一层gorm自身的一些设置。

以下示例为gorm v2版本,v1版本通过github.com/jinzhu/gorm,如mysql的驱动导入需要加_

代码结构

.+--- config|   +--- config.go|   +--- config.json+--- go.mod+--- go.sum+--- main.go+--- pool|   +--- gorm-pool.go+--- test.db

config.json

{  "database": {    "name": "test",    "type": "mysql",    "host": "localhost",    "port": "3306",    "user": "root",    "password": "root",    "table_prefix": ""  }}

config.go

package configimport ("encoding/json""os")type Database struct {Type        string `json:"type"`Host        string `json:"host"`Port        string `json:"port"`User        string `json:"user"`Password    string `json:"password"`Name        string `json:"name"`TablePrefix string `json:"table_prefix"`}var DatabaseSetting = &Database{}type Config struct {Database *Database `json:"database"`}var GlobalConfigSetting = &Config{}func init()  {// win path is abs-path, linux -> config.jsonfilePtr, err := os.Open("D:\\demo1\\src\\demo\\demo06\\go-mysql-pool-v2\\config\\config.json")if err != nil {return}defer filePtr.Close()// json decodedecoder := json.NewDecoder(filePtr)err = decoder.Decode(GlobalConfigSetting)DatabaseSetting = GlobalConfigSetting.Database}

gorm-pool.go注意设置表前缀和单复数

package poolimport ("fmt""go-mysql-pool-v2/config""gorm.io/driver/mysql""gorm.io/driver/postgres""gorm.io/driver/sqlite""gorm.io/gorm""gorm.io/gorm/schema""log""time")var db *gorm.DB// gorm v2func init()  {var dbURi stringvar dialector gorm.Dialectorif config.DatabaseSetting.Type == "mysql" {dbURi = fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8&parseTime=true",config.DatabaseSetting.User,config.DatabaseSetting.Password,config.DatabaseSetting.Host,config.DatabaseSetting.Port,config.DatabaseSetting.Name)dialector = mysql.New(mysql.Config{DSN:                       dbURi, // data source nameDefaultStringSize:         256,   // default size for string fieldsDisableDatetimePrecision:  true,  // disable datetime precision, which not supported before MySQL 5.6DontSupportRenameIndex:    true,  // drop & create when rename index, rename index not supported before MySQL 5.7, MariaDBDontSupportRenameColumn:   true,  // `change` when rename column, rename column not supported before MySQL 8, MariaDBSkipInitializeWithVersion: false, // auto configure based on currently MySQL version})} else if config.DatabaseSetting.Type == "postgres" {dbURi = fmt.Sprintf("host=%s port=%s user=%s dbname=%s sslmode=disable password=%s",config.DatabaseSetting.Host,config.DatabaseSetting.Port,config.DatabaseSetting.User,config.DatabaseSetting.Name,config.DatabaseSetting.Password)dialector = postgres.New(postgres.Config{DSN:                  "user=gorm password=gorm dbname=gorm port=9920 sslmode=disable TimeZone=Asia/Shanghai",PreferSimpleProtocol: true, // disables implicit prepared statement usage})} else { // sqlite3dbURi = fmt.Sprintf("test.db")dialector = sqlite.Open("test.db")}conn, err := gorm.Open(dialector, &gorm.Config{ // 如果不用设置表前缀及复数,nilNamingStrategy: schema.NamingStrategy{TablePrefix: config.DatabaseSetting.TablePrefix, // set table prefixSingularTable: true, // set table singular},})if err != nil {log.Println(err.Error())}sqlDB, err := conn.DB()if err != nil {log.Println("connect db server failed.")}sqlDB.SetMaxOpenConns(100)sqlDB.SetMaxIdleConns(10)sqlDB.SetConnMaxLifetime(600*time.Second)db = conn}// open apifunc GetDB() *gorm.DB {sqlDB, err := db.DB()if err != nil {log.Println("connect db server failed.")}if err = sqlDB.Ping(); err != nil {sqlDB.Close()}return db}

main

package mainimport ("go-mysql-pool-v2/pool""gorm.io/gorm""log")type Product struct {gorm.Model // default add id stats time, id as primary keyCode stringPrice uint}func main()  {log.Println("gorm init...")SetupModel()}func SetupModel()  {db := pool.GetDB()// auto migratedb.AutoMigrate(&Product{})// create recorddb.Create(&Product{Code: "L1212", Price: 1000})}

go.mod

module go-mysql-pool-v2go 1.16replace go-mysql-pool-v2 => ../go-mysql-pool-v2require (gorm.io/driver/mysql v1.3.4gorm.io/driver/postgres v1.3.4gorm.io/driver/sqlite v1.3.4gorm.io/gorm v1.23.5)

3.连接池相较于单个client

4.通用连接池

ubuntu安装ab工具,apt-get install apache2-utils// get 一般请求ab -n 1000 -c 1000 http://xxx// post json请求ab -n 100000 -c 400 -p tempPara.txt -T application/json http://xxxtempPara.txt内容:{"driverId": 17,"pageNo": 1,"pageSize": 20,"status": 1}

读到这里,这篇“golang mysql的连接池如何使用”文章已经介绍完毕,想要掌握这篇文章的知识点还需要大家自己动手实践使用过才能领会,如果想了解更多相关内容的文章,欢迎关注编程网行业资讯频道。

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-数据库
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯