INTMDB-314 Cluster Tenant Upgrade

examples/atlas-advanced-cluster/tenant-upgrade/README.md

@@ -0,0 +1,32 @@
+# MongoDB Atlas Provider -- Advanced Cluster Tenant Upgrade
+This example creates a project and cluster. It is intended to show how to upgrade from shared, aka tenant, to dedicated tier.
+
+Variables Required:
+- `atlas_org_id`: ID of the Atlas organization
+- `public_key`: Atlas public key
+- `private_key`: Atlas  private key
+- `provider_name`: Name of provider to use for cluster (TENANT, AWS, GCP)
+- `backing_provider_name`: If provider_name is tenant, the backing provider (AWS, GCP)
+- `provider_instance_size_name`: Size of the cluster (Shared: M0, M2, M5, Dedicated: M10+.)
+
+For this example, first we'll start out on the shared tier, then upgrade to a dedicated tier.
+
+Utilize the following to execute a working example, replacing the org id, public and private key with your values:
+
+Apply with the following `terraform.tfvars` to first create a shared tier cluster:
+```
+atlas_org_id                = "627a9687f7f7f7f774de306f14"
+public_key                  = <REDACTED>
+private_key                 = <REDACTED>
+provider_name               = "TENANT"
+backing_provider_name       = "AWS"
+provider_instance_size_name = "M2"
+```
+
+Apply with the following `terraform.tfvars` to upgrade the shared tier cluster you just created to dedicated tier:
+```
+atlas_org_id                = "627a9687f7f7f7f774de306f14"
+public_key                  = <REDACTED>
+private_key                 = <REDACTED>
+provider_name               = "GCP"
+provider_instance_size_name = "M10"

examples/atlas-advanced-cluster/tenant-upgrade/main.tf

@@ -0,0 +1,29 @@
+provider "mongodbatlas" {
+  public_key  = var.public_key
+  private_key = var.private_key
+}
+
+resource "mongodbatlas_advanced_cluster" "cluster" {
+  project_id   = mongodbatlas_project.project.id
+  name         = "ClusterToUpgrade"
+  cluster_type = "REPLICASET"
+
+  replication_specs {
+    num_shards = 1
+
+    region_configs {
+      electable_specs {
+        instance_size = var.provider_instance_size_name
+      }
+      provider_name         = var.provider_name
+      backing_provider_name = var.backing_provider_name
+      region_name           = "US_EAST_1"
+      priority              = 7
+    }
+  }
+}
+
+resource "mongodbatlas_project" "project" {
+  name   = "TenantUpgradeTest"
+  org_id = var.atlas_org_id
+}

examples/atlas-advanced-cluster/tenant-upgrade/variables.tf

@@ -0,0 +1,22 @@
+variable "atlas_org_id" {
+  description = "Atlas organization id"
+  default     = ""
+}
+variable "public_key" {
+  description = "Public API key to authenticate to Atlas"
+}
+variable "private_key" {
+  description = "Private API key to authenticate to Atlas"
+}
+variable "provider_name" {
+  description = "Atlas cluster provider name"
+  default     = "AWS"
+}
+variable "backing_provider_name" {
+  description = "Atlas cluster backing provider name"
+  default     = null
+}
+variable "provider_instance_size_name" {
+  description = "Atlas cluster provider instance name"
+  default     = "M10"
+}

examples/atlas-advanced-cluster/tenant-upgrade/versions.tf

@@ -0,0 +1,8 @@
+terraform {
+  required_providers {
+    mongodbatlas = {
+      source = "mongodb/mongodbatlas"
+    }
+  }
+  required_version = ">= 0.13"
+}

examples/atlas-cluster/tenant-upgrade/README.md

@@ -0,0 +1,33 @@
+# MongoDB Atlas Provider -- Cluster Tenant Upgrade
+This example creates a project and cluster. It is intended to show how to upgrade from shared, aka tenant, to dedicated tier.
+
+Variables Required:
+- `atlas_org_id`: ID of the Atlas organization
+- `public_key`: Atlas public key
+- `private_key`: Atlas  private key
+- `provider_name`: Name of provider to use for cluster (TENANT, AWS, GCP)
+- `backing_provider_name`: If provider_name is tenant, the backing provider (AWS, GCP)
+- `provider_instance_size_name`: Size of the cluster (Shared: M0, M2, M5, Dedicated: M10+.)
+
+For this example, first we'll start out on the shared tier, then upgrade to a dedicated tier.
+
+
+Utilize the following to execute a working example, replacing the org id, public and private key with your values:
+
+Apply with the following `terraform.tfvars` to first create a shared tier cluster:
+```
+atlas_org_id                = "627a9687f7f7f7f774de306f14"
+public_key                  = <REDACTED>
+private_key                 = <REDACTED>
+provider_name               = "TENANT"
+backing_provider_name       = "AWS"
+provider_instance_size_name = "M2"
+```
+
+Apply with the following `terraform.tfvars` to upgrade the shared tier cluster you just created to dedicated tier:
+```
+atlas_org_id                = "627a9687f7f7f7f774de306f14"
+public_key                  = <REDACTED>
+private_key                 = <REDACTED>
+provider_name               = "GCP"
+provider_instance_size_name = "M10"

