RUST-1253 Add AWS Lambda Examples

.evergreen/check-clippy.sh

@@ -2,7 +2,7 @@
 
 set -o errexit
 
-source ./.evergreen/configure-rust.sh
+# source ./.evergreen/configure-rust.sh
 source ./.evergreen/feature-combinations.sh
 
 # Pin clippy to the latest version. This should be updated when new versions of Rust are released.

.evergreen/run-aws-tests.sh

@@ -45,3 +45,4 @@ set -o errexit
 source ./.evergreen/configure-rust.sh
 
 RUST_BACKTRACE=1 cargo test --features aws-auth auth_aws::auth_aws
+RUST_BACKTRACE=1 cargo test --features aws-auth lambda_example::test_handler

Cargo.toml

@@ -162,6 +162,7 @@ derive_more = "0.99.13"
 function_name = "0.2.1"
 futures = "0.3"
 home = "0.5"
+lambda_runtime = "0.6.0"
 pretty_assertions = "1.1.0"
 serde_json = "1.0.64"
 semver = "1.0.0"

src/test/lambda_examples.rs

@@ -0,0 +1,5 @@
+#![allow(dead_code)]
+
+#[cfg(feature = "aws-auth")]
+mod auth;
+mod no_auth;

src/test/lambda_examples/auth.rs

@@ -0,0 +1,57 @@
+use crate as mongodb;
+
+// begin lambda connection example 2
+use async_once::AsyncOnce;
+use lambda_runtime::{service_fn, LambdaEvent};
+use lazy_static::lazy_static;
+use mongodb::{
+    bson::doc,
+    options::{AuthMechanism, ClientOptions, Credential},
+    Client,
+};
+use serde_json::Value;
+
+// Initialize a global static MongoDB Client with AWS authentication.
+//
+// The client can be accessed as follows:
+// let client = MONGODB_CLIENT.get().await;
+lazy_static! {
+    static ref MONGODB_CLIENT: AsyncOnce<Client> = AsyncOnce::new(async {
+        let uri = std::env::var("MONGODB_URI")
+            .expect("MONGODB_URI must be set to the URI of the MongoDB deployment");
+        let mut options = ClientOptions::parse(uri)
+            .await
+            .expect("Failed to parse options from URI");
+        let credential = Credential::builder()
+            .mechanism(AuthMechanism::MongoDbAws)
+            .build();
+        options.credential = Some(credential);
+        Client::with_options(options).expect("Failed to create MongoDB Client")
+    });
+}
+
+// Runs a ping operation on the "db" database and returns the response.
+async fn handler(_: LambdaEvent<Value>) -> Result<Value, lambda_runtime::Error> {
+    let client = MONGODB_CLIENT.get().await;
+    let response = client
+        .database("db")
+        .run_command(doc! { "ping": 1 }, None)
+        .await?;
+    let json = serde_json::to_value(response)?;
+    Ok(json)
+}
+
+#[tokio::main]
+async fn main() -> Result<(), lambda_runtime::Error> {
+    let service = service_fn(handler);
+    lambda_runtime::run(service).await?;
+    Ok(())
+}
+
+#[cfg_attr(feature = "tokio-runtime", tokio::test(flavor = "multi_thread"))]
+#[cfg_attr(feature = "async-std-runtime", async_std::test)]
+async fn test_handler() {
+    let event = LambdaEvent::new(Value::Null, Default::default());
+    handler(event).await.unwrap();
+}
+// end lambda connection example 2

src/test/lambda_examples/no_auth.rs

@@ -0,0 +1,48 @@
+use crate as mongodb;
+
+// begin lambda connection example 1
+use async_once::AsyncOnce;
+use lambda_runtime::{service_fn, LambdaEvent};
+use lazy_static::lazy_static;
+use mongodb::{bson::doc, Client};
+use serde_json::Value;
+
+// Initialize a global static MongoDB Client.
+//
+// The client can be accessed as follows:
+// let client = MONGODB_CLIENT.get().await;
+lazy_static! {
+    static ref MONGODB_CLIENT: AsyncOnce<Client> = AsyncOnce::new(async {
+        let uri = std::env::var("MONGODB_URI")
+            .expect("MONGODB_URI must be set to the URI of the MongoDB deployment");
+        Client::with_uri_str(uri)
+            .await
+            .expect("Failed to create MongoDB Client")
+    });
+}
+
+// Runs a ping operation on the "db" database and returns the response.
+async fn handler(_: LambdaEvent<Value>) -> Result<Value, lambda_runtime::Error> {
+    let client = MONGODB_CLIENT.get().await;
+    let response = client
+        .database("db")
+        .run_command(doc! { "ping": 1 }, None)
+        .await?;
+    let json = serde_json::to_value(response)?;
+    Ok(json)
+}
+
+#[tokio::main]
+async fn main() -> Result<(), lambda_runtime::Error> {
+    let service = service_fn(handler);
+    lambda_runtime::run(service).await?;
+    Ok(())
+}
+// end lambda connection example 1
+
+#[cfg_attr(feature = "tokio-runtime", tokio::test(flavor = "multi_thread"))]
+#[cfg_attr(feature = "async-std-runtime", async_std::test)]
+async fn test_handler() {
+    let event = LambdaEvent::new(Value::Null, Default::default());
+    handler(event).await.unwrap();
+}

src/test/mod.rs

@@ -10,6 +10,8 @@ mod db;
 #[cfg(all(not(feature = "sync"), not(feature = "tokio-sync")))]
 mod documentation_examples;
 mod index_management;
+#[cfg(all(not(feature = "sync"), not(feature = "tokio-sync")))]
+mod lambda_examples;
 pub mod spec;
 mod util;