support sub_domains

Author: mikatong

tencentcloud/services/dnspod/data_source_tc_dnspod_record_list.go

@@ -29,9 +29,17 @@ func DataSourceTencentCloudDnspodRecordList() *schema.Resource {
 			},
 
 			"sub_domain": {
-				Optional:    true,
-				Type:        schema.TypeString,
-				Description: "Retrieve resolution records based on the host header of the resolution record. Fuzzy matching is used by default. You can set the IsExactSubdomain parameter to true for precise searching.",
+				Optional:      true,
+				Type:          schema.TypeString,
+				ConflictsWith: []string{"sub_domains"},
+				Description:   "Retrieve resolution records based on the host header of the resolution record. Fuzzy matching is used by default. You can set the IsExactSubdomain parameter to true for precise searching.",
+			},
+			"sub_domains": {
+				Optional:      true,
+				Type:          schema.TypeSet,
+				Elem:          &schema.Schema{Type: schema.TypeString},
+				ConflictsWith: []string{"sub_domain"},
+				Description:   "Sub domains",
 			},
 
 			"record_type": {
@@ -381,8 +389,16 @@ func dataSourceTencentCloudDnspodRecordListRead(d *schema.ResourceData, meta int
 		paramMap["DomainId"] = helper.IntUint64(v.(int))
 	}
 
+	subDomains := make([]string, 0)
 	if v, ok := d.GetOk("sub_domain"); ok {
-		paramMap["SubDomain"] = helper.String(v.(string))
+		subDomains = append(subDomains, v.(string))
+	}
+
+	if v, ok := d.GetOk("sub_domains"); ok {
+		subDomainList := v.(*schema.Set).List()
+		for _, subDomain := range subDomainList {
+			subDomains = append(subDomains, subDomain.(string))
+		}
 	}
 
 	if v, ok := d.GetOk("record_type"); ok {
@@ -472,16 +488,19 @@ func dataSourceTencentCloudDnspodRecordListRead(d *schema.ResourceData, meta int
 
 	var recordList []*dnspod.RecordListItem
 
-	err := resource.Retry(tccommon.ReadRetryTimeout, func() *resource.RetryError {
-		result, e := service.DescribeDnspodRecordListByFilter(ctx, paramMap)
-		if e != nil {
-			return tccommon.RetryError(e)
+	for _, subDomain := range subDomains {
+		paramMap["SubDomain"] = helper.String(subDomain)
+		err := resource.Retry(tccommon.ReadRetryTimeout, func() *resource.RetryError {
+			result, e := service.DescribeDnspodRecordListByFilter(ctx, paramMap)
+			if e != nil {
+				return tccommon.RetryError(e)
+			}
+			recordList = append(recordList, result...)
+			return nil
+		})
+		if err != nil {
+			return err
 		}
-		recordList = result
-		return nil
-	})
-	if err != nil {
-		return err
 	}
 
 	ids := make([]string, 0, len(recordList))

tencentcloud/services/dnspod/data_source_tc_dnspod_record_list_test.go

@@ -11,7 +11,7 @@ import (
 func TestAccTencentCloudDnspodRecordListDataSource_basic(t *testing.T) {
 	t.Parallel()
 	resource.Test(t, resource.TestCase{
-		PreCheck:  func() { tcacctest.AccPreCheckCommon(t, tcacctest.ACCOUNT_TYPE_PREPAY) },
+		PreCheck:  func() { tcacctest.AccPreCheck(t) },
 		Providers: tcacctest.AccProviders,
 		Steps: []resource.TestStep{
 			{
@@ -22,6 +22,24 @@ func TestAccTencentCloudDnspodRecordListDataSource_basic(t *testing.T) {
 	})
 }
 
+func TestAccTencentCloudDnspodRecordListDataSource_subDomains(t *testing.T) {
+	t.Parallel()
+	resource.Test(t, resource.TestCase{
+		PreCheck:  func() { tcacctest.AccPreCheck(t) },
+		Providers: tcacctest.AccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccDnspodRecordListDataSource_subDomains,
+				Check: resource.ComposeTestCheckFunc(
+					tcacctest.AccCheckTencentCloudDataSourceID("data.tencentcloud_dnspod_record_list.subdomains"),
+					resource.TestCheckResourceAttr("data.tencentcloud_dnspod_record_list.subdomains", "record_list.#", "2"),
+					resource.TestCheckResourceAttr("data.tencentcloud_dnspod_record_list.subdomains", "instance_list.#", "2"),
+				),
+			},
+		},
+	})
+}
+
 const testAccDnspodRecordListDataSource = `
 
 data "tencentcloud_dnspod_record_list" "record_list" {
@@ -50,3 +68,11 @@ data "tencentcloud_dnspod_record_list" "record_list" {
 }
 
 `
+
+const testAccDnspodRecordListDataSource_subDomains = `
+data "tencentcloud_dnspod_record_list" "subdomains" {
+  domain              = "mikatong.xyz"
+  is_exact_sub_domain = true
+  sub_domains          = ["tes1029","tes103"]
+}
+`