/
main.go
53 lines (42 loc) · 1.37 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
// Copyright 2020 New Relic Corporation. All rights reserved.
// SPDX-License-Identifier: Apache-2.0
package main
import (
"fmt"
"net/http"
"os"
"github.com/gorilla/mux"
newrelic "github.com/newrelic/go-agent"
nrgorilla "github.com/newrelic/go-agent/_integrations/nrgorilla/v1"
)
func makeHandler(text string) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Write([]byte(text))
})
}
func mustGetEnv(key string) string {
if val := os.Getenv(key); "" != val {
return val
}
panic(fmt.Sprintf("environment variable %s unset", key))
}
func main() {
cfg := newrelic.NewConfig("Gorilla App", mustGetEnv("NEW_RELIC_LICENSE_KEY"))
cfg.Logger = newrelic.NewDebugLogger(os.Stdout)
app, err := newrelic.NewApplication(cfg)
if nil != err {
fmt.Println(err)
os.Exit(1)
}
r := mux.NewRouter()
r.Handle("/", makeHandler("index"))
r.Handle("/alpha", makeHandler("alpha"))
users := r.PathPrefix("/users").Subrouter()
users.Handle("/add", makeHandler("adding user"))
users.Handle("/delete", makeHandler("deleting user"))
// The route name will be used as the transaction name if one is set.
r.Handle("/named", makeHandler("named route")).Name("special-name-route")
// The NotFoundHandler will be instrumented if it is set.
r.NotFoundHandler = makeHandler("not found")
http.ListenAndServe(":8000", nrgorilla.InstrumentRoutes(r, app))
}