Refactor ip and port extraction

aryan-25 · aryan-25 · commit aed1a8ec546f · 2024-09-05T11:13:28.000+01:00
diff --git a/Sources/AWSLambdaRuntimeCore/NewLambdaRuntime.swift b/Sources/AWSLambdaRuntimeCore/NewLambdaRuntime.swift
@@ -40,7 +40,7 @@ package final class NewLambdaRuntime<Handler>: Sendable where Handler: Streaming
 
         let ipAndPort = runtimeEndpoint.split(separator: ":", maxSplits: 1)
         let ip = String(ipAndPort[0])
-        let port = Int(ipAndPort[1])!
+        guard let port = Int(ipAndPort[1]) else { throw NewLambdaRuntimeError(code: .invalidPort) }
 
         let handler = self.handlerMutex.withLock { maybeHandler in
             defer {
diff --git a/Sources/AWSLambdaRuntimeCore/NewLambdaRuntimeError.swift b/Sources/AWSLambdaRuntimeCore/NewLambdaRuntimeError.swift
@@ -31,6 +31,7 @@ package struct NewLambdaRuntimeError: Error {
 
         case cannotStartLambdaRuntime
         case runtimeCanOnlyBeStartedOnce
+        case invalidPort
     }
 
     package init(code: Code, underlying: (any Error)? = nil) {

Original file line number	Diff line number	Diff line change
`@@ -31,6 +31,7 @@ package struct NewLambdaRuntimeError: Error {`
`31`	`31`
`32`	`32`	`case cannotStartLambdaRuntime`
`33`	`33`	`case runtimeCanOnlyBeStartedOnce`
	`34`	`+ case invalidPort`
`34`	`35`	`}`
`35`	`36`
`36`	`37`	`package init(code: Code, underlying: (any Error)? = nil) {`