// Get ...
func Get(machine *Machine, auth aws.Auth) error {
ec2Ref := ec2.New(auth, aws.Regions[machine.Instance.Region], aws.SignV4Factory(machine.Instance.Region, "ec2"))
// Verify if cloud-config file exists
if machine.Instance.CloudConfig != "" {
_, err := os.Stat(machine.Instance.CloudConfig)
if err != nil {
return err
}
}
// get list of volumes to format
volumesToFormat := make([]volume.Volume, 0)
for key := range machine.Volumes {
volumeConfig := &machine.Volumes[key]
format := false
if volumeConfig.ID == "" && volumeConfig.SnapshotID == "" {
format = true
}
volumeConfig.AvailableZone = machine.Instance.AvailableZone
_, err := volume.Get(ec2Ref, volumeConfig)
if err != nil {
return err
}
if format == true {
volumesToFormat = append(volumesToFormat, *volumeConfig)
}
}
// Create a machine to format theses volumes
if len(volumesToFormat) > 0 {
err := FormatVolumes(ec2Ref, *machine, volumesToFormat)
if err != nil {
return err
}
}
_, err := instance.Get(ec2Ref, &machine.Instance)
if err != nil {
return err
}
err = AttachVolumes(ec2Ref, machine.Instance.ID, machine.Volumes)
if err != nil {
return err
}
err = instance.Reboot(ec2Ref, machine.Instance)
if err != nil {
return err
}
logger.Printf("The instance Id <%s> with IP Address <%s> is running with %d volume(s)!\n", machine.Instance.ID, machine.Instance.PrivateIPAddress, len(machine.Volumes))
return nil
}
// FormatVolumes ...
func FormatVolumes(ec2Ref *ec2.EC2, machine Machine, volumes []volume.Volume) error {
err := os.Mkdir("cloud-config", 0755)
if os.IsPermission(err) == true {
return err
}
name := machine.Instance.Name + "-format-volumes"
cloudConfigName := fmt.Sprintf("cloud-config/%s.yml", name)
// create specific cloud config to format volumes
var units string
for _, volumeConfig := range volumes {
units += getFormatAndMountUnit(volumeConfig)
}
err = ioutil.WriteFile(cloudConfigName, []byte(getFormatCloudConfig(units)), 0644)
if err != nil {
return err
}
formatInstance := instance.Instance{
Name: name,
CloudConfig: cloudConfigName,
ImageID: DefaultFormatInstanceImageID,
Type: DefaultFormatInstanceType,
KeyName: machine.Instance.KeyName,
SecurityGroups: machine.Instance.SecurityGroups,
SubnetID: machine.Instance.SubnetID,
AvailableZone: machine.Instance.AvailableZone,
ShutdownBehavior: "terminate",
}
_, err = instance.Get(ec2Ref, &formatInstance)
if err != nil {
return err
}
err = AttachVolumes(ec2Ref, formatInstance.ID, volumes)
if err != nil {
return err
}
err = instance.Reboot(ec2Ref, formatInstance)
if err != nil {
return err
}
logger.Printf("Waiting while %d volumes was formating...\n", len(volumes))
err = instance.WaitUntilState(ec2Ref, &formatInstance, "terminated")
logger.Println("")
if err != nil {
return err
}
return nil
}