-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.go
More file actions
117 lines (101 loc) · 4.09 KB
/
main.go
File metadata and controls
117 lines (101 loc) · 4.09 KB
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
// The randomizer-lambda command is an AWS Lambda handler that serves the Slack
// slash command API for the randomizer.
//
// The handler expects HTTP request events using the [Amazon API Gateway
// payload format version 2.0]. This makes it suitable for invocation through a
// [Lambda function URL], or through an AWS Lambda proxy integration in an
// Amazon API Gateway HTTP API.
//
// See the randomizer repository README for more information on configuring and
// deploying the randomizer on AWS Lambda.
//
// [Amazon API Gateway payload format version 2.0]: https://docs.aws.amazon.com/apigateway/latest/developerguide/http-api-develop-integrations-lambda.html
// [Lambda function URL]: https://docs.aws.amazon.com/lambda/latest/dg/lambda-urls.html
package main
import (
"context"
"log/slog"
"os"
"github.com/aws-observability/aws-otel-go/exporters/xrayudp"
"github.com/aws/aws-lambda-go/lambda"
"github.com/awslabs/aws-lambda-go-api-proxy/httpadapter"
lambdadetector "go.opentelemetry.io/contrib/detectors/aws/lambda"
"go.opentelemetry.io/contrib/instrumentation/github.com/aws/aws-lambda-go/otellambda"
"go.opentelemetry.io/contrib/instrumentation/github.com/aws/aws-lambda-go/otellambda/xrayconfig"
"go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp"
xraypropagator "go.opentelemetry.io/contrib/propagators/aws/xray"
"go.opentelemetry.io/otel"
"go.opentelemetry.io/otel/sdk/resource"
"go.opentelemetry.io/otel/sdk/trace"
semconv "go.opentelemetry.io/otel/semconv/v1.37.0"
"github.com/featherbread/randomizer/internal/slack"
"github.com/featherbread/randomizer/internal/store/dynamodb"
)
func main() {
ctx := context.Background()
logger := slog.New(slog.NewJSONHandler(os.Stderr, nil))
tokenProvider, err := slack.TokenProviderFromEnv()
if err != nil {
logger.Error("Failed to configure Slack token", "err", err)
os.Exit(2)
}
storeFactory, err := dynamodb.FactoryFromEnv(ctx)
if err != nil {
logger.Error("Failed to create DynamoDB store", "err", err)
os.Exit(2)
}
var otellambdaOptions []otellambda.Option
if xrayTracerProviderEnabled {
tp := initXRayTracerProvider(ctx, logger)
otel.SetTracerProvider(tp)
otel.SetTextMapPropagator(xraypropagator.Propagator{})
otellambdaOptions = xrayconfig.WithRecommendedOptions(tp)
defer func() {
err := tp.Shutdown(ctx)
if err != nil {
logger.Warn("Failed to shut down tracer provider", "err", err)
}
}()
}
app := slack.App{
TokenProvider: tokenProvider,
StoreFactory: storeFactory,
Logger: logger,
}
httpHandler := otelhttp.NewHandler(app, "/")
adapterHandler := httpadapter.NewV2(httpHandler).ProxyWithContext
parentHandler := otellambda.InstrumentHandler(adapterHandler, otellambdaOptions...)
lambda.Start(parentHandler)
}
// xrayTracerProviderEnabled indicates whether we should manually configure
// OpenTelemetry to export spans to Lambda's X-Ray UDP collector. This could be
// disabled if we don't want X-Ray tracing at all, or if we're configuring the
// Auto SDK via eBPF (e.g. ADOT Lambda layers).
var xrayTracerProviderEnabled = os.Getenv("AWS_XRAY_TRACER_PROVIDER_ENABLED") == "1"
func initXRayTracerProvider(ctx context.Context, logger *slog.Logger) *trace.TracerProvider {
tp := trace.NewTracerProvider(
trace.WithResource(initTraceResource(ctx, logger)))
exporter, err := xrayudp.NewSpanExporter(ctx)
if err != nil {
logger.Warn("Failed to initialize X-Ray span exporter", "err", err)
return tp
}
tp.RegisterSpanProcessor(trace.NewSimpleSpanProcessor(exporter))
return tp
}
func initTraceResource(ctx context.Context, logger *slog.Logger) *resource.Resource {
baseResource := resource.NewWithAttributes(
semconv.SchemaURL,
semconv.ServiceName(os.Getenv("AWS_LAMBDA_FUNCTION_NAME")))
lambdaResource, err := lambdadetector.NewResourceDetector().Detect(ctx)
if err != nil {
logger.Warn("Skipping Lambda resources in traces", "err", err, "step", "detect")
return baseResource
}
mergedResource, err := resource.Merge(lambdaResource, baseResource)
if err != nil {
logger.Warn("Skipping Lambda resources in traces", "err", err, "step", "merge")
return baseResource
}
return mergedResource
}