Split out statistics tracking into separate classes
[controller.git] / 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.sal.binding.api.data.DataModificationTransaction;
11 import org.opendaylight.controller.sal.binding.api.data.DataProviderService;
12 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
13 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.Meter;
14 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.MeterBuilder;
15 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.MeterKey;
16 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.NodeMeterStatistics;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.NodeMeterStatisticsBuilder;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.nodes.node.meter.MeterStatisticsBuilder;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.meter.statistics.reply.MeterStats;
21 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
22
23 final class MeterStatsTracker extends AbstractStatsTracker<MeterStats, MeterStats> {
24
25     MeterStatsTracker(InstanceIdentifier<Node> nodeIdentifier, DataProviderService dps, long lifetimeNanos) {
26         super(nodeIdentifier, dps, lifetimeNanos);
27     }
28
29     @Override
30     protected void cleanupSingleStat(DataModificationTransaction trans, MeterStats item) {
31         InstanceIdentifier<NodeMeterStatistics> meterRef = getNodeIdentifierBuilder()
32                             .augmentation(FlowCapableNode.class)
33                             .child(Meter.class,new MeterKey(item.getMeterId()))
34                             .augmentation(NodeMeterStatistics.class).build();
35         trans.removeOperationalData(meterRef);
36     }
37
38     @Override
39     protected MeterStats updateSingleStat(DataModificationTransaction trans, MeterStats item) {
40         MeterBuilder meterBuilder = new MeterBuilder();
41         MeterKey meterKey = new MeterKey(item.getMeterId());
42         meterBuilder.setKey(meterKey);
43
44         InstanceIdentifier<Meter> meterRef = getNodeIdentifierBuilder()
45                 .augmentation(FlowCapableNode.class).child(Meter.class,meterKey).build();
46
47         NodeMeterStatisticsBuilder meterStatsBuilder= new NodeMeterStatisticsBuilder();
48         meterStatsBuilder.setMeterStatistics(new MeterStatisticsBuilder(item).build());
49
50         //Update augmented data
51         meterBuilder.addAugmentation(NodeMeterStatistics.class, meterStatsBuilder.build());
52         trans.putOperationalData(meterRef, meterBuilder.build());
53         return item;
54     }
55 }