From 9487cddcc444f89663e80df93d4ad7e89979024e Mon Sep 17 00:00:00 2001
From: amanravi-squareops <aman.ravi@squareops.comm>
Date: Wed, 27 Nov 2024 11:48:10 +0530
Subject: [PATCH] added backup restore functionality and fixed security group

---
 examples/complete/main.tf                              | 10 ++++++----
 .../db-backup-restore/backup/templates/cronjob.yaml    |  2 +-
 modules/db-backup-restore/restore/templates/job.yaml   |  4 +---
 3 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/examples/complete/main.tf b/examples/complete/main.tf
index 44df0a8..e45dd02 100644
--- a/examples/complete/main.tf
+++ b/examples/complete/main.tf
@@ -9,8 +9,9 @@ locals {
   engine_version          = "15.4"
   instance_class          = "db.t4g.micro"
   storage_type            = "gp3"
+  cluster_name            = ""
   current_identity        = data.aws_caller_identity.current.arn
-  allowed_security_groups = ["sg-xxxxxxxxxxxxxxxx"]
+  allowed_security_groups = ["sg-xxxxxxxxxxxxxx"]
   custom_user_password    = ""
   additional_tags = {
     Owner      = "Organization_Name"
@@ -100,13 +101,14 @@ module "vpc" {
 
 module "rds-pg" {
   source                           = "squareops/rds-postgresql/aws"
+  version                          = "2.0.0"
   name                             = local.name
   db_name                          = "test"
-  multi_az                         = "true"
+  multi_az                         = false
   family                           = local.family
   vpc_id                           = module.vpc.vpc_id
   allowed_security_groups          = local.allowed_security_groups
-  subnet_ids                       = module.vpc.database_subnets ## db subnets
+  subnet_ids                       = module.vpc.database_subnets
   environment                      = local.environment
   kms_key_arn                      = module.kms.key_arn
   storage_type                     = local.storage_type
@@ -132,7 +134,7 @@ module "rds-pg" {
   custom_user_password             = local.custom_user_password
   #if you want backup and restore then you have to create your cluster with rds vpc id , private subnets, kms key.
   #And allow cluster security group in rds security group
-  cluster_name              = ""
+  cluster_name              = local.cluster_name
   namespace                 = local.namespace
   create_namespace          = local.create_namespace
   postgresdb_backup_enabled = false
diff --git a/modules/db-backup-restore/backup/templates/cronjob.yaml b/modules/db-backup-restore/backup/templates/cronjob.yaml
index f32d59e..01a70fc 100644
--- a/modules/db-backup-restore/backup/templates/cronjob.yaml
+++ b/modules/db-backup-restore/backup/templates/cronjob.yaml
@@ -18,7 +18,7 @@ spec:
           serviceAccountName: sa-postgres-backup
           containers:
           - name: backup-postgresdb
-            image: amanravi12/postgresql-backup:v5
+            image: squareops01/rds-postgresql-backup:v1
             imagePullPolicy: Always
             command: ["/backup/backup_script.sh"]
             env:
diff --git a/modules/db-backup-restore/restore/templates/job.yaml b/modules/db-backup-restore/restore/templates/job.yaml
index fa2e0fd..473b3ee 100644
--- a/modules/db-backup-restore/restore/templates/job.yaml
+++ b/modules/db-backup-restore/restore/templates/job.yaml
@@ -9,7 +9,7 @@ spec:
       serviceAccountName: sa-postgres-restore
       containers:
         - name: restore-postgresdb
-          image: amanravi12/postgresql-restore:v6
+          image: squareops01/rds-postgresql-restore:v1
           imagePullPolicy: Always
           command: ["/restore/restore_script.sh"]
           env:
@@ -23,8 +23,6 @@ spec:
               value: {{ .Values.restore.bucket_uri }}
             - name: DB_PORT
               value: "5432"
-            # - name: DB_NAME
-            #   value: {{ .Values.restore.DB_NAME }}
             - name: RESTORE_FILE_NAME
               value: {{ .Values.restore.backup_file_name }}
           resources: {{ .Values.restorejob.resources | toYaml | nindent 12 }}