Merge pull request #129 from magnusbaeck/connpool-sql-db

fix(otelgorm): Make metric reporting work with prepared statements
uptrace · Jun 13, 2024 · aeefc42 · aeefc42
2 parents 33d1fd3 + d3f269a
commit aeefc42
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/otelgorm/otelgorm.go b/otelgorm/otelgorm.go
@@ -55,8 +55,8 @@ type gormRegister interface {
 
 func (p otelPlugin) Initialize(db *gorm.DB) (err error) {
 	if !p.excludeMetrics {
-		if db, ok := db.ConnPool.(*sql.DB); ok {
-			otelsql.ReportDBStatsMetrics(db)
+		if sqlDB, err := db.DB(); err == nil {
+			otelsql.ReportDBStatsMetrics(sqlDB)
 		}
 	}