Merge "enginefacade: 'fixed_ip'"

This commit is contained in:
Jenkins 2016-01-25 16:06:02 +00:00 committed by Gerrit Code Review
commit 54f7925576
1 changed files with 144 additions and 155 deletions

View File

@ -354,8 +354,7 @@ def _sync_floating_ips(context, project_id, user_id):
def _sync_fixed_ips(context, project_id, user_id): def _sync_fixed_ips(context, project_id, user_id):
return dict(fixed_ips=_fixed_ip_count_by_project( return dict(fixed_ips=_fixed_ip_count_by_project(context, project_id))
context, project_id, context.session))
def _sync_security_groups(context, project_id, user_id): def _sync_security_groups(context, project_id, user_id):
@ -1116,6 +1115,7 @@ def dnsdomain_get_all(context):
@oslo_db_api.wrap_db_retry(max_retries=5, retry_on_deadlock=True, @oslo_db_api.wrap_db_retry(max_retries=5, retry_on_deadlock=True,
retry_on_request=True) retry_on_request=True)
@main_context_manager.writer
def fixed_ip_associate(context, address, instance_uuid, network_id=None, def fixed_ip_associate(context, address, instance_uuid, network_id=None,
reserved=False, virtual_interface_id=None): reserved=False, virtual_interface_id=None):
"""Keyword arguments: """Keyword arguments:
@ -1125,12 +1125,9 @@ def fixed_ip_associate(context, address, instance_uuid, network_id=None,
if not uuidutils.is_uuid_like(instance_uuid): if not uuidutils.is_uuid_like(instance_uuid):
raise exception.InvalidUUID(uuid=instance_uuid) raise exception.InvalidUUID(uuid=instance_uuid)
session = get_session()
with session.begin():
network_or_none = or_(models.FixedIp.network_id == network_id, network_or_none = or_(models.FixedIp.network_id == network_id,
models.FixedIp.network_id == null()) models.FixedIp.network_id == null())
fixed_ip_ref = model_query(context, models.FixedIp, session=session, fixed_ip_ref = model_query(context, models.FixedIp, read_deleted="no").\
read_deleted="no").\
filter(network_or_none).\ filter(network_or_none).\
filter_by(reserved=reserved).\ filter_by(reserved=reserved).\
filter_by(address=address).\ filter_by(address=address).\
@ -1150,8 +1147,7 @@ def fixed_ip_associate(context, address, instance_uuid, network_id=None,
if virtual_interface_id: if virtual_interface_id:
params['virtual_interface_id'] = virtual_interface_id params['virtual_interface_id'] = virtual_interface_id
rows_updated = model_query(context, models.FixedIp, session=session, rows_updated = model_query(context, models.FixedIp, read_deleted="no").\
read_deleted="no").\
filter_by(id=fixed_ip_ref.id).\ filter_by(id=fixed_ip_ref.id).\
filter(network_or_none).\ filter(network_or_none).\
filter_by(reserved=reserved).\ filter_by(reserved=reserved).\
@ -1169,17 +1165,15 @@ def fixed_ip_associate(context, address, instance_uuid, network_id=None,
@oslo_db_api.wrap_db_retry(max_retries=5, retry_on_deadlock=True, @oslo_db_api.wrap_db_retry(max_retries=5, retry_on_deadlock=True,
retry_on_request=True) retry_on_request=True)
@main_context_manager.writer
def fixed_ip_associate_pool(context, network_id, instance_uuid=None, def fixed_ip_associate_pool(context, network_id, instance_uuid=None,
host=None, virtual_interface_id=None): host=None, virtual_interface_id=None):
if instance_uuid and not uuidutils.is_uuid_like(instance_uuid): if instance_uuid and not uuidutils.is_uuid_like(instance_uuid):
raise exception.InvalidUUID(uuid=instance_uuid) raise exception.InvalidUUID(uuid=instance_uuid)
session = get_session()
with session.begin():
network_or_none = or_(models.FixedIp.network_id == network_id, network_or_none = or_(models.FixedIp.network_id == network_id,
models.FixedIp.network_id == null()) models.FixedIp.network_id == null())
fixed_ip_ref = model_query(context, models.FixedIp, session=session, fixed_ip_ref = model_query(context, models.FixedIp, read_deleted="no").\
read_deleted="no").\
filter(network_or_none).\ filter(network_or_none).\
filter_by(reserved=False).\ filter_by(reserved=False).\
filter_by(instance_uuid=None).\ filter_by(instance_uuid=None).\
@ -1200,8 +1194,7 @@ def fixed_ip_associate_pool(context, network_id, instance_uuid=None,
if virtual_interface_id: if virtual_interface_id:
params['virtual_interface_id'] = virtual_interface_id params['virtual_interface_id'] = virtual_interface_id
rows_updated = model_query(context, models.FixedIp, session=session, rows_updated = model_query(context, models.FixedIp, read_deleted="no").\
read_deleted="no").\
filter_by(id=fixed_ip_ref['id']).\ filter_by(id=fixed_ip_ref['id']).\
filter_by(network_id=fixed_ip_ref['network_id']).\ filter_by(network_id=fixed_ip_ref['network_id']).\
filter_by(reserved=False).\ filter_by(reserved=False).\
@ -1221,48 +1214,46 @@ def fixed_ip_associate_pool(context, network_id, instance_uuid=None,
@require_context @require_context
@main_context_manager.writer
def fixed_ip_create(context, values): def fixed_ip_create(context, values):
fixed_ip_ref = models.FixedIp() fixed_ip_ref = models.FixedIp()
fixed_ip_ref.update(values) fixed_ip_ref.update(values)
try: try:
fixed_ip_ref.save() fixed_ip_ref.save(context.session)
except db_exc.DBDuplicateEntry: except db_exc.DBDuplicateEntry:
raise exception.FixedIpExists(address=values['address']) raise exception.FixedIpExists(address=values['address'])
return fixed_ip_ref return fixed_ip_ref
@require_context @require_context
@main_context_manager.writer
def fixed_ip_bulk_create(context, ips): def fixed_ip_bulk_create(context, ips):
engine = get_engine()
with engine.begin() as conn:
try: try:
tab = models.FixedIp.__table__ tab = models.FixedIp.__table__
conn.execute(tab.insert(), ips) context.session.execute(tab.insert(), ips)
except db_exc.DBDuplicateEntry as e: except db_exc.DBDuplicateEntry as e:
raise exception.FixedIpExists(address=e.value) raise exception.FixedIpExists(address=e.value)
@require_context @require_context
@main_context_manager.writer
def fixed_ip_disassociate(context, address): def fixed_ip_disassociate(context, address):
session = get_session() _fixed_ip_get_by_address(context, address).update(
with session.begin(): {'instance_uuid': None,
_fixed_ip_get_by_address(context, address, session=session).\
update({'instance_uuid': None,
'virtual_interface_id': None}) 'virtual_interface_id': None})
@main_context_manager.writer
def fixed_ip_disassociate_all_by_timeout(context, host, time): def fixed_ip_disassociate_all_by_timeout(context, host, time):
session = get_session()
# NOTE(vish): only update fixed ips that "belong" to this # NOTE(vish): only update fixed ips that "belong" to this
# host; i.e. the network host or the instance # host; i.e. the network host or the instance
# host matches. Two queries necessary because # host matches. Two queries necessary because
# join with update doesn't work. # join with update doesn't work.
with session.begin():
host_filter = or_(and_(models.Instance.host == host, host_filter = or_(and_(models.Instance.host == host,
models.Network.multi_host == true()), models.Network.multi_host == true()),
models.Network.host == host) models.Network.host == host)
result = model_query(context, models.FixedIp, (models.FixedIp.id,), result = model_query(context, models.FixedIp, (models.FixedIp.id,),
read_deleted="no", session=session).\ read_deleted="no").\
filter(models.FixedIp.allocated == false()).\ filter(models.FixedIp.allocated == false()).\
filter(models.FixedIp.updated_at < time).\ filter(models.FixedIp.updated_at < time).\
join((models.Network, join((models.Network,
@ -1274,7 +1265,7 @@ def fixed_ip_disassociate_all_by_timeout(context, host, time):
fixed_ip_ids = [fip[0] for fip in result] fixed_ip_ids = [fip[0] for fip in result]
if not fixed_ip_ids: if not fixed_ip_ids:
return 0 return 0
result = model_query(context, models.FixedIp, session=session).\ result = model_query(context, models.FixedIp).\
filter(models.FixedIp.id.in_(fixed_ip_ids)).\ filter(models.FixedIp.id.in_(fixed_ip_ids)).\
update({'instance_uuid': None, update({'instance_uuid': None,
'leased': False, 'leased': False,
@ -1284,6 +1275,7 @@ def fixed_ip_disassociate_all_by_timeout(context, host, time):
@require_context @require_context
@main_context_manager.reader
def fixed_ip_get(context, id, get_network=False): def fixed_ip_get(context, id, get_network=False):
query = model_query(context, models.FixedIp).filter_by(id=id) query = model_query(context, models.FixedIp).filter_by(id=id)
if get_network: if get_network:
@ -1303,6 +1295,7 @@ def fixed_ip_get(context, id, get_network=False):
return result return result
@main_context_manager.reader
def fixed_ip_get_all(context): def fixed_ip_get_all(context):
result = model_query(context, models.FixedIp, read_deleted="yes").all() result = model_query(context, models.FixedIp, read_deleted="yes").all()
if not result: if not result:
@ -1312,21 +1305,18 @@ def fixed_ip_get_all(context):
@require_context @require_context
@main_context_manager.reader
def fixed_ip_get_by_address(context, address, columns_to_join=None): def fixed_ip_get_by_address(context, address, columns_to_join=None):
return _fixed_ip_get_by_address(context, address, return _fixed_ip_get_by_address(context, address,
columns_to_join=columns_to_join) columns_to_join=columns_to_join)
def _fixed_ip_get_by_address(context, address, session=None, def _fixed_ip_get_by_address(context, address, columns_to_join=None):
columns_to_join=None):
if session is None:
session = get_session()
if columns_to_join is None: if columns_to_join is None:
columns_to_join = [] columns_to_join = []
with session.begin(subtransactions=True):
try: try:
result = model_query(context, models.FixedIp, session=session) result = model_query(context, models.FixedIp)
for column in columns_to_join: for column in columns_to_join:
result = result.options(joinedload_all(column)) result = result.options(joinedload_all(column))
result = result.filter_by(address=address).first() result = result.filter_by(address=address).first()
@ -1344,15 +1334,15 @@ def _fixed_ip_get_by_address(context, address, session=None,
instance = _instance_get_by_uuid( instance = _instance_get_by_uuid(
context.elevated(read_deleted='yes'), context.elevated(read_deleted='yes'),
result['instance_uuid'], result['instance_uuid'],
session context.session
) )
nova.context.authorize_project_context(context, nova.context.authorize_project_context(context,
instance.project_id) instance.project_id)
return result return result
@require_context @require_context
@main_context_manager.reader
def fixed_ip_get_by_floating_address(context, floating_address): def fixed_ip_get_by_floating_address(context, floating_address):
return model_query(context, models.FixedIp).\ return model_query(context, models.FixedIp).\
join(models.FloatingIp, join(models.FloatingIp,
@ -1364,6 +1354,7 @@ def fixed_ip_get_by_floating_address(context, floating_address):
@require_context @require_context
@main_context_manager.reader
def fixed_ip_get_by_instance(context, instance_uuid): def fixed_ip_get_by_instance(context, instance_uuid):
if not uuidutils.is_uuid_like(instance_uuid): if not uuidutils.is_uuid_like(instance_uuid):
raise exception.InvalidUUID(uuid=instance_uuid) raise exception.InvalidUUID(uuid=instance_uuid)
@ -1387,20 +1378,19 @@ def fixed_ip_get_by_instance(context, instance_uuid):
return result return result
@main_context_manager.reader
def fixed_ip_get_by_host(context, host): def fixed_ip_get_by_host(context, host):
session = get_session() instance_uuids = _instance_get_all_uuids_by_host(context, host)
with session.begin():
instance_uuids = _instance_get_all_uuids_by_host(context, host,
session=session)
if not instance_uuids: if not instance_uuids:
return [] return []
return model_query(context, models.FixedIp, session=session).\ return model_query(context, models.FixedIp).\
filter(models.FixedIp.instance_uuid.in_(instance_uuids)).\ filter(models.FixedIp.instance_uuid.in_(instance_uuids)).\
all() all()
@require_context @require_context
@main_context_manager.reader
def fixed_ip_get_by_network_host(context, network_id, host): def fixed_ip_get_by_network_host(context, network_id, host):
result = model_query(context, models.FixedIp, read_deleted="no").\ result = model_query(context, models.FixedIp, read_deleted="no").\
filter_by(network_id=network_id).\ filter_by(network_id=network_id).\
@ -1414,6 +1404,7 @@ def fixed_ip_get_by_network_host(context, network_id, host):
@require_context @require_context
@main_context_manager.reader
def fixed_ips_by_virtual_interface(context, vif_id): def fixed_ips_by_virtual_interface(context, vif_id):
result = model_query(context, models.FixedIp, read_deleted="no").\ result = model_query(context, models.FixedIp, read_deleted="no").\
filter_by(virtual_interface_id=vif_id).\ filter_by(virtual_interface_id=vif_id).\
@ -1425,17 +1416,15 @@ def fixed_ips_by_virtual_interface(context, vif_id):
@require_context @require_context
@main_context_manager.writer
def fixed_ip_update(context, address, values): def fixed_ip_update(context, address, values):
session = get_session() _fixed_ip_get_by_address(context, address).update(values)
with session.begin():
_fixed_ip_get_by_address(context, address, session=session).\
update(values)
def _fixed_ip_count_by_project(context, project_id, session=None): def _fixed_ip_count_by_project(context, project_id):
nova.context.authorize_project_context(context, project_id) nova.context.authorize_project_context(context, project_id)
return model_query(context, models.FixedIp, (models.FixedIp.id,), return model_query(context, models.FixedIp, (models.FixedIp.id,),
read_deleted="no", session=session).\ read_deleted="no").\
join((models.Instance, join((models.Instance,
models.Instance.uuid == models.FixedIp.instance_uuid)).\ models.Instance.uuid == models.FixedIp.instance_uuid)).\
filter(models.Instance.project_id == project_id).\ filter(models.Instance.project_id == project_id).\