update at 2024-08-09 14:15:25

Author: imroc

cmd/app/cmd.go

@@ -16,7 +16,13 @@ var RootCommand = cobra.Command{
 	Use:   "tke-extend-network-controller",
 	Short: "A network controller for TKE",
 	Run: func(cmd *cobra.Command, args []string) {
-		clb.Init(viper.GetString(secretId), viper.GetString(secretKey), viper.GetString(region), viper.GetString(vpcId))
+		clb.Init(
+			viper.GetString(secretId),
+			viper.GetString(secretKey),
+			viper.GetString(region),
+			viper.GetString(vpcId),
+			viper.GetString(clusterId),
+		)
 		runManager()
 	},
 }
@@ -29,6 +35,7 @@ const (
 	secretKey              = "secret-key"
 	region                 = "region"
 	vpcId                  = "vpcid"
+	clusterId              = "clusterid"
 )
 
 var (

pkg/clb/clb.go

@@ -8,9 +8,9 @@ import (
 
 var credential *common.Credential
 
-var defaultRegion, defaultVpcId string
+var defaultRegion, defaultVpcId, clusterId string
 
-func Init(secretId, secretKey, region, vpcId string) {
+func Init(secretId, secretKey, region, vpcId, clusterID string) {
 	if secretId == "" || secretKey == "" {
 		panic("secretId and secretKey are required")
 	}
@@ -23,6 +23,7 @@ func Init(secretId, secretKey, region, vpcId string) {
 	}
 	defaultRegion = region
 	defaultVpcId = vpcId
+	clusterId = clusterID
 }
 
 func DefaultRegion() string {

pkg/clb/instance.go

@@ -53,6 +53,16 @@ func Create(ctx context.Context, region, vpcId, name string) (lbId string, err e
 	req := clb.NewCreateLoadBalancerRequest()
 	req.LoadBalancerType = common.StringPtr("OPEN")
 	req.VpcId = &vpcId
+	req.Tags = append(req.Tags,
+		&clb.TagInfo{
+			TagKey:   common.StringPtr("tke-clusterId"), // 与集群关联
+			TagValue: common.StringPtr(clusterId),
+		},
+		&clb.TagInfo{
+			TagKey:   common.StringPtr("tke-createdBy-flag"), // 用于删除集群时自动清理集群关联的自动创建的 CLB
+			TagValue: common.StringPtr("yes"),
+		},
+	)
 	client := GetClient(region)
 	resp, err := client.CreateLoadBalancer(req)
 	if err != nil {