mirror of
https://github.com/hashicorp/packer.git
synced 2026-02-25 02:42:42 -05:00
e99cd56b6c
This commit moves the Amazon builders of Packer away from the Hashicorp fork of the goamz library to the official AWS SDK for Go, in order that third party plugins may depend on the more complete official library more easily.
323 lines
10 KiB
Go
323 lines
10 KiB
Go
package common
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/awslabs/aws-sdk-go/aws"
|
|
"github.com/awslabs/aws-sdk-go/service/ec2"
|
|
|
|
"github.com/mitchellh/multistep"
|
|
"github.com/mitchellh/packer/packer"
|
|
)
|
|
|
|
type StepRunSourceInstance struct {
|
|
AssociatePublicIpAddress bool
|
|
AvailabilityZone string
|
|
BlockDevices BlockDevices
|
|
Debug bool
|
|
ExpectedRootDevice string
|
|
InstanceType string
|
|
IamInstanceProfile string
|
|
SourceAMI string
|
|
SpotPrice string
|
|
SpotPriceProduct string
|
|
SubnetId string
|
|
Tags map[string]string
|
|
UserData string
|
|
UserDataFile string
|
|
|
|
instance *ec2.Instance
|
|
spotRequest *ec2.SpotInstanceRequest
|
|
}
|
|
|
|
func (s *StepRunSourceInstance) Run(state multistep.StateBag) multistep.StepAction {
|
|
ec2conn := state.Get("ec2").(*ec2.EC2)
|
|
keyName := state.Get("keyPair").(string)
|
|
tempSecurityGroupIds := state.Get("securityGroupIds").([]string)
|
|
ui := state.Get("ui").(packer.Ui)
|
|
|
|
securityGroupIds := make([]*string, len(tempSecurityGroupIds))
|
|
for i, sg := range tempSecurityGroupIds {
|
|
securityGroupIds[i] = &sg
|
|
}
|
|
|
|
userData := s.UserData
|
|
if s.UserDataFile != "" {
|
|
contents, err := ioutil.ReadFile(s.UserDataFile)
|
|
if err != nil {
|
|
state.Put("error", fmt.Errorf("Problem reading user data file: %s", err))
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
userData = string(contents)
|
|
}
|
|
|
|
ui.Say("Launching a source AWS instance...")
|
|
imageResp, err := ec2conn.DescribeImages(&ec2.DescribeImagesInput{
|
|
ImageIDs: []*string{&s.SourceAMI},
|
|
})
|
|
if err != nil {
|
|
state.Put("error", fmt.Errorf("There was a problem with the source AMI: %s", err))
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
if len(imageResp.Images) != 1 {
|
|
state.Put("error", fmt.Errorf("The source AMI '%s' could not be found.", s.SourceAMI))
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
if s.ExpectedRootDevice != "" && *imageResp.Images[0].RootDeviceType != s.ExpectedRootDevice {
|
|
state.Put("error", fmt.Errorf(
|
|
"The provided source AMI has an invalid root device type.\n"+
|
|
"Expected '%s', got '%s'.",
|
|
s.ExpectedRootDevice, *imageResp.Images[0].RootDeviceType))
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
spotPrice := s.SpotPrice
|
|
if spotPrice == "auto" {
|
|
ui.Message(fmt.Sprintf(
|
|
"Finding spot price for %s %s...",
|
|
s.SpotPriceProduct, s.InstanceType))
|
|
|
|
// Detect the spot price
|
|
startTime := time.Now().Add(-1 * time.Hour)
|
|
resp, err := ec2conn.DescribeSpotPriceHistory(&ec2.DescribeSpotPriceHistoryInput{
|
|
InstanceTypes: []*string{&s.InstanceType},
|
|
ProductDescriptions: []*string{&s.SpotPriceProduct},
|
|
AvailabilityZone: &s.AvailabilityZone,
|
|
StartTime: &startTime,
|
|
})
|
|
if err != nil {
|
|
err := fmt.Errorf("Error finding spot price: %s", err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
var price float64
|
|
for _, history := range resp.SpotPriceHistory {
|
|
log.Printf("[INFO] Candidate spot price: %s", *history.SpotPrice)
|
|
current, err := strconv.ParseFloat(*history.SpotPrice, 64)
|
|
if err != nil {
|
|
log.Printf("[ERR] Error parsing spot price: %s", err)
|
|
continue
|
|
}
|
|
if price == 0 || current < price {
|
|
price = current
|
|
}
|
|
}
|
|
if price == 0 {
|
|
err := fmt.Errorf("No candidate spot prices found!")
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
spotPrice = strconv.FormatFloat(price, 'f', -1, 64)
|
|
}
|
|
|
|
var instanceId string
|
|
|
|
if spotPrice == "" {
|
|
runOpts := &ec2.RunInstancesInput{
|
|
KeyName: &keyName,
|
|
ImageID: &s.SourceAMI,
|
|
InstanceType: &s.InstanceType,
|
|
UserData: &userData,
|
|
MaxCount: aws.Long(1),
|
|
MinCount: aws.Long(1),
|
|
IAMInstanceProfile: &ec2.IAMInstanceProfileSpecification{Name: &s.IamInstanceProfile},
|
|
BlockDeviceMappings: s.BlockDevices.BuildLaunchDevices(),
|
|
Placement: &ec2.Placement{AvailabilityZone: &s.AvailabilityZone},
|
|
}
|
|
|
|
if s.SubnetId != "" && s.AssociatePublicIpAddress {
|
|
runOpts.NetworkInterfaces = []*ec2.InstanceNetworkInterfaceSpecification{
|
|
&ec2.InstanceNetworkInterfaceSpecification{
|
|
DeviceIndex: aws.Long(0),
|
|
AssociatePublicIPAddress: &s.AssociatePublicIpAddress,
|
|
SubnetID: &s.SubnetId,
|
|
Groups: securityGroupIds,
|
|
DeleteOnTermination: aws.Boolean(true),
|
|
},
|
|
}
|
|
} else {
|
|
runOpts.SubnetID = &s.SubnetId
|
|
runOpts.SecurityGroupIDs = securityGroupIds
|
|
}
|
|
|
|
runResp, err := ec2conn.RunInstances(runOpts)
|
|
if err != nil {
|
|
err := fmt.Errorf("Error launching source instance: %s", err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
instanceId = *runResp.Instances[0].InstanceID
|
|
} else {
|
|
ui.Message(fmt.Sprintf(
|
|
"Requesting spot instance '%s' for: %s",
|
|
s.InstanceType, spotPrice))
|
|
|
|
runSpotResp, err := ec2conn.RequestSpotInstances(&ec2.RequestSpotInstancesInput{
|
|
SpotPrice: &spotPrice,
|
|
LaunchSpecification: &ec2.RequestSpotLaunchSpecification{
|
|
KeyName: &keyName,
|
|
ImageID: &s.SourceAMI,
|
|
InstanceType: &s.InstanceType,
|
|
UserData: &userData,
|
|
SecurityGroupIDs: securityGroupIds,
|
|
IAMInstanceProfile: &ec2.IAMInstanceProfileSpecification{Name: &s.IamInstanceProfile},
|
|
SubnetID: &s.SubnetId,
|
|
NetworkInterfaces: []*ec2.InstanceNetworkInterfaceSpecification{
|
|
&ec2.InstanceNetworkInterfaceSpecification{AssociatePublicIPAddress: &s.AssociatePublicIpAddress},
|
|
},
|
|
Placement: &ec2.SpotPlacement{AvailabilityZone: &s.AvailabilityZone},
|
|
BlockDeviceMappings: s.BlockDevices.BuildLaunchDevices(),
|
|
},
|
|
})
|
|
if err != nil {
|
|
err := fmt.Errorf("Error launching source spot instance: %s", err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
s.spotRequest = runSpotResp.SpotInstanceRequests[0]
|
|
|
|
spotRequestId := s.spotRequest.SpotInstanceRequestID
|
|
ui.Message(fmt.Sprintf("Waiting for spot request (%s) to become active...", *spotRequestId))
|
|
stateChange := StateChangeConf{
|
|
Pending: []string{"open"},
|
|
Target: "active",
|
|
Refresh: SpotRequestStateRefreshFunc(ec2conn, *spotRequestId),
|
|
StepState: state,
|
|
}
|
|
_, err = WaitForState(&stateChange)
|
|
if err != nil {
|
|
err := fmt.Errorf("Error waiting for spot request (%s) to become ready: %s", *spotRequestId, err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
spotResp, err := ec2conn.DescribeSpotInstanceRequests(&ec2.DescribeSpotInstanceRequestsInput{
|
|
SpotInstanceRequestIDs: []*string{spotRequestId},
|
|
})
|
|
if err != nil {
|
|
err := fmt.Errorf("Error finding spot request (%s): %s", *spotRequestId, err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
instanceId = *spotResp.SpotInstanceRequests[0].InstanceID
|
|
}
|
|
|
|
instanceResp, err := ec2conn.DescribeInstances(&ec2.DescribeInstancesInput{InstanceIDs: []*string{&instanceId}})
|
|
if err != nil {
|
|
err := fmt.Errorf("Error finding source instance (%s): %s", instanceId, err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
s.instance = instanceResp.Reservations[0].Instances[0]
|
|
ui.Message(fmt.Sprintf("Instance ID: %s", *s.instance.InstanceID))
|
|
|
|
ui.Say(fmt.Sprintf("Waiting for instance (%s) to become ready...", *s.instance.InstanceID))
|
|
stateChange := StateChangeConf{
|
|
Pending: []string{"pending"},
|
|
Target: "running",
|
|
Refresh: InstanceStateRefreshFunc(ec2conn, s.instance),
|
|
StepState: state,
|
|
}
|
|
latestInstance, err := WaitForState(&stateChange)
|
|
if err != nil {
|
|
err := fmt.Errorf("Error waiting for instance (%s) to become ready: %s", *s.instance.InstanceID, err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
s.instance = latestInstance.(*ec2.Instance)
|
|
|
|
ec2Tags := make([]*ec2.Tag, 1, len(s.Tags)+1)
|
|
ec2Tags[0] = &ec2.Tag{Key: aws.String("Name"), Value: aws.String("Packer Builder")}
|
|
for k, v := range s.Tags {
|
|
ec2Tags = append(ec2Tags, &ec2.Tag{Key: &k, Value: &v})
|
|
}
|
|
|
|
_, err = ec2conn.CreateTags(&ec2.CreateTagsInput{
|
|
Tags: ec2Tags,
|
|
Resources: []*string{s.instance.InstanceID},
|
|
})
|
|
if err != nil {
|
|
ui.Message(
|
|
fmt.Sprintf("Failed to tag a Name on the builder instance: %s", err))
|
|
}
|
|
|
|
if s.Debug {
|
|
if *s.instance.PublicDNSName != "" {
|
|
ui.Message(fmt.Sprintf("Public DNS: %s", *s.instance.PublicDNSName))
|
|
}
|
|
|
|
if *s.instance.PublicIPAddress != "" {
|
|
ui.Message(fmt.Sprintf("Public IP: %s", *s.instance.PublicIPAddress))
|
|
}
|
|
|
|
if *s.instance.PrivateIPAddress != "" {
|
|
ui.Message(fmt.Sprintf("Private IP: %s", *s.instance.PrivateIPAddress))
|
|
}
|
|
}
|
|
|
|
state.Put("instance", s.instance)
|
|
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
func (s *StepRunSourceInstance) Cleanup(state multistep.StateBag) {
|
|
|
|
ec2conn := state.Get("ec2").(*ec2.EC2)
|
|
ui := state.Get("ui").(packer.Ui)
|
|
|
|
// Cancel the spot request if it exists
|
|
if s.spotRequest != nil {
|
|
ui.Say("Cancelling the spot request...")
|
|
input := &ec2.CancelSpotInstanceRequestsInput{
|
|
SpotInstanceRequestIDs: []*string{s.spotRequest.InstanceID},
|
|
}
|
|
if _, err := ec2conn.CancelSpotInstanceRequests(input); err != nil {
|
|
ui.Error(fmt.Sprintf("Error cancelling the spot request, may still be around: %s", err))
|
|
return
|
|
}
|
|
stateChange := StateChangeConf{
|
|
Pending: []string{"active", "open"},
|
|
Refresh: SpotRequestStateRefreshFunc(ec2conn, *s.spotRequest.SpotInstanceRequestID),
|
|
Target: "cancelled",
|
|
}
|
|
|
|
WaitForState(&stateChange)
|
|
|
|
}
|
|
|
|
// Terminate the source instance if it exists
|
|
if s.instance != nil {
|
|
|
|
ui.Say("Terminating the source AWS instance...")
|
|
if _, err := ec2conn.TerminateInstances(&ec2.TerminateInstancesInput{InstanceIDs: []*string{s.instance.InstanceID}}); err != nil {
|
|
ui.Error(fmt.Sprintf("Error terminating instance, may still be around: %s", err))
|
|
return
|
|
}
|
|
stateChange := StateChangeConf{
|
|
Pending: []string{"pending", "running", "shutting-down", "stopped", "stopping"},
|
|
Refresh: InstanceStateRefreshFunc(ec2conn, s.instance),
|
|
Target: "terminated",
|
|
}
|
|
|
|
WaitForState(&stateChange)
|
|
}
|
|
}
|