Skip to content

Commit

Permalink
Merge pull request #26864 from meetreks/f-aws_ec2_traffic_mirror_target
Browse files Browse the repository at this point in the history 
f-aws_ec2_traffic_mirror_target
  • Loading branch information
@ewbankkit
ewbankkit authored Sep 29, 2022
2 parents 8545f01 + a1f938e commit 9c21d5d
Show file tree
Hide file tree
Showing 6 changed files with 224 additions and 168 deletions.
3 changes: 3 additions & 0 deletions .changelog/26767.txt
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
```release-note:enhancement
resource/aws_ec2_traffic_mirror_target: Add `gateway_load_balancer_endpoint_id` argument
```
1 change: 1 addition & 0 deletions internal/service/ec2/errors.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -76,6 +76,7 @@ const (
errCodeInvalidSubnetCIDRReservationIDNotFound = "InvalidSubnetCidrReservationID.NotFound"
errCodeInvalidSubnetIDNotFound = "InvalidSubnetID.NotFound"
errCodeInvalidSubnetIdNotFound = "InvalidSubnetId.NotFound"
errCodeInvalidTrafficMirrorTargetIdNotFound = "InvalidTrafficMirrorTargetId.NotFound"
errCodeInvalidTransitGatewayAttachmentIDNotFound = "InvalidTransitGatewayAttachmentID.NotFound"
errCodeInvalidTransitGatewayConnectPeerIDNotFound = "InvalidTransitGatewayConnectPeerID.NotFound"
errCodeInvalidTransitGatewayPolicyTableIdNotFound = "InvalidTransitGatewayPolicyTableId.NotFound"
Expand Down
70 changes: 70 additions & 0 deletions internal/service/ec2/find.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3309,6 +3309,76 @@ func FindVPNConnectionRouteByVPNConnectionIDAndCIDR(conn *ec2.EC2, vpnConnection
}
}

func FindTrafficMirrorTarget(conn *ec2.EC2, input *ec2.DescribeTrafficMirrorTargetsInput) (*ec2.TrafficMirrorTarget, error) {
output, err := FindTrafficMirrorTargets(conn, input)

if err != nil {
return nil, err
}

if len(output) == 0 || output[0] == nil {
return nil, tfresource.NewEmptyResultError(input)
}

if count := len(output); count > 1 {
return nil, tfresource.NewTooManyResultsError(count, input)
}

return output[0], nil
}

func FindTrafficMirrorTargets(conn *ec2.EC2, input *ec2.DescribeTrafficMirrorTargetsInput) ([]*ec2.TrafficMirrorTarget, error) {
var output []*ec2.TrafficMirrorTarget

err := conn.DescribeTrafficMirrorTargetsPages(input, func(page *ec2.DescribeTrafficMirrorTargetsOutput, lastPage bool) bool {
if page == nil {
return !lastPage
}

for _, v := range page.TrafficMirrorTargets {
if v != nil {
output = append(output, v)
}
}

return !lastPage
})

if tfawserr.ErrCodeEquals(err, errCodeInvalidTrafficMirrorTargetIdNotFound) {
return nil, &resource.NotFoundError{
LastError: err,
LastRequest: input,
}
}

if err != nil {
return nil, err
}

return output, nil
}

func FindTrafficMirrorTargetByID(conn *ec2.EC2, id string) (*ec2.TrafficMirrorTarget, error) {
input := &ec2.DescribeTrafficMirrorTargetsInput{
TrafficMirrorTargetIds: aws.StringSlice([]string{id}),
}

output, err := FindTrafficMirrorTarget(conn, input)

if err != nil {
return nil, err
}

// Eventual consistency check.
if aws.StringValue(output.TrafficMirrorTargetId) != id {
return nil, &resource.NotFoundError{
LastRequest: input,
}
}

return output, nil
}

