diff options
| -rw-r--r-- | nova/db/sqlalchemy/api.py | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/nova/db/sqlalchemy/api.py b/nova/db/sqlalchemy/api.py index d0203f29e..d848d8c37 100644 --- a/nova/db/sqlalchemy/api.py +++ b/nova/db/sqlalchemy/api.py @@ -472,6 +472,16 @@ def service_update(context, service_id, values): ################### +def _compute_node_get(context, compute_id, session=None): + result = model_query(context, models.ComputeNode, session=session).\ + filter_by(id=compute_id).\ + first() + + if not result: + raise exception.ComputeHostNotFound(host=compute_id) + + return result + @require_admin_context def compute_node_get_all(context, session=None): @@ -532,7 +542,7 @@ def compute_node_update(context, compute_id, values, auto_adjust): if auto_adjust: _adjust_compute_node_values_for_utilization(context, values, session) with session.begin(subtransactions=True): - compute_ref = compute_node_get(context, compute_id, session=session) + compute_ref = _compute_node_get(context, compute_id, session=session) compute_ref.update(values) compute_ref.save(session=session) |