Handle restarting correctly, ensure power state propagates correctly

This commit is contained in:
Dane Everitt 2020-08-18 20:43:28 -07:00
parent b4875926c6
commit 956e87eb93
No known key found for this signature in database
GPG Key ID: EEA66103B3D71F53
2 changed files with 25 additions and 17 deletions

View File

@ -134,7 +134,11 @@ func (e *Environment) Stop() error {
return e.Terminate(os.Kill) return e.Terminate(os.Kill)
} }
e.setState(system.ProcessStoppingState) // If the process is already offline don't switch it back to stopping. Just leave it how
// it is and continue through to the stop handling for the process.
if e.State() != system.ProcessOfflineState {
e.setState(system.ProcessStoppingState)
}
// Only attempt to send the stop command to the instance if we are actually attached to // Only attempt to send the stop command to the instance if we are actually attached to
// the instance. If we are not for some reason, just send the container stop event. // the instance. If we are not for some reason, just send the container stop event.
@ -202,6 +206,14 @@ func (e *Environment) Terminate(signal os.Signal) error {
} }
if !c.State.Running { if !c.State.Running {
// If the container is not running but we're not already in a stopped state go ahead
// and update things to indicate we should be completely stopped now. Set to stopping
// first so crash detection is not triggered.
if e.State() != system.ProcessOfflineState {
e.setState(system.ProcessStoppingState)
e.setState(system.ProcessOfflineState)
}
return nil return nil
} }

View File

@ -95,22 +95,18 @@ func (s *Server) HandlePowerAction(action PowerAction, waitSeconds ...int) error
case PowerActionStop: case PowerActionStop:
// We're specificially waiting for the process to be stopped here, otherwise the lock is released // We're specificially waiting for the process to be stopped here, otherwise the lock is released
// too soon, and you can rack up all sorts of issues. // too soon, and you can rack up all sorts of issues.
return s.Environment.WaitForStop(10 * 60, true) return s.Environment.WaitForStop(10*60, true)
case PowerActionRestart: case PowerActionRestart:
// Only try to wait for stop if the process is currently running, otherwise just skip right to the if err := s.Environment.WaitForStop(10*60, true); err != nil {
// start event. // Even timeout errors should be bubbled back up the stack. If the process didn't stop
if r, _ := s.Environment.IsRunning(); !r { // nicely, but the terminate argument was passed then the server is stopped without an
if err := s.Environment.WaitForStop(10 * 60, true); err != nil { // error being returned.
// Even timeout errors should be bubbled back up the stack. If the process didn't stop //
// nicely, but the terminate argument was passed then the server is stopped without an // However, if terminate is not passed you'll get a context deadline error. We could
// error being returned. // probably handle that nicely here, but I'd rather just pass it back up the stack for now.
// // Either way, any type of error indicates we should not attempt to start the server back
// However, if terminate is not passed you'll get a context deadline error. We could // up.
// probably handle that nicely here, but I'd rather just pass it back up the stack for now. return err
// Either way, any type of error indicates we should not attempt to start the server back
// up.
return err
}
} }
// Now actually try to start the process by executing the normal pre-boot logic. // Now actually try to start the process by executing the normal pre-boot logic.
@ -158,4 +154,4 @@ func (s *Server) onBeforeStart() error {
} }
return nil return nil
} }