Merge pull request #30 from datarootsio/ISSUE-27/more-rds-options

Issue 27/more rds options

Author: dpfeif

README.md

@@ -113,11 +113,14 @@ For now the only authentication option is 'RBAC'. When enabling this, this modul
 | rbac\_admin\_lastname | RBAC Lastname (only when airflow\_authentication = 'rbac') | `string` | `"airflow"` | no |
 | rbac\_admin\_password | RBAC Password (only when airflow\_authentication = 'rbac') | `string` | `"admin"` | no |
 | rbac\_admin\_username | RBAC Username (only when airflow\_authentication = 'rbac') | `string` | `"admin"` | no |
+| rds\_allocated\_storage | The allocated storage for the rds db in gibibytes | `number` | `20` | no |
 | rds\_availability\_zone | Availability zone for the rds instance | `string` | `"eu-west-1a"` | no |
 | rds\_deletion\_protection | Deletion protection for the rds instance | `bool` | `false` | no |
+| rds\_engine | The database engine to use. For supported values, see the Engine parameter in [API action CreateDBInstance](https://docs.aws.amazon.com/AmazonRDS/latest/APIReference/API_CreateDBInstance.html) | `string` | `"postgres"` | no |
 | rds\_instance\_class | The class of instance you want to give to your rds db | `string` | `"db.t2.micro"` | no |
 | rds\_password | Password of rds | `string` | `""` | no |
 | rds\_skip\_final\_snapshot | Whether or not to skip the final snapshot before deleting (mainly for tests) | `bool` | `false` | no |
+| rds\_storage\_type | One of `"standard"` (magnetic), `"gp2"` (general purpose SSD), or `"io1"` (provisioned IOPS SSD) | `string` | `"standard"` | no |
 | rds\_username | Username of rds | `string` | `"airflow"` | no |
 | region | The region to deploy your solution to | `string` | `"eu-west-1"` | no |
 | resource\_prefix | A prefix for the create resources, example your company name (be aware of the resource name length) | `string` | n/a | yes |

examples/basic-local-with-requirements/main.tf

@@ -12,26 +12,26 @@ provider "aws" {
 }
 
 module "airflow" {
-    source = "datarootsio/ecs-airflow/aws"
+  source = "datarootsio/ecs-airflow/aws"
 
-    resource_prefix = "my-awesome-company"
-    resource_suffix = "env"
+  resource_prefix = "my-awesome-company"
+  resource_suffix = "env"
 
-    vpc_id             = "vpc-123456"
-    public_subnet_ids  = ["subnet-456789", "subnet-098765"]
+  vpc_id            = "vpc-123456"
+  public_subnet_ids = ["subnet-456789", "subnet-098765"]
 
-    use_https = false
+  use_https = false
 
-    airflow_executor = "Local"
-    airflow_py_requirements_path = "./requirements.txt"
-    airflow_variables = {
-      AIRFLOW__WEBSERVER__NAVBAR_COLOR : "#e27d60"
-    }
+  airflow_executor             = "Local"
+  airflow_py_requirements_path = "./requirements.txt"
+  airflow_variables = {
+    AIRFLOW__WEBSERVER__NAVBAR_COLOR : "#e27d60"
+  }
 
-    rds_username            = "airflow"
-    rds_password            = "super-secret-pass"
-    rds_availability_zone   = "eu-west-1a"
-    rds_deletion_protection = false
+  rds_username            = "airflow"
+  rds_password            = "super-secret-pass"
+  rds_availability_zone   = "eu-west-1a"
+  rds_deletion_protection = false
 }
 
 output "airflow_alb_dns" {

examples/basic-sequential-rbac/main.tf

@@ -12,22 +12,22 @@ provider "aws" {
 }
 
 module "airflow" {
-    source = "datarootsio/ecs-airflow/aws"
+  source = "datarootsio/ecs-airflow/aws"
 
-    resource_prefix = "my-awesome-company"
-    resource_suffix = "env"
+  resource_prefix = "my-awesome-company"
+  resource_suffix = "env"
 
-    vpc_id             = "vpc-123456"
-    public_subnet_ids  = ["subnet-456789", "subnet-098765"]
+  vpc_id            = "vpc-123456"
+  public_subnet_ids = ["subnet-456789", "subnet-098765"]
 
-    airflow_executor             = "Sequential"
-    airflow_py_requirements_path = "./requirements.txt"
-    airflow_authentication       = "rbac"
-    airflow_variables            = {
-      AIRFLOW__WEBSERVER__NAVBAR_COLOR : "#e27d60"
-    }
+  airflow_executor             = "Sequential"
+  airflow_py_requirements_path = "./requirements.txt"
+  airflow_authentication       = "rbac"
+  airflow_variables = {
+    AIRFLOW__WEBSERVER__NAVBAR_COLOR : "#e27d60"
+  }
 
-    use_https = false
+  use_https = false
 }
 
 output "airflow_alb_dns" {

examples/basic-sequential/main.tf

@@ -12,20 +12,20 @@ provider "aws" {
 }
 
 module "airflow" {
-    source = "datarootsio/ecs-airflow/aws"
+  source = "datarootsio/ecs-airflow/aws"
 
-    resource_prefix = "my-awesome-company"
-    resource_suffix = "env"
+  resource_prefix = "my-awesome-company"
+  resource_suffix = "env"
 
-    vpc_id             = "vpc-123456"
-    public_subnet_ids  = ["subnet-456789", "subnet-098765"]
+  vpc_id            = "vpc-123456"
+  public_subnet_ids = ["subnet-456789", "subnet-098765"]
 
-    airflow_executor = "Sequential"
-    airflow_variables = {
-      AIRFLOW__WEBSERVER__NAVBAR_COLOR : "#e27d60"
-    }
+  airflow_executor = "Sequential"
+  airflow_variables = {
+    AIRFLOW__WEBSERVER__NAVBAR_COLOR : "#e27d60"
+  }
 
-    use_https = false
+  use_https = false
 }
 
 output "airflow_alb_dns" {

rds.tf

@@ -1,9 +1,9 @@
 resource "aws_db_instance" "airflow" {
   count                     = var.postgres_uri != "" || var.airflow_executor == "Sequential" ? 0 : 1
   name                      = replace(title(local.rds_name), "-", "")
-  allocated_storage         = 20
-  storage_type              = "standard"
-  engine                    = "postgres"
+  allocated_storage         = var.rds_allocated_storage
+  storage_type              = var.rds_storage_type
+  engine                    = var.rds_engine
   engine_version            = "11.8"
   instance_class            = var.rds_instance_class
   username                  = var.rds_username
@@ -27,4 +27,4 @@ resource "aws_db_subnet_group" "airflow" {
   subnet_ids = local.rds_ecs_subnet_ids
 
   tags = local.common_tags
-}
+}

variables.tf

@@ -206,7 +206,28 @@ variable "postgres_uri" {
   default     = ""
 }
 
-// TODO: Add more rds params
+variable "rds_allocated_storage" {
+  type        = number
+  description = "The allocated storage for the rds db in gibibytes"
+  default     = 20
+}
+
+variable "rds_storage_type" {
+  type        = string
+  description = <<EOT
+  One of `"standard"` (magnetic), `"gp2"` (general purpose SSD), or `"io1"` (provisioned IOPS SSD)
+  EOT
+  default     = "standard"
+}
+
+variable "rds_engine" {
+  type        = string
+  description = <<EOT
+  The database engine to use. For supported values, see the Engine parameter in [API action CreateDBInstance](https://docs.aws.amazon.com/AmazonRDS/latest/APIReference/API_CreateDBInstance.html)
+  EOT
+  default     = "postgres"
+}
+
 variable "rds_username" {
   type        = string
   description = "Username of rds"