INTMDB-802: Create new TF Data Federation Query Limit resource and data sources

examples/federated-database-query-limit/README.md

@@ -0,0 +1,48 @@
+# Example - MongoDB Atlas Federated Database Query Limit with Atlas clusters
+
+This project aims to provide an example of using [MongoDB Atlas Federated Database Query Limit](https://www.mongodb.com/docs/atlas/data-federation/overview/).
+
+
+## Dependencies
+
+* Terraform MongoDB Atlas Provider v1.10.0
+* A MongoDB Atlas account 
+
+```
+Terraform v1.4.6
++ provider registry.terraform.io/terraform-providers/mongodbatlas v1.10.0
+```
+
+## Usage
+**1\. Create a .tfvars file**
+
+This example requires an Atlas Project to already exist. Once a project is created create the terraform.tfvars file and enter the values for all the required variables, including the project, and make sure **not to commit it**.
+
+**2\. Review the Terraform plan**
+
+Execute the below command and ensure you are happy with the plan. The `terraform plan` command lets you to preview the actions Terraform would take to modify your infrastructure, or save a speculative plan which you can apply later.
+
+``` bash
+$ terraform plan
+```
+This project currently supports the below deployments:
+
+- Two MongoDB Atlas clusters in the specified project
+- MongoDB Atlas Federated Database Instance based on Atlas clusters
+- MongoDB Atlas Federated Database Query Limit
+
+**3\. Execute the Terraform apply.**
+
+Now execute the plan to provision the Federated settings resources. The `terraform apply` command performs a plan just like `terraform plan` does, but then actually carries out the planned changes to each resource using the relevant infrastructure provider's API. It asks for confirmation from the user before making any changes, unless it was explicitly told to skip approval.
+
+``` bash
+$ terraform apply
+```
+
+**4\. Destroy the resources.**
+
+Once you are finished your testing, ensure you destroy the resources to avoid unnecessary Atlas charges. Calling the `terraform destroy` command will instruct Terraform to terminate / destroy all the resources managed. This will enable you to completely tear down and remove all resources defined in the Terraform State that have previously been deployed.
+
+``` bash
+$ terraform destroy
+```

examples/federated-database-query-limit/main.tf

@@ -0,0 +1,84 @@
+resource "mongodbatlas_advanced_cluster" "atlas_cluster_1" {
+  project_id   = var.project_id
+  name         = var.atlas_cluster_name_1
+  cluster_type = "REPLICASET"
+
+  replication_specs {
+    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           = var.provider_region_name
+      priority              = 7
+    }
+  }
+}
+
+resource "mongodbatlas_advanced_cluster" "atlas_cluster_2" {
+  project_id   = var.project_id
+  name         = var.atlas_cluster_name_2
+  cluster_type = "REPLICASET"
+
+  replication_specs {
+    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           = var.provider_region_name
+      priority              = 7
+    }
+  }
+}
+
+resource "mongodbatlas_federated_database_instance" "test-instance" {
+  project_id = var.project_id
+  name       = var.federated_instance_name
+  storage_databases {
+    name = "VirtualDatabase0"
+    collections {
+      name = "VirtualCollection0"
+      data_sources {
+        collection = var.collection_1
+        database   = var.database_1
+        store_name = mongodbatlas_advanced_cluster.atlas_cluster_1.name
+      }
+      data_sources {
+        collection = var.collection_2
+        database   = var.database_2
+        store_name = mongodbatlas_advanced_cluster.atlas_cluster_2.name
+      }
+    }
+  }
+
+  storage_stores {
+    name         = mongodbatlas_advanced_cluster.atlas_cluster_1.name
+    cluster_name = mongodbatlas_advanced_cluster.atlas_cluster_1.name
+    project_id   = var.project_id
+    provider     = "atlas"
+    read_preference {
+      mode = "secondary"
+    }
+  }
+
+  storage_stores {
+    name         = mongodbatlas_advanced_cluster.atlas_cluster_2.name
+    cluster_name = mongodbatlas_advanced_cluster.atlas_cluster_2.name
+    project_id   = var.project_id
+    provider     = "atlas"
+    read_preference {
+      mode = "secondary"
+    }
+  }
+}
+
+resource "mongodbatlas_federated_query_limit" "query_limit" {
+  project_id     = var.project_id
+  tenant_name    = mongodbatlas_federated_database_instance.test-instance.name
+  limit_name     = var.federated_query_limit
+  overrun_policy = var.overrun_policy
+  value          = var.limit_value
+}

examples/federated-database-query-limit/output.tf

@@ -0,0 +1,7 @@
+output "mongodbatlas_federated_query_limit_name" {
+  value = mongodbatlas_federated_query_limit.query_limit.limit_name
+}
+
+output "mongodbatlas_federated_query_limit_default_limit" {
+  value = mongodbatlas_federated_query_limit.query_limit.default_limit
+}

examples/federated-database-query-limit/provider.tf

@@ -0,0 +1,4 @@
+provider "mongodbatlas" {
+  public_key  = var.public_key
+  private_key = var.private_key
+}

