Move queue/meter/flow listeners into their trackers
[controller.git] / 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.md.statistics.manager.MultipartMessageManager.StatsRequestType;
12 import org.opendaylight.controller.sal.binding.api.data.DataBrokerService;
13 import org.opendaylight.controller.sal.binding.api.data.DataModificationTransaction;
14 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
15 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.Meter;
16 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.MeterBuilder;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.MeterKey;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.GetAllMeterConfigStatisticsInputBuilder;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.NodeMeterConfigStats;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.NodeMeterConfigStatsBuilder;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.OpendaylightMeterStatisticsService;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.nodes.node.meter.MeterConfigStatsBuilder;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.meter.config.stats.reply.MeterConfigStats;
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 MeterConfigStatsTracker extends AbstractListeningStatsTracker<MeterConfigStats, MeterConfigStats> {
30     private static final Logger logger = LoggerFactory.getLogger(MeterConfigStatsTracker.class);
31     private final OpendaylightMeterStatisticsService meterStatsService;
32
33     protected MeterConfigStatsTracker(OpendaylightMeterStatisticsService meterStatsService, final FlowCapableContext context, long lifetimeNanos) {
34         super(context, lifetimeNanos);
35         this.meterStatsService = meterStatsService;
36     }
37
38     @Override
39     protected void cleanupSingleStat(DataModificationTransaction trans, MeterConfigStats item) {
40         InstanceIdentifier<NodeMeterConfigStats> meterRef = getNodeIdentifierBuilder()
41                             .augmentation(FlowCapableNode.class)
42                             .child(Meter.class, new MeterKey(item.getMeterId()))
43                             .augmentation(NodeMeterConfigStats.class).build();
44         trans.removeOperationalData(meterRef);
45     }
46
47     @Override
48     protected MeterConfigStats updateSingleStat(DataModificationTransaction trans, MeterConfigStats item) {
49         MeterBuilder meterBuilder = new MeterBuilder();
50         MeterKey meterKey = new MeterKey(item.getMeterId());
51         meterBuilder.setKey(meterKey);
52
53         InstanceIdentifier<Meter> meterRef = getNodeIdentifierBuilder().augmentation(FlowCapableNode.class)
54                 .child(Meter.class,meterKey).toInstance();
55
56         NodeMeterConfigStatsBuilder meterConfig = new NodeMeterConfigStatsBuilder();
57         meterConfig.setMeterConfigStats(new MeterConfigStatsBuilder(item).build());
58
59         //Update augmented data
60         meterBuilder.addAugmentation(NodeMeterConfigStats.class, meterConfig.build());
61
62         trans.putOperationalData(meterRef, meterBuilder.build());
63         return item;
64     }
65
66     public void request() {
67         if (meterStatsService != null) {
68             GetAllMeterConfigStatisticsInputBuilder input = new GetAllMeterConfigStatisticsInputBuilder();
69             input.setNode(getNodeRef());
70
71             requestHelper(meterStatsService.getAllMeterConfigStatistics(input.build()), StatsRequestType.METER_CONFIG);
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 }