examples/atlas-cluster/tenant-upgrade/main.tf

@@ -0,0 +1,19 @@
+provider "mongodbatlas" {
+  public_key  = var.public_key
+  private_key = var.private_key
+}
+
+resource "mongodbatlas_cluster" "cluster" {
+  project_id                  = mongodbatlas_project.project.id
+  name                        = "ClusterToUpgrade"
+  cluster_type                = "REPLICASET"
+  provider_name               = var.provider_name
+  backing_provider_name       = var.backing_provider_name
+  provider_region_name        = "US_EAST_1"
+  provider_instance_size_name = var.provider_instance_size_name
+}
+
+resource "mongodbatlas_project" "project" {
+  name   = "TenantUpgradeTest"
+  org_id = var.atlas_org_id
+}

examples/atlas-cluster/tenant-upgrade/variables.tf

@@ -0,0 +1,22 @@
+variable "atlas_org_id" {
+  description = "Atlas organization id"
+  default     = ""
+}
+variable "public_key" {
+  description = "Public API key to authenticate to Atlas"
+}
+variable "private_key" {
+  description = "Private API key to authenticate to Atlas"
+}
+variable "provider_name" {
+  description = "Atlas cluster provider name"
+  default     = "AWS"
+}
+variable "backing_provider_name" {
+  description = "Atlas cluster backing provider name"
+  default     = null
+}
+variable "provider_instance_size_name" {
+  description = "Atlas cluster provider instance name"
+  default     = "M10"
+}

examples/atlas-cluster/tenant-upgrade/versions.tf

@@ -0,0 +1,8 @@
+terraform {
+  required_providers {
+    mongodbatlas = {
+      source = "mongodb/mongodbatlas"
+    }
+  }
+  required_version = ">= 0.13"
+}

mongodbatlas/resource_mongodbatlas_advanced_cluster.go

@@ -21,6 +21,8 @@ import (
 	matlas "go.mongodb.org/atlas/mongodbatlas"
 )
 
