d29a0baa80
Access to neutron.db.api's context manager is already in neutron-lib and in fact neutron is already using it as a shim. This patch switches over context manager access to use neutron-lib's accessors. Also see https://review.openstack.org/#/c/613122 Change-Id: I13eb3a25a5bd83bb00dfa4a7430324551fea0f2e
104 lines
4.0 KiB
Python
104 lines
4.0 KiB
Python
# Copyright 2014 VMware, Inc.
|
|
#
|
|
# All Rights Reserved
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
#
|
|
|
|
from neutron_lib.db import api as db_api
|
|
from oslo_db import exception as d_exc
|
|
from oslo_log import log as logging
|
|
from sqlalchemy import orm
|
|
|
|
from vmware_nsx._i18n import _
|
|
from vmware_nsx.common import exceptions as p_exc
|
|
from vmware_nsx.db import nsx_models
|
|
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
def lsn_add(context, network_id, lsn_id):
|
|
"""Add Logical Service Node information to persistent datastore."""
|
|
with db_api.CONTEXT_WRITER.using(context):
|
|
lsn = nsx_models.Lsn(network_id, lsn_id)
|
|
context.session.add(lsn)
|
|
|
|
|
|
def lsn_remove(context, lsn_id):
|
|
"""Remove Logical Service Node information from datastore given its id."""
|
|
with db_api.CONTEXT_WRITER.using(context):
|
|
context.session.query(nsx_models.Lsn).filter_by(lsn_id=lsn_id).delete()
|
|
|
|
|
|
def lsn_remove_for_network(context, network_id):
|
|
"""Remove information about the Logical Service Node given its network."""
|
|
with db_api.CONTEXT_WRITER.using(context):
|
|
context.session.query(nsx_models.Lsn).filter_by(
|
|
net_id=network_id).delete()
|
|
|
|
|
|
def lsn_get_for_network(context, network_id, raise_on_err=True):
|
|
"""Retrieve LSN information given its network id."""
|
|
query = context.session.query(nsx_models.Lsn)
|
|
try:
|
|
return query.filter_by(net_id=network_id).one()
|
|
except (orm.exc.NoResultFound, d_exc.DBError):
|
|
msg = _('Unable to find Logical Service Node for network %s')
|
|
if raise_on_err:
|
|
LOG.error(msg, network_id)
|
|
raise p_exc.LsnNotFound(entity='network',
|
|
entity_id=network_id)
|
|
else:
|
|
LOG.warning(msg, network_id)
|
|
|
|
|
|
def lsn_port_add_for_lsn(context, lsn_port_id, subnet_id, mac, lsn_id):
|
|
"""Add Logical Service Node Port information to persistent datastore."""
|
|
with db_api.CONTEXT_WRITER.using(context):
|
|
lsn_port = nsx_models.LsnPort(lsn_port_id, subnet_id, mac, lsn_id)
|
|
context.session.add(lsn_port)
|
|
|
|
|
|
def lsn_port_get_for_subnet(context, subnet_id, raise_on_err=True):
|
|
"""Return Logical Service Node Port information given its subnet id."""
|
|
with db_api.CONTEXT_READER.using(context):
|
|
try:
|
|
return (context.session.query(nsx_models.LsnPort).
|
|
filter_by(sub_id=subnet_id).one())
|
|
except (orm.exc.NoResultFound, d_exc.DBError):
|
|
if raise_on_err:
|
|
raise p_exc.LsnPortNotFound(lsn_id=None,
|
|
entity='subnet',
|
|
entity_id=subnet_id)
|
|
|
|
|
|
def lsn_port_get_for_mac(context, mac_address, raise_on_err=True):
|
|
"""Return Logical Service Node Port information given its mac address."""
|
|
with db_api.CONTEXT_READER.using(context):
|
|
try:
|
|
return (context.session.query(nsx_models.LsnPort).
|
|
filter_by(mac_addr=mac_address).one())
|
|
except (orm.exc.NoResultFound, d_exc.DBError):
|
|
if raise_on_err:
|
|
raise p_exc.LsnPortNotFound(lsn_id=None,
|
|
entity='mac',
|
|
entity_id=mac_address)
|
|
|
|
|
|
def lsn_port_remove(context, lsn_port_id):
|
|
"""Remove Logical Service Node port from the given Logical Service Node."""
|
|
with db_api.CONTEXT_WRITER.using(context):
|
|
(context.session.query(nsx_models.LsnPort).
|
|
filter_by(lsn_port_id=lsn_port_id).delete())
|