**** CubicPower OpenStack Study ****
# vim: tabstop=4 shiftwidth=4 softtabstop=4
#
# Copyright 2013 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.
#
# @author: Kaiwei Fan, VMware, Inc
import sqlalchemy as sa
from neutron.db import model_base
from neutron.extensions import routerservicetype as rst
**** CubicPower OpenStack Study ****
class RouterServiceTypeBinding(model_base.BASEV2):
router_id = sa.Column(sa.String(36),
sa.ForeignKey('routers.id', ondelete="CASCADE"),
primary_key=True)
service_type_id = sa.Column(sa.String(36),
nullable=False)
**** CubicPower OpenStack Study ****
class RouterServiceTypeDbMixin(object):
"""Mixin class to add router service type."""
**** CubicPower OpenStack Study ****
def _process_create_router_service_type_id(self, context, router):
with context.session.begin(subtransactions=True):
db = RouterServiceTypeBinding(
router_id=router['id'],
service_type_id=router[rst.SERVICE_TYPE_ID])
context.session.add(db)
return self._make_router_service_type_id_dict(db)
**** CubicPower OpenStack Study ****
def _extend_router_service_type_id_dict(self, context, router):
rsbind = self._get_router_service_type_id_binding(
context, router['id'])
if rsbind:
router[rst.SERVICE_TYPE_ID] = rsbind['service_type_id']
**** CubicPower OpenStack Study ****
def _get_router_service_type_id_binding(self, context, router_id):
query = self._model_query(context, RouterServiceTypeBinding)
query = query.filter(
RouterServiceTypeBinding.router_id == router_id)
return query.first()
**** CubicPower OpenStack Study ****
def _make_router_service_type_id_dict(self, router_service_type):
res = {'router_id': router_service_type['router_id'],
'service_type_id': router_service_type[rst.SERVICE_TYPE_ID]}
return self._fields(res, None)