neutron/services/metering/metering_plugin.py

   1 # Copyright (C) 2013 eNovance SAS <licensing@enovance.com>
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License"); you may
   4 # not use this file except in compliance with the License. You may obtain
   5 # a copy of the License at
   6 #
   7 #      http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  11 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  12 # License for the specific language governing permissions and limitations
  13 # under the License.
  14
  15 from neutron.api.rpc.agentnotifiers import metering_rpc_agent_api
  16 from neutron.common import rpc as n_rpc
  17 from neutron.common import topics
  18 from neutron.db.metering import metering_db
  19 from neutron.db.metering import metering_rpc
  20
  21
  22 class MeteringPlugin(metering_db.MeteringDbMixin):
  23     """Implementation of the Neutron Metering Service Plugin."""
  24     supported_extension_aliases = ["metering"]
  25     path_prefix = "/metering"
  26
  27     def __init__(self):
  28         super(MeteringPlugin, self).__init__()
  29
  30         self.meter_rpc = metering_rpc_agent_api.MeteringAgentNotifyAPI()
  31         self.start_rpc_listeners()
  32
  33     def start_rpc_listeners(self):
  34         self.endpoints = [metering_rpc.MeteringRpcCallbacks(self)]
  35         self.conn = n_rpc.create_connection()
  36         self.conn.create_consumer(
  37             topics.METERING_PLUGIN, self.endpoints, fanout=False)
  38         return self.conn.consume_in_threads()
  39
  40     def create_metering_label(self, context, metering_label):
  41         label = super(MeteringPlugin, self).create_metering_label(
  42             context, metering_label)
  43
  44         data = self.get_sync_data_metering(context)
  45         self.meter_rpc.add_metering_label(context, data)
  46
  47         return label
  48
  49     def delete_metering_label(self, context, label_id):
  50         data = self.get_sync_data_metering(context, label_id)
  51         label = super(MeteringPlugin, self).delete_metering_label(
  52             context, label_id)
  53
  54         self.meter_rpc.remove_metering_label(context, data)
  55
  56         return label
  57
  58     def create_metering_label_rule(self, context, metering_label_rule):
  59         rule = super(MeteringPlugin, self).create_metering_label_rule(
  60             context, metering_label_rule)
  61
  62         data = self.get_sync_data_for_rule(context, rule)
  63         self.meter_rpc.add_metering_label_rule(context, data)
  64
  65         return rule
  66
  67     def delete_metering_label_rule(self, context, rule_id):
  68         rule = super(MeteringPlugin, self).delete_metering_label_rule(
  69             context, rule_id)
  70
  71         data = self.get_sync_data_for_rule(context, rule)
  72         self.meter_rpc.remove_metering_label_rule(context, data)
  73         return rule