func FindTransitGateway(conn *ec2.EC2, input *ec2.DescribeTransitGatewaysInput) (*ec2.TransitGateway, error) {
output, err := FindTransitGateways(conn, input)

Expand Down
106 changes: 55 additions & 51 deletions internal/service/ec2/vpc_traffic_mirror_target.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -7,10 +7,10 @@ import (
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/arn"
"github.com/aws/aws-sdk-go/service/ec2"
"github.com/hashicorp/aws-sdk-go-base/v2/awsv1shim/v2/tfawserr"
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
"github.com/hashicorp/terraform-provider-aws/internal/conns"
tftags "github.com/hashicorp/terraform-provider-aws/internal/tags"
"github.com/hashicorp/terraform-provider-aws/internal/tfresource"
"github.com/hashicorp/terraform-provider-aws/internal/verify"
)

Expand All @@ -36,11 +36,22 @@ func ResourceTrafficMirrorTarget() *schema.Resource {
Optional: true,
ForceNew: true,
},
"gateway_load_balancer_endpoint_id": {
Type: schema.TypeString,
Optional: true,
ForceNew: true,
ExactlyOneOf: []string{
"gateway_load_balancer_endpoint_id",
"network_interface_id",
"network_load_balancer_arn",
},
},
"network_interface_id": {
Type: schema.TypeString,
Optional: true,
ForceNew: true,
ExactlyOneOf: []string{
"gateway_load_balancer_endpoint_id",
"network_interface_id",
"network_load_balancer_arn",
},
Expand All @@ -50,6 +61,7 @@ func ResourceTrafficMirrorTarget() *schema.Resource {
Optional: true,
ForceNew: true,
ExactlyOneOf: []string{
"gateway_load_balancer_endpoint_id",
"network_interface_id",
"network_load_balancer_arn",
},
Expand All @@ -76,6 +88,10 @@ func resourceTrafficMirrorTargetCreate(d *schema.ResourceData, meta interface{})
input.Description = aws.String(v.(string))
}

if v, ok := d.GetOk("gateway_load_balancer_endpoint_id"); ok {
input.GatewayLoadBalancerEndpointId = aws.String(v.(string))
}

if v, ok := d.GetOk("network_interface_id"); ok {
input.NetworkInterfaceId = aws.String(v.(string))
}
Expand All @@ -88,26 +104,13 @@ func resourceTrafficMirrorTargetCreate(d *schema.ResourceData, meta interface{})
input.TagSpecifications = tagSpecificationsFromKeyValueTags(tags, ec2.ResourceTypeTrafficMirrorTarget)
}

out, err := conn.CreateTrafficMirrorTarget(input)
output, err := conn.CreateTrafficMirrorTarget(input)

if err != nil {
return fmt.Errorf("Error creating traffic mirror target %v", err)
return fmt.Errorf("creating EC2 Traffic Mirror Target: %w", err)
}

d.SetId(aws.StringValue(out.TrafficMirrorTarget.TrafficMirrorTargetId))

return resourceTrafficMirrorTargetRead(d, meta)
}

func resourceTrafficMirrorTargetUpdate(d *schema.ResourceData, meta interface{}) error {
conn := meta.(*conns.AWSClient).EC2Conn

if d.HasChange("tags_all") {
o, n := d.GetChange("tags_all")

if err := UpdateTags(conn, d.Id(), o, n); err != nil {
return fmt.Errorf("error updating EC2 Traffic Mirror Target (%s) tags: %s", d.Id(), err)
}
}
d.SetId(aws.StringValue(output.TrafficMirrorTarget.TrafficMirrorTargetId))

return resourceTrafficMirrorTargetRead(d, meta)
}
Expand All @@ -117,70 +120,71 @@ func resourceTrafficMirrorTargetRead(d *schema.ResourceData, meta interface{}) e
defaultTagsConfig := meta.(*conns.AWSClient).DefaultTagsConfig
ignoreTagsConfig := meta.(*conns.AWSClient).IgnoreTagsConfig

targetId := d.Id()
input := &ec2.DescribeTrafficMirrorTargetsInput{
TrafficMirrorTargetIds: []*string{&targetId},
}
target, err := FindTrafficMirrorTargetByID(conn, d.Id())

