mirror of
https://github.com/swift-server/swift-aws-lambda-runtime.git
synced 2026-06-02 07:27:33 +00:00
Sébastien StormacqandSebastien Stormacq
e58d89148c
- Adjust notice, security reporting, code of conduct, contribution process to the standard AWS documents - Adjust GitHub issue templates to AWS standard ones. - Adjust the license header in all source files --------- Co-authored-by: Sebastien Stormacq <stormacq@amazon.lu>
62 lines
2.0 KiB
Swift
62 lines
2.0 KiB
Swift
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This source file is part of the SwiftAWSLambdaRuntime open source project
|
|
//
|
|
// Copyright SwiftAWSLambdaRuntime project authors
|
|
// Copyright (c) Amazon.com, Inc. or its affiliates.
|
|
// 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 AWSLambdaRuntime
|
|
import NIOCore
|
|
|
|
#if canImport(FoundationEssentials)
|
|
import FoundationEssentials
|
|
#else
|
|
import Foundation
|
|
#endif
|
|
|
|
struct SendNumbersWithPause: StreamingLambdaHandler {
|
|
func handle(
|
|
_ event: ByteBuffer,
|
|
responseWriter: some LambdaResponseStreamWriter,
|
|
context: LambdaContext
|
|
) async throws {
|
|
|
|
// Send HTTP status code and headers before streaming the response body
|
|
try await responseWriter.writeStatusAndHeaders(
|
|
StreamingLambdaStatusAndHeadersResponse(
|
|
statusCode: 418, // I'm a tea pot
|
|
headers: [
|
|
"Content-Type": "text/plain",
|
|
"x-my-custom-header": "streaming-example",
|
|
]
|
|
)
|
|
)
|
|
|
|
// Stream numbers with pauses to demonstrate streaming functionality
|
|
for i in 1...3 {
|
|
// Send partial data
|
|
try await responseWriter.write(ByteBuffer(string: "Number: \(i)\n"))
|
|
|
|
// Perform some long asynchronous work to simulate processing
|
|
try await Task.sleep(for: .milliseconds(1000))
|
|
}
|
|
|
|
// Send final message
|
|
try await responseWriter.write(ByteBuffer(string: "Streaming complete!\n"))
|
|
|
|
// All data has been sent. Close off the response stream.
|
|
try await responseWriter.finish()
|
|
}
|
|
}
|
|
|
|
let runtime = LambdaRuntime(handler: SendNumbersWithPause())
|
|
try await runtime.run()
|