63 lines
2.1 KiB
Go
63 lines
2.1 KiB
Go
|
// Package sqlx provides functions to trace the jmoiron/sqlx package (https://github.com/jmoiron/sqlx).
|
||
|
|
// To enable tracing, first use one of the "Register*" functions to register the sql driver that
|
||
|
|
// you will be using, then continue using the package as you normally would.
|
||
|
|
//
|
||
|
|
// For more information on registering and why this needs to happen, please check the
|
||
|
|
// github.com/DataDog/dd-trace-go/contrib/database/sql package.
|
||
|
|
//
|
||
|
|
package sqlx // import "gopkg.in/DataDog/dd-trace-go.v1/contrib/jmoiron/sqlx"
|
||
|
|
|
||
|
|
import (
|
||
|
|
sqltraced "gopkg.in/DataDog/dd-trace-go.v1/contrib/database/sql"
|
||
|
|
|
||
|
|
"github.com/jmoiron/sqlx"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Open opens a new (traced) connection to the database using the given driver and source.
|
||
|
|
// Note that the driver must formerly be registered using database/sql integration's Register.
|
||
|
|
func Open(driverName, dataSourceName string) (*sqlx.DB, error) {
|
||
|
|
db, err := sqltraced.Open(driverName, dataSourceName)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return sqlx.NewDb(db, driverName), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// MustOpen is the same as Open, but panics on error.
|
||
|
|
// To get tracing, the driver must be formerly registered using the database/sql integration's
|
||
|
|
// Register.
|
||
|
|
func MustOpen(driverName, dataSourceName string) (*sqlx.DB, error) {
|
||
|
|
db, err := sqltraced.Open(driverName, dataSourceName)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
return sqlx.NewDb(db, driverName), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Connect connects to the data source using the given driver.
|
||
|
|
// To get tracing, the driver must be formerly registered using the database/sql integration's
|
||
|
|
// Register.
|
||
|
|
func Connect(driverName, dataSourceName string) (*sqlx.DB, error) {
|
||
|
|
db, err := Open(driverName, dataSourceName)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
err = db.Ping()
|
||
|
|
if err != nil {
|
||
|
|
db.Close()
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return db, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// MustConnect connects to a database and panics on error.
|
||
|
|
// To get tracing, the driver must be formerly registered using the database/sql integration's
|
||
|
|
// Register.
|
||
|
|
func MustConnect(driverName, dataSourceName string) *sqlx.DB {
|
||
|
|
db, err := Connect(driverName, dataSourceName)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
return db
|
||
|
|
}
|