mirror of
https://github.com/hashicorp/packer.git
synced 2026-09-26 09:44:01 -04:00
James Nugent
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.
109 lines
2.9 KiB
Go
109 lines
2.9 KiB
Go
package chroot
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
|
|
"github.com/awslabs/aws-sdk-go/service/ec2"
|
|
"github.com/mitchellh/multistep"
|
|
awscommon "github.com/mitchellh/packer/builder/amazon/common"
|
|
"github.com/mitchellh/packer/packer"
|
|
)
|
|
|
|
// StepCreateVolume creates a new volume from the snapshot of the root
|
|
// device of the AMI.
|
|
//
|
|
// Produces:
|
|
// volume_id string - The ID of the created volume
|
|
type StepCreateVolume struct {
|
|
volumeId string
|
|
}
|
|
|
|
func (s *StepCreateVolume) Run(state multistep.StateBag) multistep.StepAction {
|
|
ec2conn := state.Get("ec2").(*ec2.EC2)
|
|
image := state.Get("source_image").(*ec2.Image)
|
|
instance := state.Get("instance").(*ec2.Instance)
|
|
ui := state.Get("ui").(packer.Ui)
|
|
|
|
// Determine the root device snapshot
|
|
log.Printf("Searching for root device of the image (%s)", *image.RootDeviceName)
|
|
var rootDevice *ec2.BlockDeviceMapping
|
|
for _, device := range image.BlockDeviceMappings {
|
|
if device.DeviceName == image.RootDeviceName {
|
|
rootDevice = device
|
|
break
|
|
}
|
|
}
|
|
|
|
if rootDevice == nil {
|
|
err := fmt.Errorf("Couldn't find root device!")
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
ui.Say("Creating the root volume...")
|
|
createVolume := &ec2.CreateVolumeInput{
|
|
AvailabilityZone: instance.Placement.AvailabilityZone,
|
|
Size: rootDevice.EBS.VolumeSize,
|
|
SnapshotID: rootDevice.EBS.SnapshotID,
|
|
VolumeType: rootDevice.EBS.VolumeType,
|
|
IOPS: rootDevice.EBS.IOPS,
|
|
}
|
|
log.Printf("Create args: %#v", createVolume)
|
|
|
|
createVolumeResp, err := ec2conn.CreateVolume(createVolume)
|
|
if err != nil {
|
|
err := fmt.Errorf("Error creating root volume: %s", err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
// Set the volume ID so we remember to delete it later
|
|
s.volumeId = *createVolumeResp.VolumeID
|
|
log.Printf("Volume ID: %s", s.volumeId)
|
|
|
|
// Wait for the volume to become ready
|
|
stateChange := awscommon.StateChangeConf{
|
|
Pending: []string{"creating"},
|
|
StepState: state,
|
|
Target: "available",
|
|
Refresh: func() (interface{}, string, error) {
|
|
resp, err := ec2conn.DescribeVolumes(&ec2.DescribeVolumesInput{VolumeIDs: []*string{&s.volumeId}})
|
|
if err != nil {
|
|
return nil, "", err
|
|
}
|
|
|
|
v := resp.Volumes[0]
|
|
return v, *v.State, nil
|
|
},
|
|
}
|
|
|
|
_, err = awscommon.WaitForState(&stateChange)
|
|
if err != nil {
|
|
err := fmt.Errorf("Error waiting for volume: %s", err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
state.Put("volume_id", s.volumeId)
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
func (s *StepCreateVolume) Cleanup(state multistep.StateBag) {
|
|
if s.volumeId == "" {
|
|
return
|
|
}
|
|
|
|
ec2conn := state.Get("ec2").(*ec2.EC2)
|
|
ui := state.Get("ui").(packer.Ui)
|
|
|
|
ui.Say("Deleting the created EBS volume...")
|
|
_, err := ec2conn.DeleteVolume(&ec2.DeleteVolumeInput{VolumeID: &s.volumeId})
|
|
if err != nil {
|
|
ui.Error(fmt.Sprintf("Error deleting EBS volume: %s", err))
|
|
}
|
|
}
|