Change host restart to use the driver implementation

Fixes #2140

Signed-off-by: Neil Wilson <neil@aldur.co.uk>
This commit is contained in:
Neil Wilson 2015-11-09 12:58:56 +00:00 committed by David Gageot
parent acf2d1da4c
commit c045ceec1e
1 changed files with 5 additions and 14 deletions

View File

@ -122,24 +122,15 @@ func (h *Host) Kill() error {
}
func (h *Host) Restart() error {
if drivers.MachineInState(h.Driver, state.Stopped)() {
return h.Start()
}
if drivers.MachineInState(h.Driver, state.Running)() {
if err := h.Stop(); err != nil {
return err
}
if err := mcnutils.WaitFor(drivers.MachineInState(h.Driver, state.Stopped)); err != nil {
if err := h.Driver.Restart(); err != nil {
return err
}
return mcnutils.WaitFor(drivers.MachineInState(h.Driver, state.Running))
}
if err := h.Start(); err != nil {
return err
}
if err := mcnutils.WaitFor(drivers.MachineInState(h.Driver, state.Running)); err != nil {
return err
}
return nil
}