Add support for restoring AlloyDB clusters via backup restore and PITR

mmv1/products/alloydb/Cluster.yaml

@@ -62,6 +62,24 @@ examples:
     primary_resource_id: 'full'
     vars:
       alloydb_cluster_name: 'alloydb-cluster-full'
+  - !ruby/object:Provider::Terraform::Examples
+    name: 'alloydb_cluster_restore'
+    primary_resource_id: 'source'
+    skip_test: true
+    vars:
+      alloydb_cluster_name: 'alloydb-source-cluster'
+      alloydb_backup_restored_cluster_name: 'alloydb-backup-restored'
+      alloydb_pitr_restored_cluster_name: 'alloydb-pitr-restored'
+      alloydb_backup_id: 'alloydb-backup'
+      alloydb_instance_name: 'alloydb-instance'
+      network_name: 'alloydb-network'
+    test_vars_overrides:
+      network_name: 'acctest.BootstrapSharedTestNetwork(t, "alloydb-instance-basic")'
+    ignore_read_extra:
+      - 'reconciling'
+      - 'update_time'
+custom_code: !ruby/object:Provider::Terraform::CustomCode
+  pre_create: templates/terraform/pre_create/alloydb_restore_cluster.go.erb
 parameters:
   - !ruby/object:Api::Type::String
     name: 'clusterId'
@@ -193,6 +211,42 @@ properties:
           The initial password for the user.
         required: true
         sensitive: true
+  - !ruby/object:Api::Type::NestedObject
+    name: 'restoreBackupSource'
+    ignore_read: true
+    immutable: true
+    conflicts:
+      - restore_continuous_backup_source
+    description: |
+      The source when restoring from a backup. Conflicts with 'restore_continuous_backup_source', both can't be set together.
+    properties:
+      - !ruby/object:Api::Type::String
+        name: 'backupName'
+        required: true
+        immutable: true
+        description: |
+          The name of the backup that this cluster is restored from.
+  - !ruby/object:Api::Type::NestedObject
+    name: 'restoreContinuousBackupSource'
+    ignore_read: true
+    immutable: true
+    conflicts:
+      - restore_backup_source
+    description: |
+      The source when restoring via point in time recovery (PITR). Conflicts with 'restore_backup_source', both can't be set together.
+    properties:
+      - !ruby/object:Api::Type::String
+        name: 'cluster'
+        required: true
+        immutable: true
+        description: |
+          The name of the source cluster that this cluster is restored from.
+      - !ruby/object:Api::Type::String
+        name: 'pointInTime'
+        required: true
+        immutable: true
+        description: |
+          The point in time that this cluster is restored to, in RFC 3339 format.
   - !ruby/object:Api::Type::NestedObject
     name: 'continuousBackupConfig'
     default_from_api: true

mmv1/templates/terraform/examples/alloydb_cluster_restore.tf.erb

@@ -0,0 +1,69 @@
+resource "google_alloydb_cluster" "<%= ctx[:primary_resource_id] %>" {
+  cluster_id = "<%= ctx[:vars]['alloydb_cluster_name'] %>"
+  location   = "us-central1"
+  network    = data.google_compute_network.default.id
+
+  initial_user {
+    password = "<%= ctx[:vars]['alloydb_cluster_name'] %>"
+  }
+}
+
+resource "google_alloydb_instance" "<%= ctx[:primary_resource_id] %>" {
+  cluster       = google_alloydb_cluster.<%= ctx[:primary_resource_id] %>.name
+  instance_id   = "<%= ctx[:vars]['alloydb_instance_name'] %>"
+  instance_type = "PRIMARY"
+
+  machine_config {
+    cpu_count = 2
+  }
+
+  depends_on = [google_service_networking_connection.vpc_connection]
+}
+
+resource "google_alloydb_backup" "<%= ctx[:primary_resource_id] %>" {
+  backup_id    = "<%= ctx[:vars]['alloydb_backup_id'] %>"
+  location     = "us-central1"
+  cluster_name = google_alloydb_cluster.<%= ctx[:primary_resource_id] %>.name
+
+  depends_on = [google_alloydb_instance.<%= ctx[:primary_resource_id] %>]
+}
+
+resource "google_alloydb_cluster" "restored_from_backup" {
+  cluster_id            = "<%= ctx[:vars]['alloydb_backup_restored_cluster_name'] %>"
+  location              = "us-central1"
+  network               = data.google_compute_network.default.id
+  restore_backup_source {
+    backup_name = google_alloydb_backup.<%= ctx[:primary_resource_id] %>.name
+  }
+}
+
+resource "google_alloydb_cluster" "restored_via_pitr" {
+  cluster_id             = "<%= ctx[:vars]['alloydb_pitr_restored_cluster_name'] %>"
+  location               = "us-central1"
+  network                = data.google_compute_network.default.id
+
+  restore_continuous_backup_source {
+    cluster = google_alloydb_cluster.<%= ctx[:primary_resource_id] %>.name
+    point_in_time = "2023-08-03T19:19:00.094Z"
+  }
+}
+
+data "google_project" "project" {}
+
+data "google_compute_network" "default" {
+  name = "<%= ctx[:vars]['network_name'] %>"
+}
+
+resource "google_compute_global_address" "private_ip_alloc" {
+  name          =  "<%= ctx[:vars]['alloydb_cluster_name'] %>"
+  address_type  = "INTERNAL"
+  purpose       = "VPC_PEERING"
+  prefix_length = 16
+  network       = data.google_compute_network.default.id
+}
+
+resource "google_service_networking_connection" "vpc_connection" {
+  network                 = data.google_compute_network.default.id
+  service                 = "servicenetworking.googleapis.com"
+  reserved_peering_ranges = [google_compute_global_address.private_ip_alloc.name]
+}

mmv1/templates/terraform/pre_create/alloydb_restore_cluster.go.erb

@@ -0,0 +1,33 @@
+// Read the restore variables from obj and remove them, since they do not map to anything in the cluster
+var backupSource interface{}
+var continuousBackupSource interface{}
+if val, ok := obj["restoreBackupSource"]; ok {
+	backupSource = val
+	delete(obj, "restoreBackupSource")
+}
+if val, ok := obj["restoreContinuousBackupSource"]; ok {
+	continuousBackupSource = val
+	delete(obj, "restoreContinuousBackupSource")
+}
+
+restoreClusterRequestBody := make(map[string]interface{})
+if backupSource != nil {
+	// If restoring from a backup, set the backupSource
+	restoreClusterRequestBody["backup_source"] = backupSource
+} else if continuousBackupSource != nil {
+	// Otherwise if restoring via PITR, set the continuousBackupSource
+	restoreClusterRequestBody["continuous_backup_source"] = continuousBackupSource
+}
+
+if backupSource != nil || continuousBackupSource != nil {
+	// Use restore API if this is a restore instead of a create cluster call
+	url = strings.Replace(url, "clusters?clusterId", "clusters:restore?clusterId", 1)
+
+	// Copy obj which contains the cluster into a cluster map
+	cluster := make(map[string]interface{})
+	for k,v := range obj {
+		cluster[k] = v
+	}
+	restoreClusterRequestBody["cluster"] = cluster
+	obj = restoreClusterRequestBody
+}