feature: bsusurrogate, add stop bsu backed vm step
This commit is contained in:
parent
3ed7c7ff33
commit
52962f2355
|
@ -204,6 +204,10 @@ func (b *Builder) Run(ui packer.Ui, hook packer.Hook, cache packer.Cache) (packe
|
||||||
&common.StepCleanupTempKeys{
|
&common.StepCleanupTempKeys{
|
||||||
Comm: &b.config.RunConfig.Comm,
|
Comm: &b.config.RunConfig.Comm,
|
||||||
},
|
},
|
||||||
|
&osccommon.StepStopBSUBackedVm{
|
||||||
|
Skip: false,
|
||||||
|
DisableStopVm: b.config.DisableStopVm,
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
b.runner = common.NewRunner(steps, b.config.PackerConfig, ui)
|
b.runner = common.NewRunner(steps, b.config.PackerConfig, ui)
|
||||||
|
|
|
@ -30,6 +30,12 @@ func waitUntilVmDeleted(conn *oapi.Client, vmID string) error {
|
||||||
return <-errCh
|
return <-errCh
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func waitUntilVmStopped(conn *oapi.Client, vmID string) error {
|
||||||
|
errCh := make(chan error, 1)
|
||||||
|
go waitForState(errCh, "stopped", waitUntilVmStateFunc(conn, vmID))
|
||||||
|
return <-errCh
|
||||||
|
}
|
||||||
|
|
||||||
func waitForState(errCh chan<- error, target string, refresh stateRefreshFunc) error {
|
func waitForState(errCh chan<- error, target string, refresh stateRefreshFunc) error {
|
||||||
err := common.Retry(2, 2, 0, func(_ uint) (bool, error) {
|
err := common.Retry(2, 2, 0, func(_ uint) (bool, error) {
|
||||||
state, err := refresh()
|
state, err := refresh()
|
||||||
|
|
|
@ -0,0 +1,95 @@
|
||||||
|
package common
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/aws/aws-sdk-go/aws/awserr"
|
||||||
|
"github.com/hashicorp/packer/common"
|
||||||
|
"github.com/hashicorp/packer/helper/multistep"
|
||||||
|
"github.com/hashicorp/packer/packer"
|
||||||
|
"github.com/outscale/osc-go/oapi"
|
||||||
|
)
|
||||||
|
|
||||||
|
type StepStopBSUBackedVm struct {
|
||||||
|
Skip bool
|
||||||
|
DisableStopVm bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *StepStopBSUBackedVm) Run(ctx context.Context, state multistep.StateBag) multistep.StepAction {
|
||||||
|
oapiconn := state.Get("oapi").(*oapi.Client)
|
||||||
|
vm := state.Get("vm").(*oapi.Vm)
|
||||||
|
ui := state.Get("ui").(packer.Ui)
|
||||||
|
|
||||||
|
// Skip when it is a spot vm
|
||||||
|
if s.Skip {
|
||||||
|
return multistep.ActionContinue
|
||||||
|
}
|
||||||
|
|
||||||
|
var err error
|
||||||
|
|
||||||
|
if !s.DisableStopVm {
|
||||||
|
// Stop the vm so we can create an AMI from it
|
||||||
|
ui.Say("Stopping the source vm...")
|
||||||
|
|
||||||
|
// Amazon EC2 API follows an eventual consistency model.
|
||||||
|
|
||||||
|
// This means that if you run a command to modify or describe a resource
|
||||||
|
// that you just created, its ID might not have propagated throughout
|
||||||
|
// the system, and you will get an error responding that the resource
|
||||||
|
// does not exist.
|
||||||
|
|
||||||
|
// Work around this by retrying a few times, up to about 5 minutes.
|
||||||
|
err := common.Retry(10, 60, 6, func(i uint) (bool, error) {
|
||||||
|
ui.Message(fmt.Sprintf("Stopping vm, attempt %d", i+1))
|
||||||
|
|
||||||
|
_, err = oapiconn.POST_StopVms(oapi.StopVmsRequest{
|
||||||
|
VmIds: []string{vm.VmId},
|
||||||
|
})
|
||||||
|
|
||||||
|
if err == nil {
|
||||||
|
// success
|
||||||
|
return true, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if awsErr, ok := err.(awserr.Error); ok {
|
||||||
|
if awsErr.Code() == "InvalidVmID.NotFound" {
|
||||||
|
ui.Message(fmt.Sprintf(
|
||||||
|
"Error stopping vm; will retry ..."+
|
||||||
|
"Error: %s", err))
|
||||||
|
// retry
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// errored, but not in expected way. Don't want to retry
|
||||||
|
return true, err
|
||||||
|
})
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
err := fmt.Errorf("Error stopping vm: %s", err)
|
||||||
|
state.Put("error", err)
|
||||||
|
ui.Error(err.Error())
|
||||||
|
return multistep.ActionHalt
|
||||||
|
}
|
||||||
|
|
||||||
|
} else {
|
||||||
|
ui.Say("Automatic vm stop disabled. Please stop vm manually.")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Wait for the vm to actually stop
|
||||||
|
ui.Say("Waiting for the vm to stop...")
|
||||||
|
err = waitUntilVmStopped(oapiconn, vm.VmId)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
err := fmt.Errorf("Error waiting for vm to stop: %s", err)
|
||||||
|
state.Put("error", err)
|
||||||
|
ui.Error(err.Error())
|
||||||
|
return multistep.ActionHalt
|
||||||
|
}
|
||||||
|
|
||||||
|
return multistep.ActionContinue
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *StepStopBSUBackedVm) Cleanup(multistep.StateBag) {
|
||||||
|
// No cleanup...
|
||||||
|
}
|
Loading…
Reference in New Issue