198 lines
7.4 KiB
Python
198 lines
7.4 KiB
Python
# Copyright (c) 2012 OpenStack, LLC.
|
|
#
|
|
# 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.
|
|
|
|
import logging
|
|
|
|
from sqlalchemy.orm import exc
|
|
|
|
from quantum.api import api_common
|
|
from quantum.common import exceptions as q_exc
|
|
import quantum.db.api as db
|
|
from quantum.db import models_v2
|
|
from quantum.openstack.common import cfg
|
|
from quantum.plugins.linuxbridge.common import config
|
|
from quantum.plugins.linuxbridge.db import l2network_models_v2
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
def initialize():
|
|
options = {"sql_connection": "%s" % cfg.CONF.DATABASE.sql_connection}
|
|
options.update({"sql_max_retries": cfg.CONF.DATABASE.sql_max_retries})
|
|
options.update({"reconnect_interval":
|
|
cfg.CONF.DATABASE.reconnect_interval})
|
|
options.update({"base": models_v2.model_base.BASEV2})
|
|
db.configure_db(options)
|
|
|
|
|
|
def sync_network_states(network_vlan_ranges):
|
|
"""Synchronize network_states table with current configured VLAN ranges."""
|
|
|
|
# process vlan ranges for each physical network separately
|
|
for physical_network, vlan_ranges in network_vlan_ranges.iteritems():
|
|
|
|
# determine current configured allocatable vlans for this
|
|
# physical network
|
|
vlan_ids = set()
|
|
for vlan_range in vlan_ranges:
|
|
vlan_ids |= set(xrange(vlan_range[0], vlan_range[1] + 1))
|
|
|
|
session = db.get_session()
|
|
with session.begin():
|
|
# remove from table unallocated vlans not currently allocatable
|
|
try:
|
|
states = (session.query(l2network_models_v2.NetworkState).
|
|
filter_by(physical_network=physical_network).
|
|
all())
|
|
for state in states:
|
|
try:
|
|
# see if vlan is allocatable
|
|
vlan_ids.remove(state.vlan_id)
|
|
except KeyError:
|
|
# it's not allocatable, so check if its allocated
|
|
if not state.allocated:
|
|
# it's not, so remove it from table
|
|
LOG.debug("removing vlan %s on physical network "
|
|
"%s from pool" %
|
|
(state.vlan_id, physical_network))
|
|
session.delete(state)
|
|
except exc.NoResultFound:
|
|
pass
|
|
|
|
# add missing allocatable vlans to table
|
|
for vlan_id in sorted(vlan_ids):
|
|
state = l2network_models_v2.NetworkState(physical_network,
|
|
vlan_id)
|
|
session.add(state)
|
|
|
|
|
|
def get_network_state(physical_network, vlan_id):
|
|
"""Get state of specified network"""
|
|
|
|
session = db.get_session()
|
|
try:
|
|
state = (session.query(l2network_models_v2.NetworkState).
|
|
filter_by(physical_network=physical_network,
|
|
vlan_id=vlan_id).
|
|
one())
|
|
return state
|
|
except exc.NoResultFound:
|
|
return None
|
|
|
|
|
|
def reserve_network(session):
|
|
with session.begin(subtransactions=True):
|
|
state = (session.query(l2network_models_v2.NetworkState).
|
|
filter_by(allocated=False).
|
|
first())
|
|
if not state:
|
|
raise q_exc.NoNetworkAvailable()
|
|
LOG.debug("reserving vlan %s on physical network %s from pool" %
|
|
(state.vlan_id, state.physical_network))
|
|
state.allocated = True
|
|
return (state.physical_network, state.vlan_id)
|
|
|
|
|
|
def reserve_specific_network(session, physical_network, vlan_id):
|
|
with session.begin(subtransactions=True):
|
|
try:
|
|
state = (session.query(l2network_models_v2.NetworkState).
|
|
filter_by(physical_network=physical_network,
|
|
vlan_id=vlan_id).
|
|
one())
|
|
if state.allocated:
|
|
raise q_exc.VlanIdInUse(vlan_id=vlan_id,
|
|
physical_network=physical_network)
|
|
LOG.debug("reserving specific vlan %s on physical network %s "
|
|
"from pool" % (vlan_id, physical_network))
|
|
state.allocated = True
|
|
except exc.NoResultFound:
|
|
LOG.debug("reserving specific vlan %s on physical network %s "
|
|
"outside pool" % (vlan_id, physical_network))
|
|
state = l2network_models_v2.NetworkState(physical_network, vlan_id)
|
|
state.allocated = True
|
|
session.add(state)
|
|
|
|
|
|
def release_network(session, physical_network, vlan_id, network_vlan_ranges):
|
|
with session.begin(subtransactions=True):
|
|
try:
|
|
state = (session.query(l2network_models_v2.NetworkState).
|
|
filter_by(physical_network=physical_network,
|
|
vlan_id=vlan_id).
|
|
one())
|
|
state.allocated = False
|
|
inside = False
|
|
for vlan_range in network_vlan_ranges.get(physical_network, []):
|
|
if vlan_id >= vlan_range[0] and vlan_id <= vlan_range[1]:
|
|
inside = True
|
|
break
|
|
if inside:
|
|
LOG.debug("releasing vlan %s on physical network %s to pool" %
|
|
(vlan_id, physical_network))
|
|
else:
|
|
LOG.debug("releasing vlan %s on physical network %s outside "
|
|
"pool" % (vlan_id, physical_network))
|
|
session.delete(state)
|
|
except exc.NoResultFound:
|
|
LOG.warning("vlan_id %s on physical network %s not found" %
|
|
(vlan_id, physical_network))
|
|
|
|
|
|
def add_network_binding(session, network_id, physical_network, vlan_id):
|
|
with session.begin(subtransactions=True):
|
|
binding = l2network_models_v2.NetworkBinding(network_id,
|
|
physical_network, vlan_id)
|
|
session.add(binding)
|
|
|
|
|
|
def get_network_binding(session, network_id):
|
|
try:
|
|
binding = (session.query(l2network_models_v2.NetworkBinding).
|
|
filter_by(network_id=network_id).
|
|
one())
|
|
return binding
|
|
except exc.NoResultFound:
|
|
return
|
|
|
|
|
|
def get_port_from_device(device):
|
|
"""Get port from database"""
|
|
LOG.debug("get_port_from_device() called")
|
|
session = db.get_session()
|
|
ports = session.query(models_v2.Port).all()
|
|
if not ports:
|
|
return
|
|
for port in ports:
|
|
if port['id'].startswith(device):
|
|
return port
|
|
return
|
|
|
|
|
|
def set_port_status(port_id, status):
|
|
"""Set the port status"""
|
|
LOG.debug("set_port_status as %s called", status)
|
|
session = db.get_session()
|
|
try:
|
|
port = session.query(models_v2.Port).filter_by(id=port_id).one()
|
|
port['status'] = status
|
|
if status == api_common.PORT_STATUS_DOWN:
|
|
port['device_id'] = ''
|
|
port['device_owner'] = ''
|
|
session.merge(port)
|
|
session.flush()
|
|
except exc.NoResultFound:
|
|
raise q_exc.PortNotFound(port_id=port_id)
|