Provisioning database using flag

Provisioning database using flag

README.md

@@ -11,7 +11,7 @@
 
 | Name | Version |
 |------|---------|
-| <a name="provider_aws"></a> [aws](#provider\_aws) | >= 4.0 |
+| <a name="provider_aws"></a> [aws](#provider\_aws) | 5.10.0 |
 
 ## Modules
 
@@ -31,6 +31,7 @@
 
 | Name | Description | Type | Default | Required |
 |------|-------------|------|---------|:--------:|
+| <a name="input_create_database"></a> [create\_database](#input\_create\_database) | Flag to create database or not | `bool` | `true` | no |
 | <a name="input_enabled"></a> [enabled](#input\_enabled) | Defines whether this module should create resources | `bool` | `true` | no |
 | <a name="input_invoke"></a> [invoke](#input\_invoke) | Defines whether lambda function should be invoked immediately after provisioning | `bool` | `true` | no |
 | <a name="input_lambda_name"></a> [lambda\_name](#input\_lambda\_name) | value of lambda name | `string` | n/a | yes |

lambda/index.py

@@ -64,6 +64,12 @@ def provision_db_and_user(master_secrets_json, secret_json):
         conn.autocommit = True
         cursor = conn.cursor()
 
+        # Create database
+        create_database = os.environ['CREATE_DATABASE']
+        if create_database == "true":
+            sql = "CREATE DATABASE {};".format(database_name)
+            cursor.execute(sql)
+
         # Create user
         sql = "CREATE USER {} WITH PASSWORD '{}' CREATEDB;".format(
             username, password)

main.tf

@@ -51,6 +51,7 @@ module "provisoner_lambda" {
   environment_variables = {
     RDS_HOST                      = var.rds_endpoint
     RDS_PORT                      = var.rds_port
+    CREATE_DATABASE               = var.create_database ? "true" : "false"
     DB_USER_SECRET_MANAGER_NAME   = var.rds_user_secret_name
     DB_MASTER_SECRET_MANAGER_NAME = var.rds_master_user_secret_name != null ? var.rds_master_user_secret_name : local.master_user_secret_name
   }

variables.tf

@@ -19,6 +19,12 @@ variable "vpc_config" {
   description = "VPC configuration for Lambda function"
 }
 
+variable "create_database" {
+  type        = bool
+  default     = true
+  description = "Flag to create database or not"
+}
+
 variable "invoke" {
   type        = bool
   default     = true

versions.tf

@@ -4,7 +4,7 @@ terraform {
   required_providers {
     aws = {
       source  = "hashicorp/aws"
-      version = ">= 4.0"
+      version = ">= 5.0"
     }
   }
 }