Пример #1
0
// 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
}
Пример #2
0
// 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
}