+type acCtxKey string
+
 const (
 	errorClusterAdvancedCreate             = "error creating MongoDB ClusterAdvanced: %s"
 	errorClusterAdvancedRead               = "error reading MongoDB ClusterAdvanced (%s): %s"
@@ -31,11 +33,13 @@ const (
 	errorAdvancedClusterAdvancedConfRead   = "error reading Advanced Configuration Option form MongoDB Cluster (%s): %s"
 )
 
+var upgradeRequestCtxKey acCtxKey = "upgradeRequest"
+
 func resourceMongoDBAtlasAdvancedCluster() *schema.Resource {
 	return &schema.Resource{
 		CreateWithoutTimeout: resourceMongoDBAtlasAdvancedClusterCreate,
 		ReadWithoutTimeout:   resourceMongoDBAtlasAdvancedClusterRead,
-		UpdateWithoutTimeout: resourceMongoDBAtlasAdvancedClusterUpdate,
+		UpdateWithoutTimeout: resourceMongoDBAtlasAdvancedClusterUpdateOrUpgrade,
 		DeleteWithoutTimeout: resourceMongoDBAtlasAdvancedClusterDelete,
 		Importer: &schema.ResourceImporter{
 			StateContext: resourceMongoDBAtlasAdvancedClusterImportState,
@@ -511,6 +515,42 @@ func resourceMongoDBAtlasAdvancedClusterRead(ctx context.Context, d *schema.Reso
 	return nil
 }
 
+func resourceMongoDBAtlasAdvancedClusterUpdateOrUpgrade(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
+	if upgradeRequest := getUpgradeRequest(d); upgradeRequest != nil {
+		upgradeCtx := context.WithValue(ctx, upgradeRequestCtxKey, upgradeRequest)
+		return resourceMongoDBAtlasAdvancedClusterUpgrade(upgradeCtx, d, meta)
+	}
+
+	return resourceMongoDBAtlasAdvancedClusterUpdate(ctx, d, meta)
+}
+
+func resourceMongoDBAtlasAdvancedClusterUpgrade(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
+	conn := meta.(*MongoDBClient).Atlas
+	ids := decodeStateID(d.Id())
+	projectID := ids["project_id"]
+	clusterName := ids["cluster_name"]
+
+	upgradeRequest := ctx.Value(upgradeRequestCtxKey).(*matlas.Cluster)
+
+	if upgradeRequest == nil {
+		return diag.FromErr(fmt.Errorf("upgrade called without %s in ctx", string(upgradeRequestCtxKey)))
+	}
+
+	upgradeResponse, _, err := upgradeCluster(ctx, conn, upgradeRequest, projectID, clusterName)
+
+	if err != nil {
+		return diag.FromErr(fmt.Errorf(errorClusterAdvancedUpdate, clusterName, err))
+	}
+
+	d.SetId(encodeStateID(map[string]string{
+		"cluster_id":   upgradeResponse.ID,
+		"project_id":   projectID,
+		"cluster_name": clusterName,
+	}))
+
+	return resourceMongoDBAtlasAdvancedClusterRead(ctx, d, meta)
+}
+
 func resourceMongoDBAtlasAdvancedClusterUpdate(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
 	// Get client connection.
 	conn := meta.(*MongoDBClient).Atlas
@@ -1079,6 +1119,38 @@ func replicationSpecsHashSet(v interface{}) int {
 	return schema.HashString(buf.String())
 }
 
+func getUpgradeRequest(d *schema.ResourceData) *matlas.Cluster {
+	if !d.HasChange("replication_specs") {
+		return nil
+	}
+
+	cs, us := d.GetChange("replication_specs")
+	currentSpecs := expandAdvancedReplicationSpecs(cs.(*schema.Set).List())
+	updatedSpecs := expandAdvancedReplicationSpecs(us.(*schema.Set).List())
+
+	if len(currentSpecs) != 1 || len(updatedSpecs) != 1 || len(currentSpecs[0].RegionConfigs) != 1 || len(updatedSpecs[0].RegionConfigs) != 1 {
+		return nil
+	}
+
+	currentRegion := currentSpecs[0].RegionConfigs[0]
+	updatedRegion := updatedSpecs[0].RegionConfigs[0]
+	currentSize := currentRegion.ElectableSpecs.InstanceSize
+
+	if currentRegion.ElectableSpecs.InstanceSize == updatedRegion.ElectableSpecs.InstanceSize || !(currentSize == "M0" ||
+		currentSize == "M2" ||
+		currentSize == "M5") {
+		return nil
+	}
+
+	return &matlas.Cluster{
+		ProviderSettings: &matlas.ProviderSettings{
+			ProviderName:     updatedRegion.ProviderName,
+			InstanceSizeName: updatedRegion.ElectableSpecs.InstanceSize,
+			RegionName:       updatedRegion.RegionName,
+		},
+	}
+}
+
 func updateAdvancedCluster(ctx context.Context, conn *matlas.Client, request *matlas.AdvancedCluster, projectID, name string) (*matlas.AdvancedCluster, *matlas.Response, error) {
 	cluster, resp, err := conn.AdvancedClusters.Update(ctx, projectID, name, request)
 	if err != nil {