Skip to content

Generate trace ID in correct format (LocalLambda Server) #139

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 8 commits into from
Jul 8, 2020
Merged

Generate trace ID in correct format (LocalLambda Server) #139

Changes from 1 commit
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
372b748
Generate trace ID in correct format (LocalLambda Server)
Jul 2, 2020
76c7e25
Move randomTraceID to Utils, rename to generateTraceID
Jul 2, 2020
21aede0
Create generateTraceId test
Jul 2, 2020
ed017c2
Merge branch 'master' into feature/random_trace_id
tomerd Jul 2, 2020
a59feb5
Rename generateTraceID to AmazonHeaders::generateXRayTraceID
Jul 7, 2020
e990029
Generate XRay trace ID in LambdaRuntimeClientTest
Jul 7, 2020
ee79bfd
Rename testGenerateTraceId to match tested function name
Jul 7, 2020
cefb7bc
Remove @inlinable from generateXRayTraceID
Jul 8, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 18 additions & 1 deletion Sources/AWSLambdaRuntimeCore/Lambda+LocalServer.swift
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -276,7 +276,7 @@ private enum LocalLambda {
response.headers = [
(AmazonHeaders.requestID, self.requestID),
(AmazonHeaders.invokedFunctionARN, "arn:aws:lambda:us-east-1:\(Int16.random(in: Int16.min ... Int16.max)):function:custom-runtime"),
(AmazonHeaders.traceID, "Root=\(Int16.random(in: Int16.min ... Int16.max));Parent=\(Int16.random(in: Int16.min ... Int16.max));Sampled=1"),
(AmazonHeaders.traceID, "Root=\(randomTraceID());Sampled=1"),
(AmazonHeaders.deadline, "\(DispatchWallTime.distantFuture.millisSinceEpoch)"),
]
return response
Expand All @@ -294,5 +294,22 @@ private enum LocalLambda {
case notReady
case cantBind
}

static func randomTraceID() -> String {
// see:
// - [Generating trace IDs](https://docs.aws.amazon.com/xray/latest/devguide/xray-api-sendingdata.html#xray-api-traceids)
// - [Tracing header](https://docs.aws.amazon.com/xray/latest/devguide/xray-concepts.html#xray-concepts-tracingheader)
// The version number, that is, 1.
let version: UInt = 1
// The time of the original request, in Unix epoch time, in 8 hexadecimal digits.
let now = UInt32(DispatchWallTime.now().millisSinceEpoch / 1000)
let dateValue = String(now, radix: 16, uppercase: false)
let datePadding = String(repeating: "0", count: max(0, 8 - dateValue.count))
// A 96-bit identifier for the trace, globally unique, in 24 hexadecimal digits.
let identifier = String(UInt64.random(in: UInt64.min ... UInt64.max) | 1 << 63, radix: 16, uppercase: false)
+ String(UInt32.random(in: UInt32.min ... UInt32.max) | 1 << 31, radix: 16, uppercase: false)
return "\(version)-\(datePadding)\(dateValue)-\(identifier)"
}
}

#endif