/
oracle.go
49 lines (42 loc) · 1.22 KB
/
oracle.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
// Package pporacle instruments the sijms/go-ora/v2 package (https://github.com/sijms/go-ora).
//
// This package instruments the Oracle driver calls.
// Use this package's driver in place of the Oracle driver.
//
// db, err := sql.Open("oracle-pinpoint", "oracle://scott:tiger@localhost:1521/xe")
//
// It is necessary to pass the context containing the pinpoint.Tracer to all exec and query methods on SQL driver.
//
// ctx := pinpoint.NewContext(context.Background(), tracer)
// row := db.QueryRowContext(ctx, "SELECT * FROM BONUS")
package pporacle
import (
"database/sql"
"net"
"net/url"
"strings"
"github.com/pinpoint-apm/pinpoint-go-agent"
"github.com/sijms/go-ora/v2"
)
var dbInfo = pinpoint.DBInfo{
ParseDSN: parseDSN,
}
func init() {
dbInfo.DBType = pinpoint.ServiceTypeOracle
dbInfo.QueryType = pinpoint.ServiceTypeOracleExecuteQuery
sql.Register("oracle-pinpoint", pinpoint.WrapSQLDriver(&go_ora.OracleDriver{}, dbInfo))
}
func parseDSN(info *pinpoint.DBInfo, dbUrl string) {
u, err := url.Parse(dbUrl)
if err != nil {
return
}
host, _, err := net.SplitHostPort(u.Host)
if err != nil {
host = u.Host
} else if host == "" {
host = "localhost"
}
info.DBHost = host
info.DBName = strings.Trim(u.Path, "/")
}