Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
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
44 changes: 30 additions & 14 deletions Sources/AWSLambdaRuntimeCore/ControlPlaneRequest.swift
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
//
// This source file is part of the SwiftAWSLambdaRuntime open source project
//
// Copyright (c) 2017-2021 Apple Inc. and the SwiftAWSLambdaRuntime project authors
// Copyright (c) 2021-2022 Apple Inc. and the SwiftAWSLambdaRuntime project authors
// Licensed under Apache License v2.0
//
// See LICENSE.txt for license information
Expand All @@ -29,20 +29,34 @@ enum ControlPlaneResponse: Hashable {
}

struct Invocation: Hashable {
let requestID: String
let deadlineInMillisSinceEpoch: Int64
let invokedFunctionARN: String
let traceID: String
let clientContext: String?
let cognitoIdentity: String?
var requestID: String
var deadlineInMillisSinceEpoch: UInt64
var invokedFunctionARN: String
var traceID: String
var clientContext: String?
var cognitoIdentity: String?

init(requestID: String,
deadlineInMillisSinceEpoch: UInt64,
invokedFunctionARN: String,
traceID: String,
clientContext: String?,
cognitoIdentity: String?) {
self.requestID = requestID
self.deadlineInMillisSinceEpoch = deadlineInMillisSinceEpoch
self.invokedFunctionARN = invokedFunctionARN
self.traceID = traceID
self.clientContext = clientContext
self.cognitoIdentity = cognitoIdentity
}

init(headers: HTTPHeaders) throws {
guard let requestID = headers.first(name: AmazonHeaders.requestID), !requestID.isEmpty else {
throw Lambda.RuntimeError.invocationMissingHeader(AmazonHeaders.requestID)
}

guard let deadline = headers.first(name: AmazonHeaders.deadline),
let unixTimeInMilliseconds = Int64(deadline)
let unixTimeInMilliseconds = UInt64(deadline)
else {
throw Lambda.RuntimeError.invocationMissingHeader(AmazonHeaders.deadline)
}
Expand All @@ -55,12 +69,14 @@ struct Invocation: Hashable {
throw Lambda.RuntimeError.invocationMissingHeader(AmazonHeaders.traceID)
}

self.requestID = requestID
self.deadlineInMillisSinceEpoch = unixTimeInMilliseconds
self.invokedFunctionARN = invokedFunctionARN
self.traceID = traceID
self.clientContext = headers["Lambda-Runtime-Client-Context"].first
self.cognitoIdentity = headers["Lambda-Runtime-Cognito-Identity"].first
self.init(
requestID: requestID,
deadlineInMillisSinceEpoch: unixTimeInMilliseconds,
invokedFunctionARN: invokedFunctionARN,
traceID: traceID,
clientContext: headers["Lambda-Runtime-Client-Context"].first,
cognitoIdentity: headers["Lambda-Runtime-Cognito-Identity"].first
)
}
}

Expand Down
Loading