Use Api gateway v2

SkyTruth · May 4, 2022 · fad6c04 · fad6c04
1 parent 3305b9e
commit fad6c04
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 1 deletion.
diff --git a/stack/__main__.py b/stack/__main__.py
@@ -6,4 +6,4 @@
 
 # Export the DNS name of the bucket
 pulumi.export("bucket_name", infra.bucket.url)
-pulumi.export("titiler_sentinel_url", titiler_sentinel.lambda_deployment.invoke_url)
+pulumi.export("titiler_sentinel_url", titiler_sentinel.lambda_api.api_endpoint)
diff --git a/stack/titiler_sentinel.py b/stack/titiler_sentinel.py
@@ -116,6 +116,28 @@ def create_package(code_dir: str) -> pulumi.FileArchive:
     principal="apigateway.amazonaws.com",
     function=lambda_titiler_sentinel,
 )
+lambda_api = aws.apigatewayv2.Api(
+    construct_name("lambda-titiler-api"), protocol_type="HTTP"
+)
+lambda_integration = aws.apigatewayv2.Integration(
+    construct_name("lambda-titiler-integration"),
+    api_id=lambda_api.id,
+    integration_type="AWS_PROXY",
+)
+lambda_route = aws.apigatewayv2.Route(
+    construct_name("lambda-titiler-route"),
+    api_id=lambda_api.id,
+    route_key="ANY /{proxy+}",
+    target=f"integrations/{lambda_integration.id}",
+)
+lambda_stage = aws.apigatewayv2.Stage(
+    construct_name("lambda-titiler-stage"),
+    api_id=lambda_api.id,
+    name="$default",
+    auto_deploy=True,
+    opts=pulumi.ResourceOptions(depends_on=[lambda_route]),
+)
+"""
 lambda_rest_api = aws.apigateway.RestApi(
     construct_name("lambda-titiler-rest-api"), name="Titiler Sentinel"
 )
@@ -153,3 +175,4 @@ def create_package(code_dir: str) -> pulumi.FileArchive:
     rest_api=lambda_rest_api.id,
     stage_name="default",
 )
+"""