add tags at the time volume is created

spectrocloud · Nov 7, 2024 · be7d9ad · be7d9ad
1 parent 1f66f27
commit be7d9ad
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pkg/cloud/services/ec2/instances.go b/pkg/cloud/services/ec2/instances.go
@@ -538,6 +538,7 @@ func (s *Service) runInstance(role string, i *infrav1.Instance) (*infrav1.Instan
 
 	if len(i.Tags) > 0 {
 		spec := &ec2.TagSpecification{ResourceType: aws.String(ec2.ResourceTypeInstance)}
+		volumeSpec := &ec2.TagSpecification{ResourceType: aws.String(ec2.ResourceTypeVolume)}
 		// We need to sort keys for tests to work
 		keys := make([]string, 0, len(i.Tags))
 		for k := range i.Tags {
@@ -549,9 +550,14 @@ func (s *Service) runInstance(role string, i *infrav1.Instance) (*infrav1.Instan
 				Key:   aws.String(key),
 				Value: aws.String(i.Tags[key]),
 			})
+			volumeSpec.Tags = append(volumeSpec.Tags, &ec2.Tag{
+				Key:   aws.String(key),
+				Value: aws.String(i.Tags[key]),
+			})
 		}
 
 		input.TagSpecifications = append(input.TagSpecifications, spec)
+		input.TagSpecifications = append(input.TagSpecifications, volumeSpec)
 	}
 
 	input.InstanceMarketOptions = getInstanceMarketOptionsRequest(i.SpotMarketOptions)