func (p *JujuProvisioner) RemoveUnit(app provision.App, name string) error {
var unit provision.AppUnit
for _, unit = range app.ProvisionedUnits() {
if unit.GetName() == name {
break
}
}
if unit.GetName() != name {
return fmt.Errorf("App %q does not have a unit named %q.", app.GetName(), name)
}
return p.removeUnit(app, unit)
}
func (p *JujuProvisioner) removeUnit(app provision.App, unit provision.AppUnit) error {
var (
buf bytes.Buffer
err error
)
cmd := []string{"remove-unit", unit.GetName()}
// Sometimes juju gives the "no node" error. This is one of Zookeeper bad
// behaviors. Let's try it multiple times before raising the error to the
// user, and hope that someday we run away from Zookeeper.
for i := 0; i < destroyTries; i++ {
buf.Reset()
err = runCmd(false, &buf, &buf, cmd...)
if err != nil && unitNotFound(unit.GetName(), buf.Bytes()) {
err = nil
}
if err == nil {
break
}
}
if err != nil {
return cmdError(buf.String(), err, cmd)
}
if p.elbSupport() {
pUnit := provision.Unit{
Name: unit.GetName(),
InstanceId: unit.GetInstanceId(),
}
err = p.LoadBalancer().Deregister(app, pUnit)
}
conn, collection := p.unitsCollection()
defer conn.Close()
collection.RemoveId(unit.GetName())
go p.terminateMachines(app, unit)
return err
}