fix: prepared statements enabled cause metrcis not work (#31)

zstone12 · web-flow · commit e290f954e239 · 2024-10-24T21:19:27.000+08:00
diff --git a/tracing/tracing.go b/tracing/tracing.go
@@ -62,8 +62,8 @@ type gormRegister interface {
 
 func (p otelPlugin) Initialize(db *gorm.DB) (err error) {
 	if !p.excludeMetrics {
-		if db, ok := db.ConnPool.(*sql.DB); ok {
-			metrics.ReportDBStatsMetrics(db)
+		if sqlDB, err := db.DB(); err == nil {
+			metrics.ReportDBStatsMetrics(sqlDB)
 		}
 	}
 

Original file line number	Diff line number	Diff line change
`@@ -62,8 +62,8 @@ type gormRegister interface {`
`62`	`62`
`63`	`63`	`func (p otelPlugin) Initialize(db *gorm.DB) (err error) {`
`64`	`64`	`if !p.excludeMetrics {`
`65`		`- if db, ok := db.ConnPool.(*sql.DB); ok {`
`66`		`- metrics.ReportDBStatsMetrics(db)`
	`65`	`+ if sqlDB, err := db.DB(); err == nil {`
	`66`	`+ metrics.ReportDBStatsMetrics(sqlDB)`
`67`	`67`	`}`
`68`	`68`	`}`
`69`	`69`