packer-cn/builder/profitbricks/step_take_snapshot.go

77 lines
1.8 KiB
Go
Raw Normal View History

2016-06-28 22:35:41 -04:00
package profitbricks
import (
2016-11-01 17:08:04 -04:00
"encoding/json"
2016-06-28 22:35:41 -04:00
"errors"
"fmt"
"github.com/mitchellh/multistep"
"github.com/mitchellh/packer/packer"
"github.com/profitbricks/profitbricks-sdk-go"
"time"
)
type stepTakeSnapshot struct{}
func (s *stepTakeSnapshot) Run(state multistep.StateBag) multistep.StepAction {
ui := state.Get("ui").(packer.Ui)
c := state.Get("config").(*Config)
ui.Say("Creating ProfitBricks snapshot...")
profitbricks.SetAuth(c.PBUsername, c.PBPassword)
dcId := state.Get("datacenter_id").(string)
volumeId := state.Get("volume_id").(string)
2016-07-07 04:28:46 -04:00
snapshot := profitbricks.CreateSnapshot(dcId, volumeId, c.SnapshotName)
2016-06-28 22:35:41 -04:00
2016-07-07 04:28:46 -04:00
state.Put("snapshotname", c.SnapshotName)
2016-06-28 22:35:41 -04:00
if snapshot.StatusCode > 299 {
2016-10-12 18:41:04 -04:00
var restError RestError
json.Unmarshal([]byte(snapshot.Response), &restError)
2016-11-01 17:08:04 -04:00
if len(restError.Messages) > 0 {
2016-10-12 18:41:04 -04:00
ui.Error(restError.Messages[0].Message)
} else {
ui.Error(snapshot.Response)
}
2016-06-28 22:35:41 -04:00
return multistep.ActionHalt
}
s.waitTillProvisioned(snapshot.Headers.Get("Location"), *c)
2016-06-28 22:35:41 -04:00
return multistep.ActionContinue
}
func (s *stepTakeSnapshot) Cleanup(state multistep.StateBag) {
}
func (d *stepTakeSnapshot) checkForErrors(instance profitbricks.Resp) error {
if instance.StatusCode > 299 {
return errors.New(fmt.Sprintf("Error occured %s", string(instance.Body)))
}
return nil
}
func (d *stepTakeSnapshot) waitTillProvisioned(path string, config Config) {
d.setPB(config.PBUsername, config.PBPassword, config.PBUrl)
waitCount := 50
if config.Timeout > 0 {
waitCount = config.Timeout
}
2016-06-28 22:35:41 -04:00
for i := 0; i < waitCount; i++ {
request := profitbricks.GetRequestStatus(path)
if request.Metadata.Status == "DONE" {
2016-06-28 22:35:41 -04:00
break
}
time.Sleep(10 * time.Second)
i++
}
}
func (d *stepTakeSnapshot) setPB(username string, password string, url string) {
profitbricks.SetAuth(username, password)
profitbricks.SetEndpoint(url)
}