36 lines
803 B
Go
36 lines
803 B
Go
|
package pg
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
|
||
|
|
"go.nhat.io/otelsql"
|
||
|
|
semconv "go.opentelemetry.io/otel/semconv/v1.20.0"
|
||
|
|
)
|
||
|
|
|
||
|
|
func openDB(driver, dsn string) (*sql.DB, error) {
|
||
|
|
// Register the otelsql wrapper for the provided postgres driver.
|
||
|
|
driverName, err := otelsql.Register(driver,
|
||
|
|
otelsql.AllowRoot(),
|
||
|
|
otelsql.TraceQueryWithoutArgs(),
|
||
|
|
otelsql.TraceRowsClose(),
|
||
|
|
otelsql.TraceRowsAffected(),
|
||
|
|
// otelsql.WithDatabaseName("my_database"), // Optional.
|
||
|
|
otelsql.WithSystem(semconv.DBSystemPostgreSQL), // Optional.
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Connect to a Postgres database using the postgres driver wrapper.
|
||
|
|
db, err := sql.Open(driverName, dsn)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := otelsql.RecordStats(db); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return db, nil
|
||
|
|
}
|