feat(cbh/instance_quota): support instance quota dataSource

Author: jinyangyang222

docs/data-sources/cbh_instance_quota.md

@@ -0,0 +1,34 @@
+---
+subcategory: "Cloud Bastion Host (CBH)"
+layout: "huaweicloud"
+page_title: "HuaweiCloud: huaweicloud_cbh_instance_quota"
+description: |-
+  Use this data source to get CBH instance quota within HuaweiCloud.
+---
+
+# huaweicloud_cbh_instance_quota
+
+Use this data source to get CBH instance quota within HuaweiCloud.
+
+## Example Usage
+
+```hcl
+data "huaweicloud_cbh_instance_quota" "test" {}
+```
+
+## Argument Reference
+
+The following arguments are supported:
+
+* `region` - (Optional, String) Specifies the region in which to query the CBH instance quota.
+  If omitted, the provider-level region will be used.
+
+## Attribute Reference
+
+In addition to all arguments above, the following attributes are exported:
+
+* `id` - The data source ID in UUID format.
+
+* `quota` - The maximum number of CBH instances that can be created.
+
+* `quota_used` - The current number of CBH instances created.

huaweicloud/provider.go

@@ -491,6 +491,7 @@ func Provider() *schema.Provider {
 			"huaweicloud_cbh_instances":          cbh.DataSourceCbhInstances(),
 			"huaweicloud_cbh_flavors":            cbh.DataSourceCbhFlavors(),
 			"huaweicloud_cbh_availability_zones": cbh.DataSourceAvailabilityZones(),
+			"huaweicloud_cbh_instance_quota":     cbh.DataSourceInstanceQuota(),
 
 			"huaweicloud_cc_authorizations":                               cc.DataSourceCcAuthorizations(),
 			"huaweicloud_cc_bandwidth_packages":                           cc.DataSourceCcBandwidthPackages(),

huaweicloud/services/acceptance/cbh/data_source_huaweicloud_cbh_instance_quota_test.go

@@ -0,0 +1,37 @@
+package cbh
+
+import (
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+
+	"github.com/huaweicloud/terraform-provider-huaweicloud/huaweicloud/services/acceptance"
+)
+
+func TestAccDataSourceInstanceQuota_basic(t *testing.T) {
+	var (
+		dataSourceName = "data.huaweicloud_cbh_instance_quota.test"
+		dc             = acceptance.InitDataSourceCheck(dataSourceName)
+	)
+
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck: func() {
+			acceptance.TestAccPreCheck(t)
+		},
+		ProviderFactories: acceptance.TestAccProviderFactories,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccDataSourceInstanceQuota_basic,
+				Check: resource.ComposeTestCheckFunc(
+					dc.CheckResourceExists(),
+					resource.TestCheckResourceAttrSet(dataSourceName, "quota"),
+					resource.TestCheckResourceAttrSet(dataSourceName, "quota_used"),
+				),
+			},
+		},
+	})
+}
+
+const testAccDataSourceInstanceQuota_basic string = `
+data "huaweicloud_cbh_instance_quota" "test" {}
+`

huaweicloud/services/cbh/data_source_huaweicloud_cbh_instance_quota.go

@@ -0,0 +1,87 @@
+package cbh
+
+import (
+	"context"
+	"strings"
+
+	"github.com/hashicorp/go-multierror"
+	"github.com/hashicorp/go-uuid"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+
+	"github.com/chnsz/golangsdk"
+
+	"github.com/huaweicloud/terraform-provider-huaweicloud/huaweicloud/config"
+	"github.com/huaweicloud/terraform-provider-huaweicloud/huaweicloud/utils"
+)
+
+// @API CBH GET /v2/{project_id}/cbs/instance/quota
+func DataSourceInstanceQuota() *schema.Resource {
+	return &schema.Resource{
+		ReadContext: dataSourceInstanceQuotaRead,
+
+		Schema: map[string]*schema.Schema{
+			"region": {
+				Type:     schema.TypeString,
+				Optional: true,
+				Computed: true,
+			},
+			"quota": {
+				Type:     schema.TypeInt,
+				Computed: true,
+			},
+			"quota_used": {
+				Type:     schema.TypeInt,
+				Computed: true,
+			},
+		},
+	}
+}
+
+func dataSourceInstanceQuotaRead(_ context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
+	var (
+		cfg        = meta.(*config.Config)
+		region     = cfg.GetRegion(d)
+		mErr       *multierror.Error
+		getHttpUrl = "v2/{project_id}/cbs/instance/quota"
+		product    = "cbh"
+	)
+
+	client, err := cfg.NewServiceClient(product, region)
+	if err != nil {
+		return diag.Errorf("error creating CBH client: %s", err)
+	}
+
+	getPath := client.Endpoint + getHttpUrl
+	getPath = strings.ReplaceAll(getPath, "{project_id}", client.ProjectID)
+	getOpt := golangsdk.RequestOpts{
+		KeepResponseBody: true,
+		MoreHeaders:      map[string]string{"Content-Type": "application/json"},
+	}
+
+	getResp, err := client.Request("GET", getPath, &getOpt)
+	if err != nil {
+		return diag.Errorf("error retrieving CBH instance quota, %s", err)
+	}
+
+	getRespBody, err := utils.FlattenResponse(getResp)
+	if err != nil {
+		return diag.FromErr(err)
+	}
+
+	dataSourceId, err := uuid.GenerateUUID()
+	if err != nil {
+		return diag.Errorf("unable to generate ID: %s", err)
+	}
+
+	d.SetId(dataSourceId)
+
+	mErr = multierror.Append(
+		mErr,
+		d.Set("region", region),
+		d.Set("quota", utils.PathSearch("quota", getRespBody, nil)),
+		d.Set("quota_used", utils.PathSearch("quota_used", getRespBody, nil)),
+	)
+
+	return diag.FromErr(mErr.ErrorOrNil())
+}