feat: Add MySQL driver tracing

scribd · Jun 20, 2024 · fc2cd39 · fc2cd39
1 parent 3aa40ed
commit fc2cd39
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/database/gorm.go b/pkg/database/gorm.go
@@ -6,6 +6,8 @@ import (
 	"time"
 
 	"github.com/DATA-DOG/go-txdb"
+	mysqldriver "github.com/go-sql-driver/mysql"
+	sqltrace "gopkg.in/DataDog/dd-trace-go.v1/contrib/database/sql"
 	gormtrace "gopkg.in/DataDog/dd-trace-go.v1/contrib/gorm.io/gorm.v1"
 	"gorm.io/driver/mysql"
 	"gorm.io/gorm"
@@ -18,6 +20,8 @@ func NewConnection(config *Config, environment, appName string) (*gorm.DB, error
 	serviceName := fmt.Sprintf("%s-mysql", appName)
 	dialector := getDialectorFromConfig(config, environment)
 
+	sqltrace.Register(dialector.Name(), &mysqldriver.MySQLDriver{}, sqltrace.WithServiceName(serviceName))
+
 	db, err := gormtrace.Open(dialector, nil, gormtrace.WithServiceName(serviceName))
 	if err != nil {
 		return nil, err