Fix extension compatibility issues with AWS Lambda Runtime Interface Emulator (#879)

* fix: consider status codes 200-299 successful in Extension API

* fix: Make `tracing` JSON field optional

Author: rebu-dt

lambda-extension/src/events.rs

@@ -1,7 +1,7 @@
 use serde::Deserialize;
 
 /// Request tracing information
-#[derive(Debug, Deserialize)]
+#[derive(Debug, Default, Deserialize)]
 #[serde(rename_all = "camelCase")]
 pub struct Tracing {
     /// The type of tracing exposed to the extension
@@ -20,6 +20,7 @@ pub struct InvokeEvent {
     /// The function's Amazon Resource Name
     pub invoked_function_arn: String,
     /// The request tracing information
+    #[serde(default)]
     pub tracing: Tracing,
 }
 

lambda-extension/src/extension.rs

@@ -272,7 +272,7 @@ where
                 self.log_port_number,
             )?;
             let res = client.call(req).await?;
-            if res.status() != http::StatusCode::OK {
+            if !res.status().is_success() {
                 let err = format!("unable to initialize the logs api: {}", res.status());
                 return Err(ExtensionError::boxed(err));
             }
@@ -318,7 +318,7 @@ where
                 self.telemetry_port_number,
             )?;
             let res = client.call(req).await?;
-            if res.status() != http::StatusCode::OK {
+            if !res.status().is_success() {
                 let err = format!("unable to initialize the telemetry api: {}", res.status());
                 return Err(ExtensionError::boxed(err));
             }
@@ -491,7 +491,7 @@ async fn register<'a>(
 
     let req = requests::register_request(&name, events)?;
     let res = client.call(req).await?;
-    if res.status() != http::StatusCode::OK {
+    if !res.status().is_success() {
         let err = format!("unable to register the extension: {}", res.status());
         return Err(ExtensionError::boxed(err));
     }