examples/federated-database-query-limit/variables.tf

@@ -0,0 +1,90 @@
+variable "public_key" {
+  type        = string
+  description = "Public Programmatic API key to authenticate to Atlas"
+}
+variable "private_key" {
+  type        = string
+  description = "Private Programmatic API key to authenticate to Atlas"
+}
+variable "project_id" {
+  type        = string
+  description = "MongoDB Project ID"
+}
+
+variable "collection_1" {
+  type        = string
+  description = "Human-readable label that identifies the collection in the database in first cluster"
+}
+
+variable "database_1" {
+  type        = string
+  description = "Human-readable label that identifies the database, which contains the collection in the first cluster"
+}
+
+variable "collection_2" {
+  type        = string
+  description = "Human-readable label that identifies the collection in the database in second cluster"
+}
+
+variable "database_2" {
+  type        = string
+  description = "Human-readable label that identifies the database, which contains the collection in the second cluster"
+}
+
+variable "federated_instance_name" {
+  type        = string
+  description = "MongoDB Federated Database Instance Name."
+  default     = "FederatedDatabaseInstance0"
+}
+
+variable "atlas_cluster_name_1" {
+  type        = string
+  description = "First Atlas Cluster Name."
+  default     = "ClusterFederatedTest1"
+}
+
+variable "atlas_cluster_name_2" {
+  type        = string
+  description = "Second Atlas Cluster Name."
+  default     = "ClusterFederatedTest2"
+}
+
+variable "provider_region_name" {
+  type        = string
+  description = "Physical location where MongoDB Cloud deploys your AWS-hosted MongoDB cluster nodes."
+  default     = "US_EAST_1"
+}
+
+variable "provider_instance_size_name" {
+  type        = string
+  description = "Cluster tier. Default is M10"
+  default     = "M10"
+}
+
+variable "backing_provider_name" {
+  type    = string
+  default = "AWS"
+}
+
+variable "provider_name" {
+  type    = string
+  default = "AWS"
+}
+
+variable "federated_query_limit" {
+  type        = string
+  description = "Human-readable label that identifies the user-managed limit to modify."
+  default     = "bytesProcessed.monthly"
+}
+
+variable "overrun_policy" {
+  type        = string
+  description = "Action to take when the usage limit is exceeded."
+  default     = "BLOCK"
+}
+
+variable "limit_value" {
+  type        = number
+  description = "Amount to set the federated query limit to."
+  default     = 5147483648
+}

examples/federated-database-query-limit/versions.tf

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

mongodbatlas/data_source_mongodbatlas_federated_query_limit.go

@@ -0,0 +1,83 @@
+package mongodbatlas
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+)
+
+func dataSourceMongoDBAtlasFederatedDatabaseQueryLimit() *schema.Resource {
+	return &schema.Resource{
+		ReadContext: dataSourceMongoDBAtlasFederatedDatabaseQueryLimitRead,
+		Schema:      schemaMongoDBAtlasFederatedDatabaseQueryLimitDataSource(),
+	}
+}
+
+func schemaMongoDBAtlasFederatedDatabaseQueryLimitDataSource() map[string]*schema.Schema {
+	return map[string]*schema.Schema{
+		"project_id": {
+			Type:     schema.TypeString,
+			Required: true,
+		},
+		"limit_name": {
+			Type:     schema.TypeString,
+			Required: true,
+		},
+		"tenant_name": {
+			Type:     schema.TypeString,
+			Required: true,
+		},
+		"overrun_policy": {
+			Type:     schema.TypeString,
+			Computed: true,
+		},
+		"value": {
+			Type:     schema.TypeInt,
+			Computed: true,
+		},
+		"current_usage": {
+			Type:     schema.TypeInt,
+			Computed: true,
+		},
+		"default_limit": {
+			Type:     schema.TypeInt,
+			Computed: true,
+		},
+		"last_modified_date": {
+			Type:     schema.TypeString,
+			Computed: true,
+		},
+		"maximum_limit": {
+			Type:     schema.TypeInt,
+			Computed: true,
+		},
+	}
+}
+
+func dataSourceMongoDBAtlasFederatedDatabaseQueryLimitRead(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
+	conn := meta.(*MongoDBClient).Atlas
+
+	projectID := d.Get("project_id").(string)
+	tenantName := d.Get("tenant_name").(string)
+	limitName := d.Get("limit_name").(string)
+
+	queryLimit, _, err := conn.DataFederation.GetQueryLimit(ctx, projectID, tenantName, limitName)
+
+	if err != nil {
+		return diag.FromErr(fmt.Errorf("couldn't import federated database query limit(%s) for project (%s), tenant (%s), error: %s", limitName, projectID, tenantName, err))
+	}
+
+	if err = setResourceFieldsFromFederatedDatabaseQueryLimit(d, projectID, queryLimit); err != nil {
+		return diag.FromErr(err)
+	}
+
+	d.SetId(encodeStateID(map[string]string{
+		"project_id":  projectID,
+		"tenant_name": queryLimit.TenantName,
+		"limit_name":  queryLimit.Name,
+	}))
+
+	return nil
+}