From 70e164760615ea57b3f40ed162d0a56884e22f21 Mon Sep 17 00:00:00 2001
From: Fabian Fett <fabianfett@mac.com>
Date: Thu, 5 Mar 2020 14:50:15 +0100
Subject: [PATCH 1/2] =?UTF-8?q?Don=E2=80=99t=20create=20an=20EventLoopFutu?=
 =?UTF-8?q?re=20chain?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Sources/SwiftAwsLambda/Lambda.swift | 46 +++++++++++++++++++----------
 1 file changed, 30 insertions(+), 16 deletions(-)

diff --git a/Sources/SwiftAwsLambda/Lambda.swift b/Sources/SwiftAwsLambda/Lambda.swift
index ef55e6056..873d87711 100644
--- a/Sources/SwiftAwsLambda/Lambda.swift
+++ b/Sources/SwiftAwsLambda/Lambda.swift
@@ -119,7 +119,7 @@ public enum Lambda {
             let runner = LambdaRunner(eventLoop: self.eventLoop, configuration: self.configuration, lambdaHandler: self.handler)
             return runner.initialize(logger: logger).flatMap { _ in
                 self.state = .active
-                return self.run(logger: logger, runner: runner, count: 0)
+                return self.run(runner: runner)
             }
         }
 
@@ -133,23 +133,37 @@ public enum Lambda {
             self.state = .shutdown
         }
 
-        private func run(logger: Logger, runner: LambdaRunner, count: Int) -> EventLoopFuture<Int> {
-            switch self.state {
-            case .active:
-                if self.configuration.lifecycle.maxTimes > 0, count >= self.configuration.lifecycle.maxTimes {
-                    return self.eventLoop.makeSucceededFuture(count)
+        @inline(__always)
+        private func run(runner: LambdaRunner) -> EventLoopFuture<Int> {
+            let promise = self.eventLoop.makePromise(of: Int.self)
+
+            func _run(_ count: Int) {
+                switch self.state {
+                case .active:
+                    if self.configuration.lifecycle.maxTimes > 0, count >= self.configuration.lifecycle.maxTimes {
+                        return promise.succeed(count)
+                    }
+                    var logger = self.logger
+                    logger[metadataKey: "lifecycleIteration"] = "\(count)"
+                    runner.run(logger: logger).whenComplete { result in
+                        switch result {
+                        case .success:
+                            // recursive! per aws lambda runtime spec the polling requests are to be done one at a time
+                            _run(count + 1)
+                        case .failure(let error):
+                            promise.fail(error)
+                        }
+                    }
+                case .stopping, .shutdown:
+                    promise.succeed(count)
+                default:
+                    preconditionFailure("invalid run state: \(self.state)")
                 }
-                var logger = logger
-                logger[metadataKey: "lifecycleIteration"] = "\(count)"
-                return runner.run(logger: logger).flatMap { _ in
-                    // recursive! per aws lambda runtime spec the polling requests are to be done one at a time
-                    self.run(logger: logger, runner: runner, count: count + 1)
-                }
-            case .stopping, .shutdown:
-                return self.eventLoop.makeSucceededFuture(count)
-            default:
-                preconditionFailure("invalid run state: \(self.state)")
             }
+
+            _run(0)
+
+            return promise.futureResult
         }
     }
 

From 8da62ce785d7bd7110b4ca30c6e00b313c752746 Mon Sep 17 00:00:00 2001
From: Fabian Fett <fabianfett@mac.com>
Date: Fri, 6 Mar 2020 08:36:23 +0100
Subject: [PATCH 2/2] Fix failing Unit-Test

---
 Tests/SwiftAwsLambdaTests/LambdaTest.swift | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Tests/SwiftAwsLambdaTests/LambdaTest.swift b/Tests/SwiftAwsLambdaTests/LambdaTest.swift
index 114f1d5b7..25e702abc 100644
--- a/Tests/SwiftAwsLambdaTests/LambdaTest.swift
+++ b/Tests/SwiftAwsLambdaTests/LambdaTest.swift
@@ -79,7 +79,7 @@ class LambdaTest: XCTestCase {
             }
         }
         let signal = Signal.ALRM
-        let maxTimes = 50
+        let maxTimes = 1000
         let configuration = Lambda.Configuration(lifecycle: .init(maxTimes: maxTimes, stopSignal: signal))
         let eventLoopGroup = MultiThreadedEventLoopGroup(numberOfThreads: System.coreCount)
         let future = Lambda.runAsync(eventLoopGroup: eventLoopGroup, handler: MyHandler(), configuration: configuration)