Added lambda module

Author: Emmy-github-webdev

terraform/main.tf

@@ -49,4 +49,16 @@ module "api_gateway" {
   environment       = var.environment
   lambda_invoke_arn = module.lambda.function_invoke_arn
   common_tags       = local.common_tags
+}
+
+# Lambda Module
+module "lambda" {
+  source = "./modules/lambda"
+  environment                = var.environment
+  lambda_role_arn            = module.iam.lambda_role_arn
+  dynamodb_table_name        = module.dynamodb.table_name
+  api_gateway_execution_arn  = module.api_gateway.execution_arn
+  lambda_funtion_dir          = var.lambda_funtion_dir
+  common_tags                = local.common_tags
+  depends_on = [module.api_gateway]
 }

terraform/modules/lambda/main.tf

@@ -0,0 +1,33 @@
+data "archive_file" "lambda_zip" {
+  type        = "zip"
+  source_dir  = var.lambda_funtion_dir
+  output_path = "${path.module}/${var.environment}_lambda_function.zip"
+}
+
+resource "aws_lambda_function" "health_check" {
+  function_name    = "${var.environment}-health-check-function"
+  role             = var.lambda_role_arn
+  handler          = "lambda_function.lambda_handler"
+  source_code_hash = data.archive_file.lambda_zip.output_base64sha256
+  runtime          = "python3.11"
+  filename         = data.archive_file.lambda_zip.output_path
+
+  environment {
+    variables = {
+      DYNAMODB_TABLE_NAME = var.dynamodb_table_name
+      ENVIRONMENT         = var.environment
+    }
+  }
+
+  tags = var.common_tags
+
+  depends_on = [data.archive_file.lambda_zip]
+}
+
+resource "aws_lambda_permission" "api_gateway" {
+  statement_id  = "AllowAPIGatewayInvoke"
+  action        = "lambda:InvokeFunction"
+  function_name = aws_lambda_function.health_check.function_name
+  principal     = "apigateway.amazonaws.com"
+  source_arn    = "${var.api_gateway_execution_arn}/*/*"
+}

terraform/modules/lambda/outputs.tf

@@ -0,0 +1,14 @@
+output "function_arn" {
+  description = "ARN of the Lambda function"
+  value       = aws_lambda_function.health_check.arn
+}
+
+output "function_name" {
+  description = "Name of the Lambda function"
+  value       = aws_lambda_function.health_check.function_name
+}
+
+output "function_invoke_arn" {
+  description = "Invoke ARN of the Lambda function"
+  value       = aws_lambda_function.health_check.invoke_arn
+}

terraform/modules/lambda/variables.tf

@@ -0,0 +1,29 @@
+variable "environment" {
+  description = "Environment name"
+  type        = string
+}
+
+variable "lambda_role_arn" {
+  description = "ARN of the IAM role for Lambda"
+  type        = string
+}
+
+variable "dynamodb_table_name" {
+  description = "Name of the DynamoDB table"
+  type        = string
+}
+
+variable "api_gateway_execution_arn" {
+  description = "Execution ARN of the API Gateway"
+  type        = string
+}
+
+variable "lambda_funtion_dir" {
+  description = "Path to the Lambda function source code directory"
+  type        = string
+}
+
+variable "common_tags" {
+  description = "Common tags for all resources"
+  type        = map(string)
+}