mirror of
https://github.com/swift-server/swift-aws-lambda-runtime.git
synced 2026-06-02 07:27:33 +00:00
Sébastien Stormacq
d42ae6975e
- Use the new Swift 6 `@available` macro to remove requirement on `.platform` in Package.swift. - DRY: define the swift settings once for all in `Package.swift` ### Motivation: - Remove the requirement to build on macOS 15 in `Package.swift`. This allows library builders and end users to be more flexible on their dependency requirements. - The code is optionally compiled on macOS 15 and Linux, but SPM don't enforce it anymore. - Avoid repeating ourself. Be sure the same settings are applied on all targets. ### Modifications: - Create a `var swiftSetting: [SwiftSettings]` and reuse it for all targets. - Use `AvailabilityMacro=LambdaSwift 2.0:macOS 15.0` - Add this on top of the majority struct / classes ```swift #if swift(>=6.1) @available(LambdaSwift 2.0, *) #endif ``` ### Result: When using Swift 6.1, there is no more SPM dependency on macOS 15
112 lines
3.9 KiB
Swift
112 lines
3.9 KiB
Swift
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This source file is part of the SwiftAWSLambdaRuntime open source project
|
|
//
|
|
// Copyright (c) 2017-2018 Apple Inc. and the SwiftAWSLambdaRuntime project authors
|
|
// Licensed under Apache License v2.0
|
|
//
|
|
// See LICENSE.txt for license information
|
|
// See CONTRIBUTORS.txt for the list of SwiftAWSLambdaRuntime project authors
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
import Dispatch
|
|
import Logging
|
|
import NIOCore
|
|
import NIOPosix
|
|
|
|
#if os(macOS)
|
|
import Darwin.C
|
|
#elseif canImport(Glibc)
|
|
import Glibc
|
|
#elseif canImport(Musl)
|
|
import Musl
|
|
#elseif os(Windows)
|
|
import ucrt
|
|
#else
|
|
#error("Unsupported platform")
|
|
#endif
|
|
|
|
@available(LambdaSwift 2.0, *)
|
|
public enum Lambda {
|
|
@inlinable
|
|
package static func runLoop<RuntimeClient: LambdaRuntimeClientProtocol, Handler>(
|
|
runtimeClient: RuntimeClient,
|
|
handler: Handler,
|
|
logger: Logger
|
|
) async throws where Handler: StreamingLambdaHandler {
|
|
var handler = handler
|
|
|
|
var logger = logger
|
|
do {
|
|
while !Task.isCancelled {
|
|
|
|
logger.trace("Waiting for next invocation")
|
|
let (invocation, writer) = try await runtimeClient.nextInvocation()
|
|
logger[metadataKey: "aws-request-id"] = "\(invocation.metadata.requestID)"
|
|
|
|
// when log level is trace or lower, print the first Kb of the payload
|
|
let bytes = invocation.event
|
|
let maxPayloadPreviewSize = 1024
|
|
var metadata: Logger.Metadata? = nil
|
|
if logger.logLevel <= .trace,
|
|
let buffer = bytes.getSlice(at: 0, length: min(bytes.readableBytes, maxPayloadPreviewSize))
|
|
{
|
|
metadata = [
|
|
"Event's first bytes": .string(
|
|
String(buffer: buffer) + (bytes.readableBytes > maxPayloadPreviewSize ? "..." : "")
|
|
)
|
|
]
|
|
}
|
|
logger.trace(
|
|
"Sending invocation event to lambda handler",
|
|
metadata: metadata
|
|
)
|
|
|
|
do {
|
|
try await handler.handle(
|
|
invocation.event,
|
|
responseWriter: writer,
|
|
context: LambdaContext(
|
|
requestID: invocation.metadata.requestID,
|
|
traceID: invocation.metadata.traceID,
|
|
invokedFunctionARN: invocation.metadata.invokedFunctionARN,
|
|
deadline: LambdaClock.Instant(
|
|
millisecondsSinceEpoch: invocation.metadata.deadlineInMillisSinceEpoch
|
|
),
|
|
logger: logger
|
|
)
|
|
)
|
|
logger.trace("Handler finished processing invocation")
|
|
} catch {
|
|
logger.trace("Handler failed processing invocation", metadata: ["Handler error": "\(error)"])
|
|
try await writer.reportError(error)
|
|
continue
|
|
}
|
|
logger.handler.metadata.removeValue(forKey: "aws-request-id")
|
|
}
|
|
} catch is CancellationError {
|
|
// don't allow cancellation error to propagate further
|
|
}
|
|
|
|
}
|
|
|
|
/// The default EventLoop the Lambda is scheduled on.
|
|
public static let defaultEventLoop: any EventLoop = NIOSingletons.posixEventLoopGroup.next()
|
|
}
|
|
|
|
// MARK: - Public API
|
|
|
|
@available(LambdaSwift 2.0, *)
|
|
extension Lambda {
|
|
/// Utility to access/read environment variables
|
|
public static func env(_ name: String) -> String? {
|
|
guard let value = getenv(name) else {
|
|
return nil
|
|
}
|
|
return String(cString: value)
|
|
}
|
|
}
|