opendaylight/md-sal/statistics-manager/src/main/java/org/opendaylight/controller/md/statistics/manager/MeterConfigStatsTracker.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.GetAllMeterConfigStatisticsInputBuilder;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.NodeMeterConfigStats;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.NodeMeterConfigStatsBuilder;
  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.MeterConfigStatsBuilder;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.meter.config.stats.reply.MeterConfigStats;
  23 import org.opendaylight.yangtools.yang.binding.DataObject;
  24 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  25 import org.slf4j.Logger;
  26 import org.slf4j.LoggerFactory;
  27
  28 final class MeterConfigStatsTracker extends AbstractListeningStatsTracker<MeterConfigStats, MeterConfigStats> {
  29     private static final Logger logger = LoggerFactory.getLogger(MeterConfigStatsTracker.class);
  30     private final OpendaylightMeterStatisticsService meterStatsService;
  31
  32     protected MeterConfigStatsTracker(OpendaylightMeterStatisticsService meterStatsService, final FlowCapableContext context) {
  33         super(context);
  34         this.meterStatsService = meterStatsService;
  35     }
  36
  37     @Override
  38     protected void cleanupSingleStat(DataModificationTransaction trans, MeterConfigStats item) {
  39         InstanceIdentifier<NodeMeterConfigStats> meterRef = getNodeIdentifierBuilder()
  40                             .augmentation(FlowCapableNode.class)
  41                             .child(Meter.class, new MeterKey(item.getMeterId()))
  42                             .augmentation(NodeMeterConfigStats.class).build();
  43         trans.removeOperationalData(meterRef);
  44     }
  45
  46     @Override
  47     protected MeterConfigStats updateSingleStat(DataModificationTransaction trans, MeterConfigStats item) {
  48         MeterBuilder meterBuilder = new MeterBuilder();
  49         MeterKey meterKey = new MeterKey(item.getMeterId());
  50         meterBuilder.setKey(meterKey);
  51
  52         InstanceIdentifier<Meter> meterRef = getNodeIdentifierBuilder().augmentation(FlowCapableNode.class)
  53                 .child(Meter.class,meterKey).toInstance();
  54
  55         NodeMeterConfigStatsBuilder meterConfig = new NodeMeterConfigStatsBuilder();
  56         meterConfig.setMeterConfigStats(new MeterConfigStatsBuilder(item).build());
  57
  58         //Update augmented data
  59         meterBuilder.addAugmentation(NodeMeterConfigStats.class, meterConfig.build());
  60
  61         trans.putOperationalData(meterRef, meterBuilder.build());
  62         return item;
  63     }
  64
  65     @Override
  66     public void request() {
  67         if (meterStatsService != null) {
  68             GetAllMeterConfigStatisticsInputBuilder input = new GetAllMeterConfigStatisticsInputBuilder();
  69             input.setNode(getNodeRef());
  70
  71             requestHelper(meterStatsService.getAllMeterConfigStatistics(input.build()));
  72         }
  73     }
  74
  75     @Override
  76     public void onDataChanged(DataChangeEvent<InstanceIdentifier<?>, DataObject> change) {
  77         final DataModificationTransaction trans = startTransaction();
  78
  79         for (InstanceIdentifier<?> key : change.getRemovedConfigurationData()) {
  80             if (Meter.class.equals(key.getTargetType())) {
  81                 @SuppressWarnings("unchecked")
  82                 InstanceIdentifier<Meter> meter = (InstanceIdentifier<Meter>)key;
  83
  84                 InstanceIdentifier<?> nodeMeterStatisticsAugmentation =
  85                         InstanceIdentifier.builder(meter).augmentation(NodeMeterConfigStats.class).toInstance();
  86                 trans.removeOperationalData(nodeMeterStatisticsAugmentation);
  87             }
  88         }
  89
  90         trans.commit();
  91     }
  92
  93     @Override
  94     protected InstanceIdentifier<?> listenPath() {
  95         return getNodeIdentifierBuilder().augmentation(FlowCapableNode.class).child(Meter.class).build();
  96     }
  97
  98     @Override
  99     protected String statName() {
 100         return "Meter Config";
 101     }
 102
 103     @Override
 104     public void start(final DataBrokerService dbs) {
 105         if (meterStatsService == null) {
 106             logger.debug("No Meter Statistics service, not subscribing to meter on node {}", getNodeIdentifier());
 107             return;
 108         }
 109
 110         super.start(dbs);
 111     }
 112 }