1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

use aws_lambda_events::apigw::{ApiGatewayProxyRequest, ApiGatewayProxyResponse};
use futures::FutureExt;
use lambda_runtime::{service_fn, LambdaEvent};

use std::future::Future;

/// Start the Lambda runtime to handle requests for the specified API using the specified
/// middleware.
///
/// # Example
///
/// ```rust,ignore
/// // Replace `my_api` with the name of your crate and `backend` with the name of the module
/// // passed to `ApiLambda::new()`.
/// use my_api::backend::Api;
/// use my_api::backend_handler::BackendApiHandler;
/// use openapi_lambda::run_lambda;
///
/// #[tokio::main]
/// pub async fn main() {
///   let api = BackendApiHandler::new(...);
///   let middleware = ...; // Instantiate your middleware here.
///
///   run_lambda(|event| api.dispatch_request(event, &middleware)).await
/// }
/// ```
pub async fn run_lambda<F, Fut>(mut dispatch_event: F)
where
  F: FnMut(LambdaEvent<ApiGatewayProxyRequest>) -> Fut,
  Fut: Future<Output = ApiGatewayProxyResponse>,
{
  lambda_runtime::run(service_fn(|event: LambdaEvent<ApiGatewayProxyRequest>| {
    dispatch_event(event).map(Result::<_, std::convert::Infallible>::Ok)
  }))
  .await
  .expect("Lambda run loop should never exit")
}