/
main.go
62 lines (53 loc) · 1.8 KB
/
main.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
50
51
52
53
54
55
56
57
58
59
60
61
62
// Copyright 2020, 2021 New Relic Corporation. All rights reserved.
// SPDX-License-Identifier: Apache-2.0
//
// Example of using nrpgx to instrument a Postgres database application
// using the jackc/pgx driver with database/sql.
//
// To run this example, be sure the environment variable NEW_RELIC_LICENSE_KEY
// is set to your license key. Postgres must be running on the default port
// 5432 on localhost, and have a password "docker". An easy (albeit insecure)
// way to test this is to issue the following command to run a postgres database
// in a docker container:
// docker run --rm -e POSTGRES_PASSWORD=docker -p 5432:5432 postgres
//
// Run that in the background or in a separate window, and then run this program
// to access that database.
//
package main
import (
"context"
"database/sql"
"fmt"
"os"
"time"
_ "github.com/newrelic/go-agent/v3/integrations/nrpgx"
"github.com/newrelic/go-agent/v3/newrelic"
)
func main() {
// docker run --rm -e POSTGRES_PASSWORD=docker -p 5432:5432 postgres
db, err := sql.Open("nrpgx", "host=localhost port=5432 user=postgres dbname=postgres password=docker sslmode=disable")
if err != nil {
panic(err)
}
app, err := newrelic.NewApplication(
newrelic.ConfigAppName("PostgreSQL App"),
newrelic.ConfigLicense(os.Getenv("NEW_RELIC_LICENSE_KEY")),
newrelic.ConfigDebugLogger(os.Stdout),
)
if err != nil {
panic(err)
}
//
// N.B.: We do not recommend using app.WaitForConnection in production code.
//
app.WaitForConnection(5 * time.Second)
txn := app.StartTransaction("postgresQuery")
ctx := newrelic.NewContext(context.Background(), txn)
row := db.QueryRowContext(ctx, "SELECT count(*) FROM pg_catalog.pg_tables")
var count int
row.Scan(&count)
txn.End()
app.Shutdown(5 * time.Second)
fmt.Println("number of entries in pg_catalog.pg_tables", count)
}