mirror of
https://github.com/open-telemetry/opentelemetry-go.git
synced 2024-12-14 10:13:10 +02:00
0425c09c31
* Add dice example * Add manual instrumentation * Update changelog * Add comments * Use naked return consistently * Move handleFunc to main * refactor: Extract newHTTPHandler * Fix comment * Improve comments * Update example/dice/otel.go Co-authored-by: Tyler Yahn <MrAlias@users.noreply.github.com> * Simplify BaseContext * Update doc.go * Update example/dice/main.go Co-authored-by: Phillip Carter <pcarter@fastmail.com> * Rename span and metric names --------- Co-authored-by: Tyler Yahn <MrAlias@users.noreply.github.com> Co-authored-by: Phillip Carter <pcarter@fastmail.com>
100 lines
2.6 KiB
Go
100 lines
2.6 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"time"
|
|
|
|
"go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp"
|
|
)
|
|
|
|
func main() {
|
|
if err := run(); err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
}
|
|
|
|
func run() (err error) {
|
|
// Handle SIGINT (CTRL+C) gracefully.
|
|
ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt)
|
|
defer stop()
|
|
|
|
// Set up OpenTelemetry.
|
|
serviceName := "dice"
|
|
serviceVersion := "0.1.0"
|
|
otelShutdown, err := setupOTelSDK(ctx, serviceName, serviceVersion)
|
|
if err != nil {
|
|
return
|
|
}
|
|
// Handle shutdown properly so nothing leaks.
|
|
defer func() {
|
|
err = errors.Join(err, otelShutdown(context.Background()))
|
|
}()
|
|
|
|
// Start HTTP server.
|
|
srv := &http.Server{
|
|
Addr: ":8080",
|
|
BaseContext: func(_ net.Listener) context.Context { return ctx },
|
|
ReadTimeout: time.Second,
|
|
WriteTimeout: 10 * time.Second,
|
|
Handler: newHTTPHandler(),
|
|
}
|
|
srvErr := make(chan error, 1)
|
|
go func() {
|
|
srvErr <- srv.ListenAndServe()
|
|
}()
|
|
|
|
// Wait for interruption.
|
|
select {
|
|
case err = <-srvErr:
|
|
// Error when starting HTTP server.
|
|
return
|
|
case <-ctx.Done():
|
|
// Wait for first CTRL+C.
|
|
// Stop receiving signal notifications as soon as possible.
|
|
stop()
|
|
}
|
|
|
|
// When Shutdown is called, ListenAndServe immediately returns ErrServerClosed.
|
|
err = srv.Shutdown(context.Background())
|
|
return
|
|
}
|
|
|
|
func newHTTPHandler() http.Handler {
|
|
mux := http.NewServeMux()
|
|
|
|
// handleFunc is a replacement for mux.HandleFunc
|
|
// which enriches the handler's HTTP instrumentation with the pattern as the http.route.
|
|
handleFunc := func(pattern string, handlerFunc func(http.ResponseWriter, *http.Request)) {
|
|
// Configure the "http.route" for the HTTP instrumentation.
|
|
handler := otelhttp.WithRouteTag(pattern, http.HandlerFunc(handlerFunc))
|
|
mux.Handle(pattern, handler)
|
|
}
|
|
|
|
// Register handlers.
|
|
handleFunc("/rolldice", rolldice)
|
|
|
|
// Add HTTP instrumentation for the whole server.
|
|
handler := otelhttp.NewHandler(mux, "/")
|
|
return handler
|
|
}
|