add vpc sweeper (#925)

Author: hellertang

tencentcloud/basic_test.go

@@ -12,6 +12,11 @@ the following must be changed to your resource id.
 var appid string = os.Getenv("TENCENTCLOUD_APPID")
 var ownerUin string = os.Getenv("TENCENTCLOUD_OWNER_UIN")
 
+const (
+	keepResource = "keep"
+	defaultResource = "Default"
+)
+
 const (
 	defaultRegion      = "ap-guangzhou"
 	defaultVpcId       = "vpc-86v957zb"
@@ -50,7 +55,7 @@ const (
 // Tke Exclusive Network Environment
 const (
 	tkeExclusiveVpcId     = "vpc-391sv4w3"
-	tkeExclusiveVpcName   = "tke_exclusive_vpc"
+	tkeExclusiveVpcName   = "keep_tke_exclusive_vpc"
 	tkeExclusiveSubnetId  = "subnet-ljyn7h30"
 	defaultTkeClusterId   = "cls-ely08ic4"
 	defaultTkeClusterName = "preset_tke_cluster"

tencentcloud/common.go

@@ -81,6 +81,13 @@ func getEnvDefault(key string, defVal int) int {
 	return int
 }
 
+// string to time.Time
+func stringTotime(t string) time.Time {
+	template := "2006-01-02 15:04:05"
+	stamp, _ := time.ParseInLocation(template, t, time.Local)
+	return stamp
+}
+
 // getLogId get logId for trace, return a new logId if ctx is nil
 func getLogId(ctx context.Context) string {
 	if ctx != nil {

tencentcloud/resource_tc_vpc_test.go

@@ -3,6 +3,8 @@ package tencentcloud
 import (
 	"context"
 	"fmt"
+	"log"
+	"strings"
 	"testing"
 	"time"
 
@@ -11,6 +13,56 @@ import (
 	"github.com/hashicorp/terraform-plugin-sdk/terraform"
 )
 
+func init() {
+	resource.AddTestSweepers("tencentcloud_vpc", &resource.Sweeper{
+		Name: "tencentcloud_vpc",
+		F:    testSweepVpcInstance,
+	})
+}
+
+func testSweepVpcInstance(region string) error {
+	logId := getLogId(contextNil)
+	ctx := context.WithValue(context.TODO(), logIdKey, logId)
+
+	sharedClient, err := sharedClientForRegion(region)
+	if err != nil {
+		return fmt.Errorf("getting tencentcloud client error: %s", err.Error())
+	}
+	client := sharedClient.(*TencentCloudClient)
+
+	vpcService := VpcService{
+		client: client.apiV3Conn,
+	}
+
+	instances, err := vpcService.DescribeVpcs(ctx, "", "", nil, nil, "", "")
+	if err != nil {
+		return fmt.Errorf("get instance list error: %s", err.Error())
+	}
+
+	for _, v := range instances {
+		instanceId := v.vpcId
+		instanceName := v.name
+
+		now := time.Now()
+
+		createTime := stringTotime(v.createTime)
+		interval := now.Sub(createTime).Minutes()
+		if strings.HasPrefix(instanceName, keepResource) || strings.HasPrefix(instanceName, defaultResource) {
+			continue
+		}
+		// less than 30 minute, not delete
+		if int64(interval) < 30 {
+			continue
+		}
+
+		if err = vpcService.DeleteVpc(ctx, instanceId); err != nil {
+			log.Printf("[ERROR] sweep instance %s error: %s", instanceId, err.Error())
+		}
+	}
+
+	return nil
+}
+
 func TestAccTencentCloudVpcV3Basic(t *testing.T) {
 	t.Parallel()
 	resource.Test(t, resource.TestCase{