fix:tencent clb add sg detail fields (#89)

Co-authored-by: Lyc-heng <[email protected]>

Author: Lyc-heng

collector/tencent/collector/clb/clb.go

@@ -18,12 +18,13 @@ package clb
 import (
 	"context"
 
+	"github.com/cloudrec/tencent/collector"
 	"github.com/core-sdk/constant"
 	"github.com/core-sdk/log"
 	"github.com/core-sdk/schema"
-	"github.com/cloudrec/tencent/collector"
 	clb "github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/clb/v20180317"
 	"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common"
+	vpc "github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/vpc/v20170312"
 	"go.uber.org/zap"
 )
 
@@ -47,27 +48,31 @@ type LBDetail struct {
 	LoadBalancer clb.LoadBalancer
 	Listeners    []*clb.ListenerBackend
 	SecureGroups []*string
+	SecurityGroupDetail []*vpc.SecurityGroupPolicySet
 }
 
 func ListCLBResource(ctx context.Context, service schema.ServiceInterface, res chan<- any) error {
-	cli := service.(*collector.Services).CLB
+	services := service.(*collector.Services)
+	clbClient := services.CLB
+	vpcClient := services.VPC
 
 	request := clb.NewDescribeLoadBalancersRequest()
 	request.Limit = common.Int64Ptr(100)
 	request.Offset = common.Int64Ptr(0)
 
 	var count uint64
 	for {
-		response, err := cli.DescribeLoadBalancers(request)
+		response, err := clbClient.DescribeLoadBalancers(request)
 		if err != nil {
 			log.CtxLogger(ctx).Warn("DescribeLoadBalancers error", zap.Error(err))
 			return err
 		}
 		for _, lb := range response.Response.LoadBalancerSet {
 			d := &LBDetail{
 				LoadBalancer: *lb,
-				Listeners:    describeTargets(ctx, cli, lb.LoadBalancerId),
+				Listeners:    describeTargets(ctx, clbClient, lb.LoadBalancerId),
 				SecureGroups: lb.SecureGroups,
+				SecurityGroupDetail: describeSecurityGroups(ctx, vpcClient, lb.SecureGroups),
 			}
 			res <- d
 		}
@@ -93,3 +98,22 @@ func describeTargets(ctx context.Context, cli *clb.Client, LoadBalancerId *strin
 	}
 	return response.Response.Listeners
 }
+
+func describeSecurityGroups(ctx context.Context, cli *vpc.Client, securityGroupIds []*string) []*vpc.SecurityGroupPolicySet {
+
+	var securityGroupInfo []*vpc.SecurityGroupPolicySet
+	for _, securityGroupId := range securityGroupIds {
+		request := vpc.NewDescribeSecurityGroupPoliciesRequest()
+		request.SecurityGroupId = securityGroupId
+
+		response, err := cli.DescribeSecurityGroupPolicies(request)
+		if err != nil {
+			log.CtxLogger(ctx).Warn("Describe SecurityGroupPolicies error", zap.Error(err))
+			return nil
+		}
+
+		securityGroupInfo = append(securityGroupInfo, response.Response.SecurityGroupPolicySet)
+	}
+
+	return securityGroupInfo
+}

collector/tencent/collector/services.go

@@ -111,6 +111,10 @@ func (s *Services) InitServices(cloudAccountParam schema.CloudAccountParam) (err
 		if err != nil {
 			log.GetWLogger().Error(fmt.Sprintf("failed to initialize CLB client in region:%s, err:%s", param.Region, err.Error()))
 		}
+		s.VPC, err = createVPCClient(param.Region, s.Credential)
+		if err != nil {
+			log.GetWLogger().Error(fmt.Sprintf("failed to initialize VPC client in region:%s, err:%s", param.Region, err.Error()))
+		}
 	case CDB:
 		s.CDB, err = createCDBClient(param.Region, s.Credential)
 		if err != nil {