Feat/support cos acceleration (#1537)

* update dts mode

* feat:support acceleration field for cos bucket

* add changelog

* use tc client instead of s3 client

Author: lyu571

.changelog/1537.txt

@@ -0,0 +1,3 @@
+```release-note:enhancement
+resource/tencentcloud_cos_bucket: add `acceleration_enable` field
+```

tencentcloud/resource_tc_cos_bucket.go

@@ -241,8 +241,10 @@ package tencentcloud
 import (
 	"bytes"
 	"context"
+	"encoding/json"
 	"encoding/xml"
 	"fmt"
+	"io/ioutil"
 	"log"
 	"reflect"
 	"time"
@@ -434,6 +436,12 @@ func resourceTencentCloudCosBucket() *schema.Resource {
 				Default:     false,
 				Description: "Enable bucket versioning.",
 			},
+			"acceleration_enable": {
+				Type:        schema.TypeBool,
+				Optional:    true,
+				Default:     false,
+				Description: "Enable bucket acceleration.",
+			},
 			"force_clean": {
 				Type:        schema.TypeBool,
 				Optional:    true,
@@ -905,6 +913,15 @@ func resourceTencentCloudCosBucketRead(d *schema.ResourceData, meta interface{})
 		return fmt.Errorf("setting versioning_enable error: %v", err)
 	}
 
+	// read the acceleration
+	acceleration, err := cosService.GetBucketAccleration(ctx, bucket)
+	if err != nil {
+		return err
+	}
+	if err = d.Set("acceleration_enable", acceleration); err != nil {
+		return fmt.Errorf("setting acceleration_enable error: %v", err)
+	}
+
 	replicaResult, err := cosService.GetBucketReplication(ctx, bucket)
 	if err != nil {
 		return err
@@ -950,6 +967,7 @@ func resourceTencentCloudCosBucketUpdate(d *schema.ResourceData, meta interface{
 	ctx := context.WithValue(context.TODO(), logIdKey, logId)
 
 	client := meta.(*TencentCloudClient).apiV3Conn.UseCosClient()
+	tcClient := meta.(*TencentCloudClient).apiV3Conn.UseTencentCosClient(d.Id())
 	cosService := CosService{client: meta.(*TencentCloudClient).apiV3Conn}
 
 	d.Partial(true)
@@ -1027,6 +1045,14 @@ func resourceTencentCloudCosBucketUpdate(d *schema.ResourceData, meta interface{
 		d.SetPartial("versioning_enable")
 	}
 
+	if d.HasChange("acceleration_enable") {
+		err := resourceTencentCloudCosBucketAccelerationUpdate(ctx, tcClient, d)
+		if err != nil {
+			return err
+		}
+		d.SetPartial("acceleration_enable")
+	}
+
 	if d.HasChange("replica_role") || d.HasChange("replica_rules") {
 		err := resourceTencentCloudCosBucketReplicaUpdate(ctx, cosService, d)
 
@@ -1163,6 +1189,33 @@ func resourceTencentCloudCosBucketVersioningUpdate(ctx context.Context, client *
 	return nil
 }
 
+func resourceTencentCloudCosBucketAccelerationUpdate(ctx context.Context, client *cos.Client, d *schema.ResourceData) error {
+	logId := getLogId(ctx)
+
+	bucket := d.Get("bucket").(string)
+	enabled := d.Get("acceleration_enable").(bool)
+	status := "Suspended"
+	if enabled {
+		status = "Enabled"
+	}
+
+	opt := &cos.BucketPutAccelerateOptions{
+		Status: status,
+	}
+	response, err := client.Bucket.PutAccelerate(ctx, opt)
+	if err != nil {
+		log.Printf("[CRITAL]%s api[%s] fail, status [%s], reason[%s]\n",
+			logId, "put bucket acceleration", opt.Status, err.Error())
+		return fmt.Errorf("cos put bucket acceleration error: %s, bucket: %s", err.Error(), bucket)
+	}
+	rb, _ := ioutil.ReadAll(response.Body)
+	body, _ := json.Marshal(rb)
+	log.Printf("[DEBUG]%s api[%s] success, status [%s], response body [%s]\n",
+		logId, "put bucket acceleration", opt.Status, string(body))
+
+	return err
+}
+
 func resourceTencentCloudCosBucketReplicaUpdate(ctx context.Context, service CosService, d *schema.ResourceData) error {
 	bucket := d.Get("bucket").(string)
 	oldRole, newRole := d.GetChange("replica_role")

tencentcloud/resource_tc_cos_bucket_test.go

@@ -87,6 +87,7 @@ func TestAccTencentCloudCosBucketResource_basic(t *testing.T) {
 					testAccCheckCosBucketExists("tencentcloud_cos_bucket.bucket_basic"),
 					resource.TestCheckResourceAttr("tencentcloud_cos_bucket.bucket_basic", "encryption_algorithm", "AES256"),
 					resource.TestCheckResourceAttr("tencentcloud_cos_bucket.bucket_basic", "versioning_enable", "true"),
+					resource.TestCheckResourceAttr("tencentcloud_cos_bucket.bucket_basic", "acceleration_enable", "true"),
 					resource.TestCheckResourceAttr("tencentcloud_cos_bucket.bucket_basic", "force_clean", "true"),
 					resource.TestCheckResourceAttrSet("tencentcloud_cos_bucket.bucket_basic", "cos_bucket_url"),
 				),
@@ -559,6 +560,7 @@ resource "tencentcloud_cos_bucket" "bucket_basic" {
   acl                  = "private"
   encryption_algorithm = "AES256"
   versioning_enable    = true
+  acceleration_enable  = true
   force_clean          = true
 }
 `, userInfoData)

tencentcloud/service_tencentcloud_cos.go

@@ -737,6 +737,37 @@ func (me *CosService) GetBucketVersioning(ctx context.Context, bucket string) (v
 	return
 }
 
+func (me *CosService) GetBucketAccleration(ctx context.Context, bucket string) (accelerationEnable bool, errRet error) {
+	logId := getLogId(ctx)
+
+	request := s3.GetBucketAccelerateConfigurationInput{
+		Bucket: aws.String(bucket),
+	}
+	ratelimit.Check("GetBucketAccelerateConfiguration")
+	response, err := me.client.UseCosClient().GetBucketAccelerateConfiguration(&request)
+	if err != nil {
+		awsError, ok := err.(awserr.Error)
+		if ok && awsError.Code() == "NoSuchAccelerateConfiguration" {
+			return
+		}
+		log.Printf("[CRITAL]%s api[%s] fail, request body [%s], reason[%s]\n",
+			logId, "get bucket acceleration", request.String(), err.Error())
+		errRet = fmt.Errorf("cos get bucket acceleration error: %s, bucket: %s", err.Error(), bucket)
+		return
+	}
+
+	log.Printf("[DEBUG]%s api[%s] success, request body [%s], response body [%s]\n",
+		logId, "get bucket acceleration", request.String(), response.String())
+
+	if response.Status == nil || *response.Status == "Suspended" {
+		accelerationEnable = false
+	} else if *response.Status == "Enabled" {
+		accelerationEnable = true
+	}
+
+	return
+}
+
 func (me *CosService) GetBucketLogStatus(ctx context.Context, bucket string) (logEnable bool, logTargetBucket string, logPrefix string, errRet error) {
 	logId := getLogId(ctx)
 

website/docs/r/cos_bucket.html.markdown

@@ -242,6 +242,7 @@ resource "tencentcloud_cos_bucket" "mycos" {
 The following arguments are supported:
 
 * `bucket` - (Required, String, ForceNew) The name of a bucket to be created. Bucket format should be [custom name]-[appid], for example `mycos-1258798060`.
+* `acceleration_enable` - (Optional, Bool) Enable bucket acceleration.
 * `acl_body` - (Optional, String) ACL XML body for multiple grant info. NOTE: this argument will overwrite `acl`. Check https://intl.cloud.tencent.com/document/product/436/7737 for more detail.
 * `acl` - (Optional, String) The canned ACL to apply. Valid values: private, public-read, and public-read-write. Defaults to private.
 * `cors_rules` - (Optional, List) A rule of Cross-Origin Resource Sharing (documented below).