GORM数据库连接池对接Prometheus

一、背景与介绍

Golang的database/sql包定了关于操作数据库的相关接口，但是没有去做对应数据库的实现。这些实现是预留给开发者或者对应厂商进行实现的。

其中让我比较关注的是Golang的sql包有没有实现连接池pool的机制呢? 毕竟Golang是静态语言，类似Java我们看到很多连接池的实现例如阿里巴巴的Apache druid。给开发者提供了一个高效的数据库连接池，提高应用的性能，连接池保持与后端数据库的长连接，在需要的时候直接使用即可。使用完毕后，将连接归还到连接池中。并且还提供可视化界面，查看连接池的运行情况、连接数情况、活跃数、空闲数等等。

我们通过源码可以发现，Golang底层的sql包是实现了数据库连接池的逻辑。但是Golang的连接池，都是隐式/惰性使用，只有在执行到查询SQL或者执行SQL的时候，才从pool中拿到连接进行执行，不会像Java一样显式使用，需要从pool中拿到连接对象，再通过这个连接对象执行SQL.

二、GORM配置prometheus

1、对接Prometheus的目的

GORM支持配置Promtehues的export, 这样我们再结合Grafana面板就能很直观地对Go程序的SQL连接池进行监控。

可以监控到当前连接池的连接数有没有超过最大连接数、空闲连接数等等，由此来保证我们程序的正常运行。

一般我们的Go项目都使用到GROM，既然GORM实现了Prometheus的export，那么就使用内置功能进行测试。其实我们也可以自己实现这个exporter，通过DB对象的Stats()方法，返回连接池统计数据，再封装为exporter接口暴露出来即可。

2、开启Promtheus exporter

db.go测试代码:

Go 复制代码

package db

import (
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
	"gorm.io/plugin/prometheus"
	"log"
	"os"
	"sync"
	"time"
)

var db *gorm.DB
var once sync.Once
var getDbConnErr error
var dsn = "root:root@tcp(127.0.0.1:3307)/test?charset=utf8mb4&parseTime=True&loc=Local"

// GetDbConn 获取mysql连接
func GetDbConn() *gorm.DB {
	once.Do(func() {
		logfile, _ := os.OpenFile("./sql.log", os.O_CREATE|os.O_APPEND|os.O_RDWR, 0666)
		newLogger := logger.New(log.New(logfile, "\r\n", log.LstdFlags), logger.Config{
			SlowThreshold: time.Second,
			LogLevel:      logger.Info,
		})
		db, getDbConnErr = gorm.Open(mysql.Open(dsn), &gorm.Config{
			Logger: newLogger,
		})
		if getDbConnErr != nil {
			panic(getDbConnErr)
		}
		db.Use(prometheus.New(prometheus.Config{
			DBName:          "db1",                       // 使用 `DBName` 作为指标 label
			RefreshInterval: 15,                          // 指标刷新频率（默认为 15 秒）
			PushAddr:        "prometheus pusher address", // 如果配置了 `PushAddr`，则推送指标
			StartServer:     true,                        // 启用一个 http 服务来暴露指标
			HTTPServerPort:  8080,                        // 配置 http 服务监听端口，默认端口为 8080 （如果您配置了多个，只有第一个 `HTTPServerPort` 会被使用）
			MetricsCollector: []prometheus.MetricsCollector{
				&prometheus.MySQL{
					Prefix:        "gorm_mysql_stats_",
					Interval:      15,
					VariableNames: []string{"Threads_running"},
				},
			}, // 用户自定义指标
		}))
		DB, _ := db.DB()
		// 最大连接数
		DB.SetMaxOpenConns(2)
		// 最大空闲连接数 <= 最大连接数
		DB.SetMaxIdleConns(1)
		// 连接数的生命周期, 从创建开始记时，到时间点会被销毁
		DB.SetConnMaxLifetime(time.Hour * 7)
		// 最大空闲时间, 这个链接距离上一次被使用的时间段，超过这个时间则认为该连接已过期
		DB.SetConnMaxIdleTime(time.Hour)
	})
	return db
}

main.go测试代码:

Go 复制代码

package main

import (
	db "awesomeProject1/db"
	"fmt"
	_ "gorm.io/driver/mysql"
	"gorm.io/gorm"
	"math/rand"
	"time"
)

func main() {
	dbConn := db.GetDbConn()
	for i := 0; i < 10; i++ {
		go func(conn *gorm.DB) {
			for {
				time.Sleep(time.Second * time.Duration(rand.Intn(10)))
				rows := dbConn.Raw("select * from users")
				var res []map[string]interface{}
				fmt.Println(rows.Scan(&res))
			}
		}(dbConn)
	}
	for {
		time.Sleep(time.Second * 2)
		fmt.Println("running....")
	}
}

访问http://localhost:8080/metrics 查看到指标数据:

3、采集metrics到Promtehus

promethues.yaml配置添加采集任务:

4、Grafana配置监控面板

1、通过图表能看到，我们配置了连接池最大连接数是2个, 但是有10个协程去执行查询SQL,观察wait_count指标曲线在持续上涨, 此时有26个协程还在阻塞等待获取数据库连接，同时查看右边是等待获取数据库连接等待耗时曲线，也是一直在上涨，此时我们应该判定数据库连接池满了。那么例如生产环境，看到这种曲线要么我们设置的连接池最大连接数过小，要么mysql的连接数已经跑满了。此时应该告警，查看异常的情况。

2、我们将最大连接数调整为20以后再观察监控面板:

最大连接数已从2调整为20(2->20), 此时再看监控面板，等待的协程数量wait_count:0、同时协程等待获取数据库连接的时间已经降为了0ns, 此时数据库连接池不再是爆满的状态。

三、总结

无论Golang的什么ORM框架，GORM也好、XORM也罢，底层还是去使用了database/sql这个包，这个包底层实现了连接池的抽象，同时提供了获取连接池相关统计信息的接口。我们再结合Prometheus将这些指标暴露出来，结合Grafana这套告警体系，就可以实时监控和观察我们Go程序MySQL连接池的情况。