Merge pull request #210 from sachinh/sachinh-feature-integration-msk-to-lambda-rust

yualexan · web-flow · commit 62f54f12fb96 · 2025-01-10T09:07:19.000-08:00
Added Rust implementation for msk-to-lambda integration
diff --git a/integration-msk-to-lambda/main.rs b/integration-msk-to-lambda/main.rs
@@ -0,0 +1,48 @@
+use aws_lambda_events::event::kafka::KafkaEvent;
+use lambda_runtime::{run, service_fn, tracing, Error, LambdaEvent};
+use base64::prelude::*;
+use serde_json::{Value};
+use tracing::{info};
+
+/// Pre-Requisites:
+/// 1. Install Cargo Lambda - see https://www.cargo-lambda.info/guide/getting-started.html
+/// 2. Add packages tracing, tracing-subscriber, serde_json, base64
+///
+/// This is the main body for the function.
+/// Write your code inside it.
+/// There are some code example in the following URLs:
+/// - https://github.com/awslabs/aws-lambda-rust-runtime/tree/main/examples
+/// - https://github.com/aws-samples/serverless-rust-demo/
+
+async fn function_handler(event: LambdaEvent<KafkaEvent>) -> Result<Value, Error> {
+
+    let payload = event.payload.records;
+
+    for (_name, records) in payload.iter() {
+
+        for record in records {
+
+         let record_text = record.value.as_ref().ok_or("Value is None")?;
+         info!("Record: {}", &record_text);
+
+         // perform Base64 decoding
+         let record_bytes = BASE64_STANDARD.decode(record_text)?;
+         let message = std::str::from_utf8(&record_bytes)?;
+         
+         info!("Message: {}", message);
+        }
+
+    }
+
+    Ok(().into())
+}
+
+#[tokio::main]
+async fn main() -> Result<(), Error> {
+
+    // required to enable CloudWatch error logging by the runtime
+    tracing::init_default_subscriber();
+    info!("Setup CW subscriber!");
+
+    run(service_fn(function_handler)).await
+}
