// UnbindUnit makes the unbind between the service instance and an unit.
func (si *ServiceInstance) UnbindUnit(app bind.App, unit bind.Unit) error {
endpoint, err := si.Service().getClient("production")
if err != nil {
return err
}
conn, err := db.Conn()
if err != nil {
return err
}
defer conn.Close()
updateOp := bson.M{"$pull": bson.M{"units": unit.GetID()}}
err = conn.ServiceInstances().Update(bson.M{"name": si.Name, "service_name": si.ServiceName, "units": unit.GetID()}, updateOp)
if err != nil {
if err == mgo.ErrNotFound {
return ErrUnitNotBound
}
return err
}
err = endpoint.UnbindUnit(si, app, unit)
if err != nil {
rollbackErr := si.update(bson.M{"$addToSet": bson.M{"units": unit.GetID()}})
if rollbackErr != nil {
log.Errorf("[unbind unit] could not add bound unit back to db after failure: %s", rollbackErr)
}
return err
}
return nil
}
