feat:support dbbrain 2 datasource (#1425)

Author: lyu571

tencentcloud/data_source_tc_dbbrain_security_audit_log_export_tasks.go

@@ -0,0 +1,227 @@
+/*
+Use this data source to query detailed information of dbbrain securityAuditLogExportTasks
+
+Example Usage
+
+```hcl
+resource "tencentcloud_dbbrain_security_audit_log_export_task" "task" {
+  sec_audit_group_id = "sec_audit_group_id"
+  start_time = "start_time"
+  end_time = "end_time"
+  product = "mysql"
+  danger_levels = [0,1,2]
+}
+
+data "tencentcloud_dbbrain_security_audit_log_export_tasks" "tasks" {
+	sec_audit_group_id = "sec_audit_group_id"
+	product = "mysql"
+	async_request_ids = [tencentcloud_dbbrain_security_audit_log_export_task.task.async_request_id]
+}
+```
+*/
+package tencentcloud
+
+import (
+	"context"
+	"log"
+
+	"github.com/hashicorp/terraform-plugin-sdk/helper/resource"
+	"github.com/hashicorp/terraform-plugin-sdk/helper/schema"
+	dbbrain "github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/dbbrain/v20210527"
+	"github.com/tencentcloudstack/terraform-provider-tencentcloud/tencentcloud/internal/helper"
+)
+
+func dataSourceTencentCloudDbbrainSecurityAuditLogExportTasks() *schema.Resource {
+	return &schema.Resource{
+		Read: dataSourceTencentCloudDbbrainSecurityAuditLogExportTasksRead,
+		Schema: map[string]*schema.Schema{
+			"sec_audit_group_id": {
+				Type:        schema.TypeString,
+				Required:    true,
+				Description: "security audit group id.",
+			},
+
+			"product": {
+				Type:        schema.TypeString,
+				Required:    true,
+				Description: "product, optional value is mysql.",
+			},
+
+			"async_request_ids": {
+				Type: schema.TypeSet,
+				Elem: &schema.Schema{
+					Type: schema.TypeInt,
+				},
+				Optional:    true,
+				Description: "async request id list.",
+			},
+
+			"list": {
+				Type:        schema.TypeList,
+				Computed:    true,
+				Description: "security audit log export task list.",
+				Elem: &schema.Resource{
+					Schema: map[string]*schema.Schema{
+						"async_request_id": {
+							Type:        schema.TypeInt,
+							Computed:    true,
+							Description: "async request id.",
+						},
+						"start_time": {
+							Type:        schema.TypeString,
+							Computed:    true,
+							Description: "start time.",
+						},
+						"end_time": {
+							Type:        schema.TypeString,
+							Computed:    true,
+							Description: "end time.",
+						},
+						"create_time": {
+							Type:        schema.TypeString,
+							Computed:    true,
+							Description: "create time.",
+						},
+						"status": {
+							Type:        schema.TypeString,
+							Computed:    true,
+							Description: "status.",
+						},
+						"progress": {
+							Type:        schema.TypeInt,
+							Computed:    true,
+							Description: "task progress.",
+						},
+						"log_start_time": {
+							Type:        schema.TypeString,
+							Computed:    true,
+							Description: "log start time.",
+						},
+						"log_end_time": {
+							Type:        schema.TypeString,
+							Computed:    true,
+							Description: "log end time.",
+						},
+						"total_size": {
+							Type:        schema.TypeInt,
+							Computed:    true,
+							Description: "the total size of log.",
+						},
+						"danger_levels": {
+							Type: schema.TypeSet,
+							Elem: &schema.Schema{
+								Type: schema.TypeInt,
+							},
+							Computed:    true,
+							Description: "danger level list.",
+						},
+					},
+				},
+			},
+
+			"result_output_file": {
+				Type:        schema.TypeString,
+				Optional:    true,
+				Description: "Used to save results.",
+			},
+		},
+	}
+}
+
+func dataSourceTencentCloudDbbrainSecurityAuditLogExportTasksRead(d *schema.ResourceData, meta interface{}) error {
+	defer logElapsed("data_source.tencentcloud_dbbrain_security_audit_log_export_tasks.read")()
+	defer inconsistentCheck(d, meta)()
+
+	logId := getLogId(contextNil)
+	ctx := context.WithValue(context.TODO(), logIdKey, logId)
+	var sag_id string
+
+	paramMap := make(map[string]interface{})
+	if v, ok := d.GetOk("sec_audit_group_id"); ok {
+		paramMap["sec_audit_group_id"] = helper.String(v.(string))
+		sag_id = v.(string)
+	}
+
+	if v, ok := d.GetOk("product"); ok {
+		paramMap["product"] = helper.String(v.(string))
+	}
+
+	if v, ok := d.GetOk("async_request_ids"); ok {
+		async_request_idSet := v.(*schema.Set).List()
+		tmpList := make([]*uint64, 0, len(async_request_idSet))
+		for i := range async_request_idSet {
+			async_request_id := async_request_idSet[i].(int)
+			tmpList = append(tmpList, helper.IntUint64(async_request_id))
+		}
+		paramMap["async_request_ids"] = tmpList
+	}
+
+	dbbrainService := DbbrainService{client: meta.(*TencentCloudClient).apiV3Conn}
+
+	var tasks []*dbbrain.SecLogExportTaskInfo
+	err := resource.Retry(readRetryTimeout, func() *resource.RetryError {
+		results, e := dbbrainService.DescribeDbbrainSecurityAuditLogExportTasksByFilter(ctx, paramMap)
+		if e != nil {
+			return retryError(e)
+		}
+		tasks = results
+		return nil
+	})
+	if err != nil {
+		log.Printf("[CRITAL]%s read Dbbrain tasks failed, reason:%+v", logId, err)
+		return err
+	}
+
+	ids := make([]string, 0, len(tasks))
+	taskList := make([]map[string]interface{}, 0, len(tasks))
+
+	if tasks != nil {
+
+		for _, task := range tasks {
+			taskMap := map[string]interface{}{}
+			if task.AsyncRequestId != nil {
+				taskMap["async_request_id"] = task.AsyncRequestId
+			}
+			if task.StartTime != nil {
+				taskMap["start_time"] = task.StartTime
+			}
+			if task.EndTime != nil {
+				taskMap["end_time"] = task.EndTime
+			}
+			if task.CreateTime != nil {
+				taskMap["create_time"] = task.CreateTime
+			}
+			if task.Status != nil {
+				taskMap["status"] = task.Status
+			}
+			if task.Progress != nil {
+				taskMap["progress"] = task.Progress
+			}
+			if task.LogStartTime != nil {
+				taskMap["log_start_time"] = task.LogStartTime
+			}
+			if task.LogEndTime != nil {
+				taskMap["log_end_time"] = task.LogEndTime
+			}
+			if task.TotalSize != nil {
+				taskMap["total_size"] = task.TotalSize
+			}
+			if task.DangerLevels != nil {
+				taskMap["danger_levels"] = task.DangerLevels
+			}
+			ids = append(ids, sag_id+FILED_SP+helper.UInt64ToStr(*task.AsyncRequestId))
+			taskList = append(taskList, taskMap)
+		}
+		d.SetId(helper.DataResourceIdsHash(ids))
+		_ = d.Set("list", taskList)
+	}
+
+	output, ok := d.GetOk("result_output_file")
+	if ok && output.(string) != "" {
+		if e := writeToFile(output.(string), taskList); e != nil {
+			return e
+		}
+	}
+
+	return nil
+}

tencentcloud/data_source_tc_dbbrain_security_audit_log_export_tasks_test.go

@@ -0,0 +1,50 @@
+package tencentcloud
+
+import (
+	"fmt"
+	"testing"
+	"time"
+
+	"github.com/hashicorp/terraform-plugin-sdk/helper/resource"
+)
+
+func TestAccTencentCloudDbbrainSecurityAuditLogExportTasksDataSource(t *testing.T) {
+	t.Parallel()
+	loc, _ := time.LoadLocation("Asia/Chongqing")
+	startTime := time.Now().Add(-2 * time.Hour).In(loc).Format("2006-01-02T15:04:05+08:00")
+	endTime := time.Now().Add(2 * time.Hour).In(loc).Format("2006-01-02T15:04:05+08:00")
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccDataSourceDbbrainSecurityAuditLogExportTasks(startTime, endTime),
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckTencentCloudDataSourceID("data.tencentcloud_dbbrain_security_audit_log_export_tasks.tasks"),
+					resource.TestCheckResourceAttrSet("data.tencentcloud_dbbrain_security_audit_log_export_tasks.tasks", "list.#"),
+				),
+			},
+		},
+	})
+}
+
+func testAccDataSourceDbbrainSecurityAuditLogExportTasks(st, et string) string {
+	return fmt.Sprintf(`
+
+resource "tencentcloud_dbbrain_security_audit_log_export_task" "task" {
+  sec_audit_group_id = "%s"
+  start_time = "%s"
+  end_time = "%s"
+  product = "mysql"
+  danger_levels = [0,1,2]
+}
+
+data "tencentcloud_dbbrain_security_audit_log_export_tasks" "tasks" {
+	sec_audit_group_id = "%s"
+	product = "mysql"
+	async_request_ids = [tencentcloud_dbbrain_security_audit_log_export_task.task.async_request_id]
+}
+
+`, defaultDbBrainsagId, st, et, defaultDbBrainsagId)
+}