opendaylight/md-sal/statistics-manager/src/main/java/org/opendaylight/controller/md/statistics/manager/MeterStatsTracker.java

   1 /*
   2  * Copyright IBM Corporation, 2013.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.controller.md.statistics.manager;
   9
  10 import org.opendaylight.controller.md.sal.common.api.data.DataChangeEvent;
  11 import org.opendaylight.controller.sal.binding.api.data.DataBrokerService;
  12 import org.opendaylight.controller.sal.binding.api.data.DataModificationTransaction;
  13 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
  14 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.Meter;
  15 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.MeterBuilder;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.MeterKey;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.GetAllMeterStatisticsInputBuilder;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.NodeMeterStatistics;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.NodeMeterStatisticsBuilder;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.OpendaylightMeterStatisticsService;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.nodes.node.meter.MeterStatisticsBuilder;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.meter.statistics.reply.MeterStats;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.meter.statistics.reply.MeterStatsBuilder;
  24 import org.opendaylight.yangtools.yang.binding.DataObject;
  25 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  26 import org.slf4j.Logger;
  27 import org.slf4j.LoggerFactory;
  28
  29 final class MeterStatsTracker extends AbstractListeningStatsTracker<MeterStats, MeterStats> {
  30     private static final Logger logger = LoggerFactory.getLogger(MeterStatsTracker.class);
  31     private final OpendaylightMeterStatisticsService meterStatsService;
  32
  33     MeterStatsTracker(OpendaylightMeterStatisticsService meterStatsService, final FlowCapableContext context) {
  34         super(context);
  35         this.meterStatsService = meterStatsService;
  36     }
  37
  38     @Override
  39     protected void cleanupSingleStat(DataModificationTransaction trans, MeterStats item) {
  40         InstanceIdentifier<NodeMeterStatistics> meterRef = getNodeIdentifierBuilder()
  41                             .augmentation(FlowCapableNode.class)
  42                             .child(Meter.class,new MeterKey(item.getMeterId()))
  43                             .augmentation(NodeMeterStatistics.class).build();
  44         trans.removeOperationalData(meterRef);
  45     }
  46
  47     @Override
  48     protected MeterStats updateSingleStat(DataModificationTransaction trans, MeterStats item) {
  49         MeterBuilder meterBuilder = new MeterBuilder();
  50         MeterKey meterKey = new MeterKey(item.getMeterId());
  51         meterBuilder.setKey(meterKey);
  52
  53         InstanceIdentifier<Meter> meterRef = getNodeIdentifierBuilder()
  54                 .augmentation(FlowCapableNode.class).child(Meter.class,meterKey).build();
  55
  56         NodeMeterStatisticsBuilder meterStatsBuilder= new NodeMeterStatisticsBuilder();
  57         meterStatsBuilder.setMeterStatistics(new MeterStatisticsBuilder(item).build());
  58
  59         //Update augmented data
  60         meterBuilder.addAugmentation(NodeMeterStatistics.class, meterStatsBuilder.build());
  61         trans.putOperationalData(meterRef, meterBuilder.build());
  62         return item;
  63     }
  64
  65     @Override
  66     public void request() {
  67         if (meterStatsService != null) {
  68             GetAllMeterStatisticsInputBuilder input = new GetAllMeterStatisticsInputBuilder();
  69             input.setNode(getNodeRef());
  70
  71             requestHelper(meterStatsService.getAllMeterStatistics(input.build()));
  72         }
  73     }
  74
  75     @Override
  76     public void onDataChanged(DataChangeEvent<InstanceIdentifier<?>, DataObject> change) {
  77         for (InstanceIdentifier<?> key : change.getCreatedConfigurationData().keySet()) {
  78             if (Meter.class.equals(key.getTargetType())) {
  79                 request();
  80             }
  81         }
  82
  83         final DataModificationTransaction trans = startTransaction();
  84         for (InstanceIdentifier<?> key : change.getRemovedConfigurationData()) {
  85             if (Meter.class.equals(key.getTargetType())) {
  86                 @SuppressWarnings("unchecked")
  87                 InstanceIdentifier<Meter> meter = (InstanceIdentifier<Meter>)key;
  88
  89                 InstanceIdentifier<?> nodeMeterStatisticsAugmentation =
  90                         meter.augmentation(NodeMeterStatistics.class);
  91                 trans.removeOperationalData(nodeMeterStatisticsAugmentation);
  92             }
  93         }
  94         trans.commit();
  95     }
  96
  97     @Override
  98     protected InstanceIdentifier<?> listenPath() {
  99         return getNodeIdentifierBuilder().augmentation(FlowCapableNode.class).child(Meter.class).build();
 100     }
 101
 102     @Override
 103     protected String statName() {
 104         return "Meter";
 105     }
 106
 107     @Override
 108     public void start(final DataBrokerService dbs) {
 109         if (meterStatsService == null) {
 110             logger.debug("No Meter Statistics service, not subscribing to meters on node {}", getNodeIdentifier());
 111             return;
 112         }
 113
 114         super.start(dbs);
 115     }
 116
 117     @Override
 118     protected MeterStats createInvariantKey(MeterStats item) {
 119         MeterStatsBuilder meterStatsBuilder = new MeterStatsBuilder();
 120         meterStatsBuilder.setKey(item.getKey());
 121         meterStatsBuilder.setMeterId(item.getMeterId());
 122         return meterStatsBuilder.build();
 123     }
 124 }