out, err := conn.DescribeTrafficMirrorTargets(input)
if tfawserr.ErrCodeEquals(err, "InvalidTrafficMirrorTargetId.NotFound") {
log.Printf("[WARN] EC2 Traffic Mirror Target (%s) not found, removing from state", d.Id())
if !d.IsNewResource() && tfresource.NotFound(err) {
log.Printf("[WARN] EC2 Traffic Mirror Target %s not found, removing from state", d.Id())
d.SetId("")
return nil
}

if err != nil {
return fmt.Errorf("error describing EC2 Traffic Mirror Target (%s): %w", targetId, err)
}

if out == nil || len(out.TrafficMirrorTargets) == 0 {
log.Printf("[WARN] EC2 Traffic Mirror Target (%s) not found, removing from state", d.Id())
d.SetId("")
return nil
return fmt.Errorf("reading EC2 Traffic Mirror Target (%s): %w", d.Id(), err)
}

target := out.TrafficMirrorTargets[0]
ownerID := aws.StringValue(target.OwnerId)
arn := arn.ARN{
Partition: meta.(*conns.AWSClient).Partition,
Service: ec2.ServiceName,
Region: meta.(*conns.AWSClient).Region,
AccountID: ownerID,
Resource: fmt.Sprintf("traffic-mirror-target/%s", d.Id()),
}.String()
d.Set("arn", arn)
d.Set("description", target.Description)
d.Set("gateway_load_balancer_endpoint_id", target.GatewayLoadBalancerEndpointId)
d.Set("network_interface_id", target.NetworkInterfaceId)
d.Set("network_load_balancer_arn", target.NetworkLoadBalancerArn)
d.Set("owner_id", ownerID)

tags := KeyValueTags(target.Tags).IgnoreAWS().IgnoreConfig(ignoreTagsConfig)

//lintignore:AWSR002
if err := d.Set("tags", tags.RemoveDefaultConfig(defaultTagsConfig).Map()); err != nil {
return fmt.Errorf("error setting tags: %w", err)
return fmt.Errorf("setting tags: %w", err)
}

if err := d.Set("tags_all", tags.Map()); err != nil {
return fmt.Errorf("error setting tags_all: %w", err)
return fmt.Errorf("setting tags_all: %w", err)
}

d.Set("owner_id", target.OwnerId)
return nil
}

arn := arn.ARN{
Partition: meta.(*conns.AWSClient).Partition,
Service: ec2.ServiceName,
Region: meta.(*conns.AWSClient).Region,
AccountID: aws.StringValue(target.OwnerId),
Resource: fmt.Sprintf("traffic-mirror-target/%s", d.Id()),
}.String()
func resourceTrafficMirrorTargetUpdate(d *schema.ResourceData, meta interface{}) error {
conn := meta.(*conns.AWSClient).EC2Conn

d.Set("arn", arn)
if d.HasChange("tags_all") {
o, n := d.GetChange("tags_all")

return nil
if err := UpdateTags(conn, d.Id(), o, n); err != nil {
return fmt.Errorf("updating EC2 Traffic Mirror Target (%s) tags: %w", d.Id(), err)
}
}

return resourceTrafficMirrorTargetRead(d, meta)
}

func resourceTrafficMirrorTargetDelete(d *schema.ResourceData, meta interface{}) error {
conn := meta.(*conns.AWSClient).EC2Conn

targetId := d.Id()
input := &ec2.DeleteTrafficMirrorTargetInput{
TrafficMirrorTargetId: &targetId,
}
log.Printf("[DEBUG] Deleting EC2 Traffic Mirror Target: %s", d.Id())
_, err := conn.DeleteTrafficMirrorTarget(&ec2.DeleteTrafficMirrorTargetInput{
TrafficMirrorTargetId: aws.String(d.Id()),
})

_, err := conn.DeleteTrafficMirrorTarget(input)
if nil != err {
return fmt.Errorf("error deleting EC2 Traffic Mirror Target (%s): %w", targetId, err)
return fmt.Errorf("deleting EC2 Traffic Mirror Target (%s): %w", d.Id(), err)
}

return nil
Expand Down
Loading

0 comments on commit 9c21d5d

Please sign in to comment.