add use_default_security_group

mergermarket · Aug 21, 2024 · 10e967e · 10e967e
1 parent adfba38
commit 10e967e
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/main.tf b/main.tf
@@ -1,10 +1,11 @@
 locals {
-  security_group_ids = var.use_default_sg == true ? [data.aws_security_group.default[0].id] : var.security_group_ids
+  security_group_ids = var.use_default_security_group == true ? [data.aws_security_group.default[0].id] : var.security_group_ids
 }
 
 data "aws_security_group" "default" {
   count = var.use_default_sg == true ? 1 : 0
   name = "${terraform.workspace}-default-lambda-sg"
+  vpc_id = var.vpc_id
 }
 resource "aws_lambda_function" "lambda_function" {
   s3_bucket     = var.s3_bucket

diff --git a/variables.tf b/variables.tf
@@ -108,4 +108,15 @@ variable "architectures" {
 variable "use_default_sg" {
   description = "Whether to use the default security group for the Lambda function."
   default     = false
+}
+
+variable "vpc_id" {
+  description = "The VPC ID in which the Lambda runs"
+  default     = ""
+}
+
+variable "use_default_security_group" {
+  type = bool
+  description = "Whether to use the default security group for the Lambda function."
+  default = false
 }