forked from newrelic/go-agent
-
Notifications
You must be signed in to change notification settings - Fork 0
/
nrawssdk.go
87 lines (81 loc) · 3.28 KB
/
nrawssdk.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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
// Copyright 2020 New Relic Corporation. All rights reserved.
// SPDX-License-Identifier: Apache-2.0
// Package nrawssdk instruments https://github.com/aws/aws-sdk-go requests.
package nrawssdk
import (
"github.com/aws/aws-sdk-go/aws/request"
internal "github.com/newrelic/go-agent/_integrations/nrawssdk/internal"
agentinternal "github.com/newrelic/go-agent/internal"
)
func init() { agentinternal.TrackUsage("integration", "library", "aws-sdk-go") }
func startSegment(req *request.Request) {
input := internal.StartSegmentInputs{
HTTPRequest: req.HTTPRequest,
ServiceName: req.ClientInfo.ServiceName,
Operation: req.Operation.Name,
Region: req.ClientInfo.SigningRegion,
Params: req.Params,
}
req.HTTPRequest = internal.StartSegment(input)
}
func endSegment(req *request.Request) {
ctx := req.HTTPRequest.Context()
internal.EndSegment(ctx, req.HTTPResponse.Header)
}
// InstrumentHandlers will add instrumentation to the given *request.Handlers.
//
// A Segment will be created for each out going request. The Transaction must
// be added to the `http.Request`'s Context in order for the segment to be
// recorded. For DynamoDB calls, these segments will be
// `newrelic.DatastoreSegment` type and for all others they will be
// `newrelic.ExternalSegment` type.
//
// Additional attributes will be added to Transaction Trace Segments and Span
// Events: aws.region, aws.requestId, and aws.operation.
//
// To add instrumentation to the Session and see segments created for each
// invocation that uses the Session, call InstrumentHandlers with the session's
// Handlers and add the current Transaction to the `http.Request`'s Context:
//
// ses := session.New()
// // Add instrumentation to handlers
// nrawssdk.InstrumentHandlers(&ses.Handlers)
// lambdaClient = lambda.New(ses, aws.NewConfig())
//
// req, out := lambdaClient.InvokeRequest(&lambda.InvokeInput{
// ClientContext: aws.String("MyApp"),
// FunctionName: aws.String("Function"),
// InvocationType: aws.String("Event"),
// LogType: aws.String("Tail"),
// Payload: []byte("{}"),
// }
// // Add txn to http.Request's context
// req.HTTPRequest = newrelic.RequestWithTransactionContext(req.HTTPRequest, txn)
// err := req.Send()
//
// To add instrumentation to a Request and see a segment created just for the
// individual request, call InstrumentHandlers with the `request.Request`'s
// Handlers and add the current Transaction to the `http.Request`'s Context:
//
// req, out := lambdaClient.InvokeRequest(&lambda.InvokeInput{
// ClientContext: aws.String("MyApp"),
// FunctionName: aws.String("Function"),
// InvocationType: aws.String("Event"),
// LogType: aws.String("Tail"),
// Payload: []byte("{}"),
// }
// // Add instrumentation to handlers
// nrawssdk.InstrumentHandlers(&req.Handlers)
// // Add txn to http.Request's context
// req.HTTPRequest = newrelic.RequestWithTransactionContext(req.HTTPRequest, txn)
// err := req.Send()
func InstrumentHandlers(handlers *request.Handlers) {
handlers.Send.SetFrontNamed(request.NamedHandler{
Name: "StartNewRelicSegment",
Fn: startSegment,
})
handlers.Send.SetBackNamed(request.NamedHandler{
Name: "EndNewRelicSegment",
Fn: endSegment,
})
}