Move trace logic to separate package

ddlambda.go

@@ -4,7 +4,7 @@ import (
 	"context"
 	"encoding/json"
 
-	"github.com/DataDog/dd-lambda-go/internal"
+	"github.com/DataDog/dd-lambda-go/internal/trace"
 )
 
 type (
@@ -15,17 +15,17 @@ type (
 // It returns a modified handler that can be passed directly to the lambda.Start function.
 func WrapHandler(handler interface{}) interface{} {
 	hl := handlerListener{}
-	return internal.WrapHandlerWithListener(handler, &hl)
+	return trace.WrapHandlerWithListener(handler, &hl)
 }
 
 // GetTraceHeaders reads a map containing the DataDog trace headers from a context object.
 func GetTraceHeaders(ctx context.Context) map[string]string {
-	result := internal.GetTraceHeaders(ctx, true)
+	result := trace.GetTraceHeaders(ctx, true)
 	return result
 }
 
 func (hl *handlerListener) HandlerStarted(ctx context.Context, msg json.RawMessage) context.Context {
-	ctx, _ = internal.ExtractTraceContext(ctx, msg)
+	ctx, _ = trace.ExtractTraceContext(ctx, msg)
 	return ctx
 }
 

internal/constants.go → internal/trace/constants.go

@@ -1,4 +1,4 @@
-package internal
+package trace
 
 const (
 	traceIDHeader          = "x-datadog-trace-id"

internal/trace_context.go → internal/trace/trace_context.go

@@ -1,4 +1,4 @@
-package internal
+package trace
 
 import (
 	"context"

internal/trace_context_test.go → internal/trace/trace_context_test.go

@@ -1,4 +1,4 @@
-package internal
+package trace
 
 import (
 	"context"

internal/wrap_handler.go → internal/trace/wrap_handler.go

@@ -1,4 +1,4 @@
-package internal
+package trace
 
 import (
 	"context"

internal/wrap_handler_test.go → internal/trace/wrap_handler_test.go

@@ -1,4 +1,4 @@
-package internal
+package trace
 
 import (
 	"context"