Tag Azure resources with UID

2024-10-01 01:36:09 -04:00 · 2022-08-29 11:55:46 +02:00 · 2022-08-29 11:55:46 +02:00 · 1861dc2744
commit 1861dc2744
parent 69abe17c96
5 changed files with 74 additions and 25 deletions
--- a/cli/internal/azure/client/applicationinsight.go
+++ b/cli/internal/azure/client/applicationinsight.go
@ -14,6 +14,7 @@ func (c *Client) CreateApplicationInsight(ctx context.Context) error {
 		Properties: &armapplicationinsights.ComponentProperties{
 			ApplicationType: to.Ptr(armapplicationinsights.ApplicationTypeWeb),
 		},
+		Tags: map[string]*string{"uid": to.Ptr(c.uid)},
 	}

 	_, err := c.applicationInsightsAPI.CreateOrUpdate(
--- a/cli/internal/azure/client/compute.go
+++ b/cli/internal/azure/client/compute.go
@ -113,6 +113,7 @@ func (c *Client) createScaleSet(ctx context.Context, input CreateScaleSetInput)
 	scaleSet := azure.ScaleSet{
 		Name:                           input.Name,
 		NamePrefix:                     input.NamePrefix,
+		UID:                            c.uid,
 		Location:                       c.location,
 		InstanceType:                   input.InstanceType,
 		StateDiskSizeGB:                input.StateDiskSizeGB,
--- a/cli/internal/azure/client/network.go
+++ b/cli/internal/azure/client/network.go
@ -40,6 +40,7 @@ func (c *Client) CreateVirtualNetwork(ctx context.Context) error {
 		ctx, c.resourceGroup, createNetworkInput.name,
 		armnetwork.VirtualNetwork{
 			Name:     to.Ptr(createNetworkInput.name), // this is supposed to be read-only
+			Tags:     map[string]*string{"uid": to.Ptr(c.uid)},
 			Location: to.Ptr(createNetworkInput.location),
 			Properties: &armnetwork.VirtualNetworkPropertiesFormat{
 				AddressSpace: &armnetwork.AddressSpace{
@ -101,6 +102,7 @@ func (c *Client) CreateSecurityGroup(ctx context.Context, input NetworkSecurityG
 		ctx, c.resourceGroup, createNetworkSecurityGroupInput.name,
 		armnetwork.SecurityGroup{
 			Name:     to.Ptr(createNetworkSecurityGroupInput.name),
+			Tags:     map[string]*string{"uid": to.Ptr(c.uid)},
 			Location: to.Ptr(createNetworkSecurityGroupInput.location),
 			Properties: &armnetwork.SecurityGroupPropertiesFormat{
 				SecurityRules: createNetworkSecurityGroupInput.rules,
@ -125,6 +127,7 @@ func (c *Client) createPublicIPAddress(ctx context.Context, name string) (*armne
 	poller, err := c.publicIPAddressesAPI.BeginCreateOrUpdate(
 		ctx, c.resourceGroup, name,
 		armnetwork.PublicIPAddress{
+			Tags:     map[string]*string{"uid": to.Ptr(c.uid)},
 			Location: to.Ptr(c.location),
 			SKU: &armnetwork.PublicIPAddressSKU{
 				Name: to.Ptr(armnetwork.PublicIPAddressSKUNameStandard),
--- a/cli/internal/azure/loadbalancer.go
+++ b/cli/internal/azure/loadbalancer.go
@ -47,15 +47,9 @@ func (l LoadBalancer) Azure() armnetwork.LoadBalancer {
 				},
 			},
 			BackendAddressPools: []*armnetwork.BackendAddressPool{
-				{
-					Name: to.Ptr(backEndAddressPoolNodeName),
-				},
-				{
-					Name: to.Ptr(backEndAddressPoolControlPlaneName),
-				},
-				{
-					Name: to.Ptr("all"),
-				},
+				{Name: to.Ptr(backEndAddressPoolNodeName)},
+				{Name: to.Ptr(backEndAddressPoolControlPlaneName)},
+				{Name: to.Ptr("all")},
 			},
 			Probes: []*armnetwork.Probe{
 				{
@ -92,18 +86,27 @@ func (l LoadBalancer) Azure() armnetwork.LoadBalancer {
 					Name: to.Ptr("kubeLoadBalancerRule"),
 					Properties: &armnetwork.LoadBalancingRulePropertiesFormat{
 						FrontendIPConfiguration: &armnetwork.SubResource{
-							ID: to.Ptr("/subscriptions/" + l.Subscription + "/resourceGroups/" + l.ResourceGroup + "/providers/Microsoft.Network/loadBalancers/" + l.Name + "/frontendIPConfigurations/" + frontEndIPConfigName),
+							ID: to.Ptr("/subscriptions/" + l.Subscription +
+								"/resourceGroups/" + l.ResourceGroup +
+								"/providers/Microsoft.Network/loadBalancers/" + l.Name +
+								"/frontendIPConfigurations/" + frontEndIPConfigName),
 						},
 						FrontendPort: to.Ptr[int32](constants.KubernetesPort),
 						BackendPort:  to.Ptr[int32](constants.KubernetesPort),
 						Protocol:     to.Ptr(armnetwork.TransportProtocolTCP),
 						Probe: &armnetwork.SubResource{
-							ID: to.Ptr("/subscriptions/" + l.Subscription + "/resourceGroups/" + l.ResourceGroup + "/providers/Microsoft.Network/loadBalancers/" + l.Name + "/probes/" + kubeHealthProbeName),
+							ID: to.Ptr("/subscriptions/" + l.Subscription +
+								"/resourceGroups/" + l.ResourceGroup +
+								"/providers/Microsoft.Network/loadBalancers/" + l.Name +
+								"/probes/" + kubeHealthProbeName),
 						},
 						DisableOutboundSnat: to.Ptr(true),
 						BackendAddressPools: []*armnetwork.SubResource{
 							{
-								ID: to.Ptr("/subscriptions/" + l.Subscription + "/resourceGroups/" + l.ResourceGroup + "/providers/Microsoft.Network/loadBalancers/" + l.Name + "/backendAddressPools/" + backEndAddressPoolControlPlaneName),
+								ID: to.Ptr("/subscriptions/" + l.Subscription +
+									"/resourceGroups/" + l.ResourceGroup +
+									"/providers/Microsoft.Network/loadBalancers/" + l.Name +
+									"/backendAddressPools/" + backEndAddressPoolControlPlaneName),
 							},
 						},
 					},
@ -112,18 +115,27 @@ func (l LoadBalancer) Azure() armnetwork.LoadBalancer {
 					Name: to.Ptr("verifyLoadBalancerRule"),
 					Properties: &armnetwork.LoadBalancingRulePropertiesFormat{
 						FrontendIPConfiguration: &armnetwork.SubResource{
-							ID: to.Ptr("/subscriptions/" + l.Subscription + "/resourceGroups/" + l.ResourceGroup + "/providers/Microsoft.Network/loadBalancers/" + l.Name + "/frontendIPConfigurations/" + frontEndIPConfigName),
+							ID: to.Ptr("/subscriptions/" + l.Subscription +
+								"/resourceGroups/" + l.ResourceGroup +
+								"/providers/Microsoft.Network/loadBalancers/" + l.Name +
+								"/frontendIPConfigurations/" + frontEndIPConfigName),
 						},
 						FrontendPort: to.Ptr[int32](constants.VerifyServiceNodePortGRPC),
 						BackendPort:  to.Ptr[int32](constants.VerifyServiceNodePortGRPC),
 						Protocol:     to.Ptr(armnetwork.TransportProtocolTCP),
 						Probe: &armnetwork.SubResource{
-							ID: to.Ptr("/subscriptions/" + l.Subscription + "/resourceGroups/" + l.ResourceGroup + "/providers/Microsoft.Network/loadBalancers/" + l.Name + "/probes/" + verifyHealthProbeName),
+							ID: to.Ptr("/subscriptions/" + l.Subscription +
+								"/resourceGroups/" + l.ResourceGroup +
+								"/providers/Microsoft.Network/loadBalancers/" + l.Name +
+								"/probes/" + verifyHealthProbeName),
 						},
 						DisableOutboundSnat: to.Ptr(true),
 						BackendAddressPools: []*armnetwork.SubResource{
 							{
-								ID: to.Ptr("/subscriptions/" + l.Subscription + "/resourceGroups/" + l.ResourceGroup + "/providers/Microsoft.Network/loadBalancers/" + l.Name + "/backendAddressPools/" + backEndAddressPoolControlPlaneName),
+								ID: to.Ptr("/subscriptions/" + l.Subscription +
+									"/resourceGroups/" + l.ResourceGroup +
+									"/providers/Microsoft.Network/loadBalancers/" + l.Name +
+									"/backendAddressPools/" + backEndAddressPoolControlPlaneName),
 							},
 						},
 					},
@ -132,18 +144,27 @@ func (l LoadBalancer) Azure() armnetwork.LoadBalancer {
 					Name: to.Ptr("coordLoadBalancerRule"),
 					Properties: &armnetwork.LoadBalancingRulePropertiesFormat{
 						FrontendIPConfiguration: &armnetwork.SubResource{
-							ID: to.Ptr("/subscriptions/" + l.Subscription + "/resourceGroups/" + l.ResourceGroup + "/providers/Microsoft.Network/loadBalancers/" + l.Name + "/frontendIPConfigurations/" + frontEndIPConfigName),
+							ID: to.Ptr("/subscriptions/" + l.Subscription +
+								"/resourceGroups/" + l.ResourceGroup +
+								"/providers/Microsoft.Network/loadBalancers/" + l.Name +
+								"/frontendIPConfigurations/" + frontEndIPConfigName),
 						},
 						FrontendPort: to.Ptr[int32](constants.BootstrapperPort),
 						BackendPort:  to.Ptr[int32](constants.BootstrapperPort),
 						Protocol:     to.Ptr(armnetwork.TransportProtocolTCP),
 						Probe: &armnetwork.SubResource{
-							ID: to.Ptr("/subscriptions/" + l.Subscription + "/resourceGroups/" + l.ResourceGroup + "/providers/Microsoft.Network/loadBalancers/" + l.Name + "/probes/" + coordHealthProbeName),
+							ID: to.Ptr("/subscriptions/" + l.Subscription +
+								"/resourceGroups/" + l.ResourceGroup +
+								"/providers/Microsoft.Network/loadBalancers/" + l.Name +
+								"/probes/" + coordHealthProbeName),
 						},
 						DisableOutboundSnat: to.Ptr(true),
 						BackendAddressPools: []*armnetwork.SubResource{
 							{
-								ID: to.Ptr("/subscriptions/" + l.Subscription + "/resourceGroups/" + l.ResourceGroup + "/providers/Microsoft.Network/loadBalancers/" + l.Name + "/backendAddressPools/" + backEndAddressPoolControlPlaneName),
+								ID: to.Ptr("/subscriptions/" + l.Subscription +
+									"/resourceGroups/" + l.ResourceGroup +
+									"/providers/Microsoft.Network/loadBalancers/" + l.Name +
+									"/backendAddressPools/" + backEndAddressPoolControlPlaneName),
 							},
 						},
 					},
@ -152,18 +173,27 @@ func (l LoadBalancer) Azure() armnetwork.LoadBalancer {
 					Name: to.Ptr("debudLoadBalancerRule"),
 					Properties: &armnetwork.LoadBalancingRulePropertiesFormat{
 						FrontendIPConfiguration: &armnetwork.SubResource{
-							ID: to.Ptr("/subscriptions/" + l.Subscription + "/resourceGroups/" + l.ResourceGroup + "/providers/Microsoft.Network/loadBalancers/" + l.Name + "/frontendIPConfigurations/" + frontEndIPConfigName),
+							ID: to.Ptr("/subscriptions/" + l.Subscription +
+								"/resourceGroups/" + l.ResourceGroup +
+								"/providers/Microsoft.Network/loadBalancers/" + l.Name +
+								"/frontendIPConfigurations/" + frontEndIPConfigName),
 						},
 						FrontendPort: to.Ptr[int32](constants.DebugdPort),
 						BackendPort:  to.Ptr[int32](constants.DebugdPort),
 						Protocol:     to.Ptr(armnetwork.TransportProtocolTCP),
 						Probe: &armnetwork.SubResource{
-							ID: to.Ptr("/subscriptions/" + l.Subscription + "/resourceGroups/" + l.ResourceGroup + "/providers/Microsoft.Network/loadBalancers/" + l.Name + "/probes/" + debugdHealthProbeName),
+							ID: to.Ptr("/subscriptions/" + l.Subscription +
+								"/resourceGroups/" + l.ResourceGroup +
+								"/providers/Microsoft.Network/loadBalancers/" + l.Name +
+								"/probes/" + debugdHealthProbeName),
 						},
 						DisableOutboundSnat: to.Ptr(true),
 						BackendAddressPools: []*armnetwork.SubResource{
 							{
-								ID: to.Ptr("/subscriptions/" + l.Subscription + "/resourceGroups/" + l.ResourceGroup + "/providers/Microsoft.Network/loadBalancers/" + l.Name + "/backendAddressPools/" + backEndAddressPoolControlPlaneName),
+								ID: to.Ptr("/subscriptions/" + l.Subscription +
+									"/resourceGroups/" + l.ResourceGroup +
+									"/providers/Microsoft.Network/loadBalancers/" + l.Name +
+									"/backendAddressPools/" + backEndAddressPoolControlPlaneName),
 							},
 						},
 					},
@ -175,11 +205,17 @@ func (l LoadBalancer) Azure() armnetwork.LoadBalancer {
 					Properties: &armnetwork.OutboundRulePropertiesFormat{
 						FrontendIPConfigurations: []*armnetwork.SubResource{
 							{
-								ID: to.Ptr("/subscriptions/" + l.Subscription + "/resourceGroups/" + l.ResourceGroup + "/providers/Microsoft.Network/loadBalancers/" + l.Name + "/frontendIPConfigurations/" + frontEndIPConfigName),
+								ID: to.Ptr("/subscriptions/" + l.Subscription +
+									"/resourceGroups/" + l.ResourceGroup +
+									"/providers/Microsoft.Network/loadBalancers/" + l.Name +
+									"/frontendIPConfigurations/" + frontEndIPConfigName),
 							},
 						},
 						BackendAddressPool: &armnetwork.SubResource{
-							ID: to.Ptr("/subscriptions/" + l.Subscription + "/resourceGroups/" + l.ResourceGroup + "/providers/Microsoft.Network/loadBalancers/" + l.Name + "/backendAddressPools/all"),
+							ID: to.Ptr("/subscriptions/" + l.Subscription +
+								"/resourceGroups/" + l.ResourceGroup +
+								"/providers/Microsoft.Network/loadBalancers/" + l.Name +
+								"/backendAddressPools/all"),
 						},
 						Protocol: to.Ptr(armnetwork.LoadBalancerOutboundRuleProtocolAll),
 					},
--- a/cli/internal/azure/scaleset.go
+++ b/cli/internal/azure/scaleset.go
@ -13,6 +13,7 @@ import (
 type ScaleSet struct {
 	Name                           string
 	NamePrefix                     string
+	UID                            string
 	Subscription                   string
 	ResourceGroup                  string
 	Location                       string
@ -101,10 +102,16 @@ func (s ScaleSet) Azure() armcomputev2.VirtualMachineScaleSet {
 											},
 											LoadBalancerBackendAddressPools: []*armcomputev2.SubResource{
 												{
-													ID: to.Ptr("/subscriptions/" + s.Subscription + "/resourcegroups/" + s.ResourceGroup + "/providers/Microsoft.Network/loadBalancers/" + s.LoadBalancerName + "/backendAddressPools/" + s.LoadBalancerBackendAddressPool),
+													ID: to.Ptr("/subscriptions/" + s.Subscription +
+														"/resourcegroups/" + s.ResourceGroup +
+														"/providers/Microsoft.Network/loadBalancers/" + s.LoadBalancerName +
+														"/backendAddressPools/" + s.LoadBalancerBackendAddressPool),
 												},
 												{
-													ID: to.Ptr("/subscriptions/" + s.Subscription + "/resourcegroups/" + s.ResourceGroup + "/providers/Microsoft.Network/loadBalancers/" + s.LoadBalancerName + "/backendAddressPools/all"),
+													ID: to.Ptr("/subscriptions/" + s.Subscription +
+														"/resourcegroups/" + s.ResourceGroup +
+														"/providers/Microsoft.Network/loadBalancers/" + s.LoadBalancerName +
+														"/backendAddressPools/all"),
 												},
 											},
 										},
@ -134,6 +141,7 @@ func (s ScaleSet) Azure() armcomputev2.VirtualMachineScaleSet {
 				s.UserAssignedIdentity: {},
 			},
 		},
+		Tags: map[string]*string{"uid": to.Ptr(s.UID)},
 	}
 }