84 lines
3.3 KiB
Python
84 lines
3.3 KiB
Python
# Copyright (C) 2014 eNovance SAS <licensing@enovance.com>
|
|
#
|
|
# 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.api.definitions import portbindings
|
|
from neutron_lib import constants
|
|
from neutron_lib.plugins import directory
|
|
from sqlalchemy import func
|
|
|
|
from neutron.callbacks import events
|
|
from neutron.callbacks import registry
|
|
from neutron.callbacks import resources
|
|
from neutron.db import l3_agentschedulers_db as l3_sch_db
|
|
from neutron.db.models import agent as agent_model
|
|
from neutron.db.models import l3agent as rb_model
|
|
|
|
|
|
class L3_HA_scheduler_db_mixin(l3_sch_db.AZL3AgentSchedulerDbMixin):
|
|
|
|
def get_l3_agents_ordered_by_num_routers(self, context, agent_ids):
|
|
if not agent_ids:
|
|
return []
|
|
query = (context.session.query(agent_model.Agent, func.count(
|
|
rb_model.RouterL3AgentBinding.router_id)
|
|
.label('count')).
|
|
outerjoin(rb_model.RouterL3AgentBinding).
|
|
group_by(agent_model.Agent.id).
|
|
filter(agent_model.Agent.id.in_(agent_ids)).
|
|
order_by('count'))
|
|
|
|
return [record[0] for record in query]
|
|
|
|
def _get_agents_dict_for_router(self, agents_and_states):
|
|
agents = []
|
|
for agent, ha_state in agents_and_states:
|
|
l3_agent_dict = self._make_agent_dict(agent)
|
|
l3_agent_dict['ha_state'] = ha_state
|
|
agents.append(l3_agent_dict)
|
|
return {'agents': agents}
|
|
|
|
def list_l3_agents_hosting_router(self, context, router_id):
|
|
with context.session.begin(subtransactions=True):
|
|
router_db = self._get_router(context, router_id)
|
|
if router_db.extra_attributes.ha:
|
|
bindings = self.get_l3_bindings_hosting_router_with_ha_states(
|
|
context, router_id)
|
|
else:
|
|
bindings = self._get_l3_bindings_hosting_routers(
|
|
context, [router_id])
|
|
bindings = [(binding.l3_agent, None) for binding in bindings]
|
|
|
|
return self._get_agents_dict_for_router(bindings)
|
|
|
|
|
|
def _notify_l3_agent_ha_port_update(resource, event, trigger, **kwargs):
|
|
new_port = kwargs.get('port')
|
|
original_port = kwargs.get('original_port')
|
|
context = kwargs.get('context')
|
|
host = new_port[portbindings.HOST_ID]
|
|
|
|
if new_port and original_port and host:
|
|
new_device_owner = new_port.get('device_owner', '')
|
|
if (new_device_owner == constants.DEVICE_OWNER_ROUTER_HA_INTF and
|
|
new_port['status'] == constants.PORT_STATUS_ACTIVE and
|
|
original_port['status'] != new_port['status']):
|
|
l3plugin = directory.get_plugin(constants.L3)
|
|
l3plugin.l3_rpc_notifier.routers_updated_on_host(
|
|
context, [new_port['device_id']], host)
|
|
|
|
|
|
def subscribe():
|
|
registry.subscribe(
|
|
_notify_l3_agent_ha_port_update, resources.PORT, events.AFTER_UPDATE)
|