func (bm *BaseMachine) HandleStart(ctx context.Context) error {
origState := bm.State()
currentState := origState
bm.PushEvent("Checking machine state", 10, machinestate.Starting)
realState, meta, err := bm.machine.Info(ctx)
if err != nil {
return err
}
defer func() {
bm.Log.Debug("exit: origState=%s, currentState=%s, err=%v", origState, currentState, err)
if err != nil && origState != currentState {
modelhelper.ChangeMachineState(bm.ObjectId, "Machine is marked as "+origState.String(), origState)
}
}()
bm.Log.Debug("origState=%s, currentState=%s, realState=%s", origState, currentState, realState)
if !realState.In(machinestate.Running, machinestate.Starting) {
currentState = machinestate.Starting
bm.PushEvent("Starting machine", 25, currentState)
err = modelhelper.ChangeMachineState(bm.ObjectId, "Machine is starting", currentState)
if err != nil {
return err
}
meta, err = bm.machine.Start(ctx)
if err != nil {
return stack.NewEventerError(err)
}
}
bm.PushEvent("Checking remote machine", 75, currentState)
dialState, err := bm.WaitKlientReady(0)
if err != nil {
bm.Log.Debug("waiting for klient failed with error: %s", err)
currentState = machinestate.Stopped
return stack.NewEventerError(err)
}
currentState = machinestate.Running
if err := bm.updateMachine(dialState, meta, currentState); err != nil {
return fmt.Errorf("failed to update machine: %s", err)
}
return nil
}
func (bm *BaseMachine) HandleStop(ctx context.Context) error {
origState := bm.State()
currentState := origState
bm.PushEvent("Checking machine state", 10, machinestate.Stopping)
realState, meta, err := bm.machine.Info(ctx)
if err != nil {
return err
}
defer func() {
bm.Log.Debug("stop exit: origState=%s, currentState=%s, err=%v", origState, currentState, err)
if err != nil && origState != currentState {
modelhelper.ChangeMachineState(bm.ObjectId, "Machine is marked as "+origState.String(), origState)
}
}()
bm.Log.Debug("stop origState=%s, currentState=%s, realState=%s", origState, currentState, realState)
if !realState.In(machinestate.Stopping, machinestate.Stopped) {
currentState = machinestate.Stopping
bm.PushEvent("Stopping machine", 25, currentState)
err = modelhelper.ChangeMachineState(bm.ObjectId, "Machine is stopping", currentState)
if err != nil {
return err
}
meta, err = bm.machine.Stop(ctx)
if err != nil {
return stack.NewEventerError(err)
}
currentState = machinestate.Stopped
}
if err := bm.updateMachine(nil, meta, currentState); err != nil {
return fmt.Errorf("failed to update machine: %s", err)
}
return nil
}