forked from hashicorp/terraform-provider-aws
/
data_source_aws_ec2_transit_gateway_route_table.go
86 lines (67 loc) · 2.43 KB
/
data_source_aws_ec2_transit_gateway_route_table.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
package aws
import (
"errors"
"fmt"
"log"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/service/ec2"
"github.com/hashicorp/terraform/helper/schema"
)
func dataSourceAwsEc2TransitGatewayRouteTable() *schema.Resource {
return &schema.Resource{
Read: dataSourceAwsEc2TransitGatewayRouteTableRead,
Schema: map[string]*schema.Schema{
"default_association_route_table": {
Type: schema.TypeBool,
Computed: true,
},
"default_propagation_route_table": {
Type: schema.TypeBool,
Computed: true,
},
"filter": dataSourceFiltersSchema(),
"id": {
Type: schema.TypeString,
Optional: true,
},
"transit_gateway_id": {
Type: schema.TypeString,
Computed: true,
},
"tags": tagsSchemaComputed(),
},
}
}
func dataSourceAwsEc2TransitGatewayRouteTableRead(d *schema.ResourceData, meta interface{}) error {
conn := meta.(*AWSClient).ec2conn
input := &ec2.DescribeTransitGatewayRouteTablesInput{}
if v, ok := d.GetOk("filter"); ok {
input.Filters = buildAwsDataSourceFilters(v.(*schema.Set))
}
if v, ok := d.GetOk("id"); ok {
input.TransitGatewayRouteTableIds = []*string{aws.String(v.(string))}
}
log.Printf("[DEBUG] Reading EC2 Transit Gateways: %s", input)
output, err := conn.DescribeTransitGatewayRouteTables(input)
if err != nil {
return fmt.Errorf("error reading EC2 Transit Gateway Route Table: %s", err)
}
if output == nil || len(output.TransitGatewayRouteTables) == 0 {
return errors.New("error reading EC2 Transit Gateway Route Table: no results found")
}
if len(output.TransitGatewayRouteTables) > 1 {
return errors.New("error reading EC2 Transit Gateway Route Table: multiple results found, try adjusting search criteria")
}
transitGatewayRouteTable := output.TransitGatewayRouteTables[0]
if transitGatewayRouteTable == nil {
return errors.New("error reading EC2 Transit Gateway Route Table: empty result")
}
d.Set("default_association_route_table", aws.BoolValue(transitGatewayRouteTable.DefaultAssociationRouteTable))
d.Set("default_propagation_route_table", aws.BoolValue(transitGatewayRouteTable.DefaultPropagationRouteTable))
if err := d.Set("tags", tagsToMap(transitGatewayRouteTable.Tags)); err != nil {
return fmt.Errorf("error setting tags: %s", err)
}
d.Set("transit_gateway_id", aws.StringValue(transitGatewayRouteTable.TransitGatewayId))
d.SetId(aws.StringValue(transitGatewayRouteTable.TransitGatewayRouteTableId))
return nil
}