func (w *upgradesteps) waitForOtherControllers(info *state.UpgradeInfo) error {
watcher := info.Watch()
defer watcher.Stop()
maxWait := getUpgradeStartTimeout(w.isMaster)
timeout := time.After(maxWait)
for {
select {
case <-watcher.Changes():
if err := info.Refresh(); err != nil {
return errors.Trace(err)
}
if w.isMaster {
if ready, err := info.AllProvisionedControllersReady(); err != nil {
return errors.Trace(err)
} else if ready {
// All controllers ready to start upgrade
err := info.SetStatus(state.UpgradeRunning)
return errors.Trace(err)
}
} else {
if info.Status() == state.UpgradeFinishing {
// Master is done, ok to proceed
return nil
}
}
case <-timeout:
if w.isMaster {
if err := info.Abort(); err != nil {
return errors.Annotate(err, "unable to abort upgrade")
}
}
return errors.Errorf("timed out after %s", maxWait)
case <-w.tomb.Dying():
return tomb.ErrDying
}
}
}
