Merge #9033: Fix crash in aws_ami data source with name_regex

This commit is contained in:
Martin Atkins 2016-09-24 08:27:57 -07:00
commit 7d2b51e6c5
2 changed files with 25 additions and 11 deletions

View File

@ -36,6 +36,7 @@ BUG FIXES:
* provider/aws: Ensure that origin_access_identity is a required value within the CloudFront distribution s3_config block [GH-8777] * provider/aws: Ensure that origin_access_identity is a required value within the CloudFront distribution s3_config block [GH-8777]
* provider/aws: Corrected Seoul S3 Website Endpoint format [GH-9032] * provider/aws: Corrected Seoul S3 Website Endpoint format [GH-9032]
* provider/aws: Fix failed remove S3 lifecycle_rule [GH-9031] * provider/aws: Fix failed remove S3 lifecycle_rule [GH-9031]
* provider/aws: Fix crashing bug in `aws_ami` data source when using `name_regex` [GH-9033]
## 0.7.4 (September 19, 2016) ## 0.7.4 (September 19, 2016)

View File

@ -236,10 +236,19 @@ func dataSourceAwsAmiRead(d *schema.ResourceData, meta interface{}) error {
} }
var filteredImages []*ec2.Image var filteredImages []*ec2.Image
if nameRegexOk == true { if nameRegexOk {
r := regexp.MustCompile(nameRegex.(string)) r := regexp.MustCompile(nameRegex.(string))
for _, image := range resp.Images { for _, image := range resp.Images {
if r.MatchString(*image.Name) == true { // Check for a very rare case where the response would include no
// image name. No name means nothing to attempt a match against,
// therefore we are skipping such image.
if image.Name == nil || *image.Name == "" {
log.Printf("[WARN] Unable to find AMI name to match against "+
"for image ID %q owned by %q, nothing to do.",
*image.ImageId, *image.OwnerId)
continue
}
if r.MatchString(*image.Name) {
filteredImages = append(filteredImages, image) filteredImages = append(filteredImages, image)
} }
} }
@ -249,19 +258,23 @@ func dataSourceAwsAmiRead(d *schema.ResourceData, meta interface{}) error {
var image *ec2.Image var image *ec2.Image
if len(filteredImages) < 1 { if len(filteredImages) < 1 {
return fmt.Errorf("Your query returned no results. Please change your filters and try again.") return fmt.Errorf("Your query returned no results. Please change your search criteria and try again.")
} else if len(filteredImages) > 1 { }
if (d.Get("most_recent").(bool)) == true {
log.Printf("[DEBUG] aws_ami - multiple results found and most_recent is set") if len(filteredImages) > 1 {
recent := d.Get("most_recent").(bool)
log.Printf("[DEBUG] aws_ami - multiple results found and `most_recent` is set to: %t", recent)
if recent {
image = mostRecentAmi(filteredImages) image = mostRecentAmi(filteredImages)
} else { } else {
log.Printf("[DEBUG] aws_ami - multiple results found and most_recent not set") return fmt.Errorf("Your query returned more than one result. Please try a more " +
return fmt.Errorf("Your query returned more than one result. Please try a more specific search, or set most_recent to true.") "specific search criteria, or set `most_recent` attribute to true.")
} }
} else { }
log.Printf("[DEBUG] aws_ami - Single AMI found: %s", *filteredImages[0].ImageId)
image = filteredImages[0] image = filteredImages[0]
} log.Printf("[DEBUG] aws_ami - Single AMI found: %s", *image.ImageId)
return amiDescriptionAttributes(d, image) return amiDescriptionAttributes(d, image)
} }