diff --git a/pkg/database/gorm.go b/pkg/database/gorm.go index 4dab3a8..a801a8b 100644 --- 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