Fix: Random Crash tonisuter#9

Andrea-Scuderi · Aug 4, 2019 · 8f07caf · 8f07caf
1 parent 3f7d56a
commit 8f07caf
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/Examples/SquareNumber/Sources/SquareNumber/main.swift b/Examples/SquareNumber/Sources/SquareNumber/main.swift
@@ -12,7 +12,10 @@ func squareNumber(input: Input, context: Context) -> Output {
     let squaredNumber = input.number * input.number
     return Output(result: squaredNumber)
 }
-
-let runtime = try Runtime()
-runtime.registerLambda("squareNumber", handlerFunction: squareNumber)
-try runtime.start()
+do {
+    let runtime = try Runtime()
+    runtime.registerLambda("squareNumber", handlerFunction: squareNumber)
+    try runtime.start()
+} catch (let error) {
+    log(error)
+}
diff --git a/Sources/AWSLambdaSwift/Runtime.swift b/Sources/AWSLambdaSwift/Runtime.swift
@@ -21,7 +21,12 @@ public class Runtime {
     var handlers: [String: Handler]
 
     public init() throws {
-        self.urlSession = URLSession(configuration: .default)
+
+        let configuration = URLSessionConfiguration.default
+        configuration.timeoutIntervalForRequest = 3600
+
+
+        self.urlSession = URLSession(configuration: configuration)
         self.handlers = [:]
 
         let environment = ProcessInfo.processInfo.environment