/
main.go
70 lines (59 loc) · 1.64 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
63
64
65
66
67
68
69
70
// Copyright 2020 New Relic Corporation. All rights reserved.
// SPDX-License-Identifier: Apache-2.0
// An application that illustrates Distributed Tracing or Cross Application
// Tracing when using NewRoundTripper.
package main
import (
"fmt"
"net/http"
"os"
"time"
"github.com/newrelic/go-agent"
)
func mustGetEnv(key string) string {
if val := os.Getenv(key); "" != val {
return val
}
panic(fmt.Sprintf("environment variable %s unset", key))
}
func doRequest(txn newrelic.Transaction) error {
for _, addr := range []string{"segments", "mysql"} {
url := fmt.Sprintf("http://localhost:8000/%s", addr)
req, err := http.NewRequest("GET", url, nil)
if nil != err {
return err
}
client := &http.Client{}
// Using NewRoundTripper automatically instruments all request
// for Distributed Tracing and Cross Application Tracing.
client.Transport = newrelic.NewRoundTripper(txn, nil)
resp, err := client.Do(req)
if nil != err {
return err
}
fmt.Println("response code is", resp.StatusCode)
}
return nil
}
func main() {
cfg := newrelic.NewConfig("Client App RoundTripper", mustGetEnv("NEW_RELIC_LICENSE_KEY"))
cfg.Logger = newrelic.NewDebugLogger(os.Stdout)
cfg.DistributedTracer.Enabled = true
app, err := newrelic.NewApplication(cfg)
if nil != err {
fmt.Println(err)
os.Exit(1)
}
// Wait for the application to connect.
if err = app.WaitForConnection(5 * time.Second); nil != err {
fmt.Println(err)
}
txn := app.StartTransaction("client-txn", nil, nil)
err = doRequest(txn)
if nil != err {
txn.NoticeError(err)
}
txn.End()
// Shut down the application to flush data to New Relic.
app.Shutdown(10 * time.Second)
}