Support container based lambdas by optionally providing image_uri or s3_bucket and s3_key

Author: jheister

main.tf

@@ -3,8 +3,9 @@ terraform {
 }
 
 resource "aws_lambda_function" "lambda_function" {
-  s3_bucket                      = var.s3_bucket
-  s3_key                         = var.s3_key
+  image_uri                      = var.image_uri != "" ? var.image_uri : null
+  s3_bucket                      = var.s3_bucket != "" ? var.s3_bucket : null
+  s3_key                         = var.s3_key != "" ? var.s3_key : null
   function_name                  = var.function_name
   role                           = aws_iam_role.iam_for_lambda.arn
   handler                        = var.handler

variables.tf

@@ -1,9 +1,16 @@
+variable "image_uri" {
+  description = "Optional ECR image (for image based lambda)"
+  default     = ""
+}
+
 variable "s3_bucket" {
   description = "The name of the bucket containing your uploaded Lambda deployment package."
+  default     = ""
 }
 
 variable "s3_key" {
   description = "The s3 key for your Lambda deployment package."
+  default     = ""
 }
 
 variable "function_name" {