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
|
// Copyright 2020 New Relic Corporation. All rights reserved.
// SPDX-License-Identifier: Apache-2.0
package main
import (
"context"
"encoding/json"
"fmt"
"os"
"time"
elasticsearch "github.com/elastic/go-elasticsearch/v7"
nrelasticsearch "github.com/newrelic/go-agent/v3/integrations/nrelasticsearch-v7"
"github.com/newrelic/go-agent/v3/newrelic"
)
func main() {
// Step 1: Use nrelasticsearch.NewRoundTripper to assign the
// elasticsearch.Config's Transport field.
cfg := elasticsearch.Config{
Transport: nrelasticsearch.NewRoundTripper(nil),
}
app, err := newrelic.NewApplication(
newrelic.ConfigAppName("Elastic App"),
newrelic.ConfigLicense(os.Getenv("NEW_RELIC_LICENSE_KEY")),
newrelic.ConfigDebugLogger(os.Stdout),
)
if nil != err {
panic(err)
}
app.WaitForConnection(5 * time.Second)
txn := app.StartTransaction("elastic")
// Step 2: Ensure that all calls using the elasticsearch client have a
// context which includes the newrelic.Transaction.
ctx := newrelic.NewContext(context.Background(), txn)
es, err := elasticsearch.NewClient(cfg)
if err != nil {
panic(err)
}
res, err := es.Info(es.Info.WithContext(ctx))
if err != nil {
panic(err)
}
if res.IsError() {
panic(err)
}
var r map[string]interface{}
if err := json.NewDecoder(res.Body).Decode(&r); err != nil {
panic(err)
}
fmt.Println("ELASTIC SEARCH INFO", elasticsearch.Version, r["version"].(map[string]interface{})["number"])
txn.End()
app.Shutdown(5 * time.Second)
}
|
