137 lines
3.3 KiB
Go
137 lines
3.3 KiB
Go
package aws
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"sort"
|
|
|
|
"github.com/aws/aws-sdk-go/service/ec2"
|
|
"github.com/davecgh/go-spew/spew"
|
|
"github.com/hashicorp/terraform/helper/schema"
|
|
)
|
|
|
|
func dataSourceAwsEbsVolume() *schema.Resource {
|
|
return &schema.Resource{
|
|
Read: dataSourceAwsEbsVolumeRead,
|
|
|
|
Schema: map[string]*schema.Schema{
|
|
"filter": dataSourceFiltersSchema(),
|
|
"most_recent": {
|
|
Type: schema.TypeBool,
|
|
Optional: true,
|
|
Default: false,
|
|
ForceNew: true,
|
|
},
|
|
"availability_zone": {
|
|
Type: schema.TypeString,
|
|
Computed: true,
|
|
},
|
|
"encrypted": {
|
|
Type: schema.TypeBool,
|
|
Computed: true,
|
|
},
|
|
"iops": {
|
|
Type: schema.TypeInt,
|
|
Computed: true,
|
|
},
|
|
"volume_type": {
|
|
Type: schema.TypeString,
|
|
Computed: true,
|
|
},
|
|
"size": {
|
|
Type: schema.TypeInt,
|
|
Computed: true,
|
|
},
|
|
"snapshot_id": {
|
|
Type: schema.TypeString,
|
|
Computed: true,
|
|
},
|
|
"kms_key_id": {
|
|
Type: schema.TypeString,
|
|
Computed: true,
|
|
},
|
|
"volume_id": {
|
|
Type: schema.TypeString,
|
|
Computed: true,
|
|
},
|
|
"tags": dataSourceTagsSchema(),
|
|
},
|
|
}
|
|
}
|
|
|
|
func dataSourceAwsEbsVolumeRead(d *schema.ResourceData, meta interface{}) error {
|
|
conn := meta.(*AWSClient).ec2conn
|
|
|
|
filters, filtersOk := d.GetOk("filter")
|
|
|
|
params := &ec2.DescribeVolumesInput{}
|
|
if filtersOk {
|
|
params.Filters = buildAwsDataSourceFilters(filters.(*schema.Set))
|
|
}
|
|
|
|
resp, err := conn.DescribeVolumes(params)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Printf("Found These Volumes %s", spew.Sdump(resp.Volumes))
|
|
|
|
filteredVolumes := resp.Volumes[:]
|
|
|
|
var volume *ec2.Volume
|
|
if len(filteredVolumes) < 1 {
|
|
return fmt.Errorf("Your query returned no results. Please change your search criteria and try again.")
|
|
}
|
|
|
|
if len(filteredVolumes) > 1 {
|
|
recent := d.Get("most_recent").(bool)
|
|
log.Printf("[DEBUG] aws_ebs_volume - multiple results found and `most_recent` is set to: %t", recent)
|
|
if recent {
|
|
volume = mostRecentVolume(filteredVolumes)
|
|
} else {
|
|
return fmt.Errorf("Your query returned more than one result. Please try a more " +
|
|
"specific search criteria, or set `most_recent` attribute to true.")
|
|
}
|
|
} else {
|
|
// Query returned single result.
|
|
volume = filteredVolumes[0]
|
|
}
|
|
|
|
log.Printf("[DEBUG] aws_ebs_volume - Single Volume found: %s", *volume.VolumeId)
|
|
return volumeDescriptionAttributes(d, volume)
|
|
}
|
|
|
|
type volumeSort []*ec2.Volume
|
|
|
|
func (a volumeSort) Len() int { return len(a) }
|
|
func (a volumeSort) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
|
|
func (a volumeSort) Less(i, j int) bool {
|
|
itime := *a[i].CreateTime
|
|
jtime := *a[j].CreateTime
|
|
return itime.Unix() < jtime.Unix()
|
|
}
|
|
|
|
func mostRecentVolume(volumes []*ec2.Volume) *ec2.Volume {
|
|
sortedVolumes := volumes
|
|
sort.Sort(volumeSort(sortedVolumes))
|
|
return sortedVolumes[len(sortedVolumes)-1]
|
|
}
|
|
|
|
func volumeDescriptionAttributes(d *schema.ResourceData, volume *ec2.Volume) error {
|
|
d.SetId(*volume.VolumeId)
|
|
d.Set("volume_id", volume.VolumeId)
|
|
d.Set("availability_zone", volume.AvailabilityZone)
|
|
d.Set("encrypted", volume.Encrypted)
|
|
d.Set("iops", volume.Iops)
|
|
d.Set("kms_key_id", volume.KmsKeyId)
|
|
d.Set("size", volume.Size)
|
|
d.Set("snapshot_id", volume.SnapshotId)
|
|
d.Set("volume_type", volume.VolumeType)
|
|
|
|
if err := d.Set("tags", dataSourceTags(volume.Tags)); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|