Move statistics request functions into trackers
[controller.git] / opendaylight / md-sal / statistics-manager / src / main / java / org / opendaylight / controller / md / statistics / manager / NodeConnectorStatsTracker.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.yang.gen.v1.urn.opendaylight.flow.transaction.rev131103.TransactionId;
12 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnector;
13 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorBuilder;
14 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorKey;
15 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.FlowCapableNodeConnectorStatisticsData;
16 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.FlowCapableNodeConnectorStatisticsDataBuilder;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.GetAllNodeConnectorsStatisticsInputBuilder;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.OpendaylightPortStatisticsService;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.flow.capable.node.connector.statistics.FlowCapableNodeConnectorStatisticsBuilder;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.node.connector.statistics.and.port.number.map.NodeConnectorStatisticsAndPortNumberMap;
21 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
22 import org.slf4j.Logger;
23 import org.slf4j.LoggerFactory;
24
25 import com.google.common.base.Preconditions;
26 import com.google.common.util.concurrent.ListenableFuture;
27
28 final class NodeConnectorStatsTracker extends AbstractStatsTracker<NodeConnectorStatisticsAndPortNumberMap, NodeConnectorStatisticsAndPortNumberMap> {
29     private static final Logger logger = LoggerFactory.getLogger(NodeConnectorStatsTracker.class);
30     private final OpendaylightPortStatisticsService portStatsService;
31
32     NodeConnectorStatsTracker(final OpendaylightPortStatisticsService portStatsService, final FlowCapableContext context, long lifetimeNanos) {
33         super(context, lifetimeNanos);
34         this.portStatsService = Preconditions.checkNotNull(portStatsService);
35     }
36
37     @Override
38     protected void cleanupSingleStat(DataModificationTransaction trans, NodeConnectorStatisticsAndPortNumberMap item) {
39         // TODO Auto-generated method stub
40     }
41
42     @Override
43     protected NodeConnectorStatisticsAndPortNumberMap updateSingleStat(DataModificationTransaction trans, NodeConnectorStatisticsAndPortNumberMap item) {
44         FlowCapableNodeConnectorStatisticsBuilder statisticsBuilder
45                                         = new FlowCapableNodeConnectorStatisticsBuilder();
46         statisticsBuilder.setBytes(item.getBytes());
47         statisticsBuilder.setCollisionCount(item.getCollisionCount());
48         statisticsBuilder.setDuration(item.getDuration());
49         statisticsBuilder.setPackets(item.getPackets());
50         statisticsBuilder.setReceiveCrcError(item.getReceiveCrcError());
51         statisticsBuilder.setReceiveDrops(item.getReceiveDrops());
52         statisticsBuilder.setReceiveErrors(item.getReceiveErrors());
53         statisticsBuilder.setReceiveFrameError(item.getReceiveFrameError());
54         statisticsBuilder.setReceiveOverRunError(item.getReceiveOverRunError());
55         statisticsBuilder.setTransmitDrops(item.getTransmitDrops());
56         statisticsBuilder.setTransmitErrors(item.getTransmitErrors());
57
58         //Augment data to the node-connector
59         FlowCapableNodeConnectorStatisticsDataBuilder statisticsDataBuilder =
60                 new FlowCapableNodeConnectorStatisticsDataBuilder();
61
62         statisticsDataBuilder.setFlowCapableNodeConnectorStatistics(statisticsBuilder.build());
63
64         InstanceIdentifier<NodeConnector> nodeConnectorRef = getNodeIdentifierBuilder()
65                 .child(NodeConnector.class, new NodeConnectorKey(item.getNodeConnectorId())).build();
66
67         // FIXME: can we bypass this read?
68         NodeConnector nodeConnector = (NodeConnector)trans.readOperationalData(nodeConnectorRef);
69         if(nodeConnector != null){
70             final FlowCapableNodeConnectorStatisticsData stats = statisticsDataBuilder.build();
71             logger.debug("Augmenting port statistics {} to port {}",stats,nodeConnectorRef.toString());
72             NodeConnectorBuilder nodeConnectorBuilder = new NodeConnectorBuilder();
73             nodeConnectorBuilder.addAugmentation(FlowCapableNodeConnectorStatisticsData.class, stats);
74             trans.putOperationalData(nodeConnectorRef, nodeConnectorBuilder.build());
75         }
76
77         return item;
78     }
79
80     public ListenableFuture<TransactionId> request() {
81         final GetAllNodeConnectorsStatisticsInputBuilder input = new GetAllNodeConnectorsStatisticsInputBuilder();
82         input.setNode(getNodeRef());
83
84         return requestHelper(portStatsService.getAllNodeConnectorsStatistics(input.build()));
85     }
86 }