dd0d136c6504c619cf8870e0b69695b7812d320e
[openflowplugin.git] / openflowplugin-impl / src / main / java / org / opendaylight / openflowplugin / impl / statistics / services / direct / QueueDirectStatisticsService.java
1 /*
2  * Copyright (c) 2015 Cisco Systems, Inc. and others.  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
9 package org.opendaylight.openflowplugin.impl.statistics.services.direct;
10
11 import java.util.ArrayList;
12 import java.util.List;
13 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
14 import org.opendaylight.openflowplugin.api.OFConstants;
15 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
16 import org.opendaylight.openflowplugin.api.openflow.device.RequestContextStack;
17 import org.opendaylight.openflowplugin.openflow.md.util.InventoryDataServiceUtil;
18 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Counter32;
19 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Counter64;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.direct.statistics.rev160511.GetQueueStatisticsInput;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.direct.statistics.rev160511.GetQueueStatisticsOutput;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.direct.statistics.rev160511.GetQueueStatisticsOutputBuilder;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNodeConnector;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.port.rev130925.queues.Queue;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.port.rev130925.queues.QueueBuilder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.port.rev130925.queues.QueueKey;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.queue.rev130925.QueueId;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnector;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorKey;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.model.statistics.types.rev130925.duration.DurationBuilder;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MultipartType;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReply;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyQueueCase;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.queue._case.MultipartReplyQueue;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.queue._case.multipart.reply.queue.QueueStats;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.request.MultipartRequestBody;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.request.multipart.request.body.MultipartRequestQueueCaseBuilder;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.request.multipart.request.body.multipart.request.queue._case.MultipartRequestQueueBuilder;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.queue.statistics.rev131216.FlowCapableNodeConnectorQueueStatisticsData;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.queue.statistics.rev131216.FlowCapableNodeConnectorQueueStatisticsDataBuilder;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.queue.statistics.rev131216.flow.capable.node.connector.queue.statistics.FlowCapableNodeConnectorQueueStatistics;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.queue.statistics.rev131216.flow.capable.node.connector.queue.statistics.FlowCapableNodeConnectorQueueStatisticsBuilder;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.queue.statistics.rev131216.queue.id.and.statistics.map.QueueIdAndStatisticsMap;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.queue.statistics.rev131216.queue.id.and.statistics.map.QueueIdAndStatisticsMapBuilder;
46 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
47
48 /**
49  * The Queue direct statistics service.
50  */
51 public class QueueDirectStatisticsService extends AbstractDirectStatisticsService<GetQueueStatisticsInput, GetQueueStatisticsOutput> {
52     /**
53      * Instantiates a new Queue direct statistics service.
54      *
55      * @param requestContextStack the request context stack
56      * @param deviceContext       the device context
57      */
58     public QueueDirectStatisticsService(RequestContextStack requestContextStack, DeviceContext deviceContext) {
59         super(MultipartType.OFPMPQUEUE, requestContextStack, deviceContext);
60     }
61
62     @Override
63     protected MultipartRequestBody buildRequestBody(GetQueueStatisticsInput input) {
64         final MultipartRequestQueueBuilder mprQueueBuilder = new MultipartRequestQueueBuilder();
65
66         if (input.getQueueId() != null) {
67             mprQueueBuilder.setQueueId(input.getQueueId().getValue());
68         } else {
69             mprQueueBuilder.setQueueId(OFConstants.OFPQ_ALL);
70         }
71
72         if (input.getNodeConnectorId() != null) {
73             mprQueueBuilder.setPortNo(InventoryDataServiceUtil.portNumberfromNodeConnectorId(getOfVersion(), input.getNodeConnectorId()));
74         } else {
75             mprQueueBuilder.setPortNo(OFConstants.OFPP_ANY);
76         }
77
78         return new MultipartRequestQueueCaseBuilder()
79                 .setMultipartRequestQueue(mprQueueBuilder.build())
80                 .build();
81     }
82
83     @Override
84     protected GetQueueStatisticsOutput buildReply(List<MultipartReply> input, boolean success) {
85         final List<QueueIdAndStatisticsMap> queueIdAndStatisticsMap = new ArrayList<>();
86
87         if (success) {
88             for (final MultipartReply mpReply : input) {
89                 final MultipartReplyQueueCase caseBody = (MultipartReplyQueueCase) mpReply.getMultipartReplyBody();
90                 final MultipartReplyQueue replyBody = caseBody.getMultipartReplyQueue();
91
92                 for (final QueueStats queueStats : replyBody.getQueueStats()) {
93                     final DurationBuilder durationBuilder = new DurationBuilder()
94                             .setSecond(new Counter32(queueStats.getDurationSec()))
95                             .setNanosecond(new Counter32(queueStats.getDurationNsec()));
96
97                     final QueueIdAndStatisticsMapBuilder statsBuilder = new QueueIdAndStatisticsMapBuilder()
98                             .setNodeConnectorId(InventoryDataServiceUtil.nodeConnectorIdfromDatapathPortNo(
99                                     getDatapathId(), queueStats.getPortNo(), getOfVersion()))
100                             .setTransmissionErrors(new Counter64(queueStats.getTxErrors()))
101                             .setTransmittedBytes(new Counter64(queueStats.getTxBytes()))
102                             .setTransmittedPackets(new Counter64(queueStats.getTxPackets()))
103                             .setQueueId(new QueueId(queueStats.getQueueId()))
104                             .setDuration(durationBuilder.build());
105
106                     queueIdAndStatisticsMap.add(statsBuilder.build());
107                 }
108             }
109         }
110
111         return new GetQueueStatisticsOutputBuilder()
112                 .setQueueIdAndStatisticsMap(queueIdAndStatisticsMap)
113                 .build();
114     }
115
116     @Override
117     protected void storeStatistics(GetQueueStatisticsOutput output) throws Exception {
118         final InstanceIdentifier<Node> nodePath = getDeviceInfo().getNodeInstanceIdentifier();
119
120         for (final QueueIdAndStatisticsMap queueStatistics : output.getQueueIdAndStatisticsMap()) {
121             if (queueStatistics.getQueueId() != null) {
122                 final QueueKey qKey = new QueueKey(queueStatistics.getQueueId());
123
124                 final FlowCapableNodeConnectorQueueStatistics statChild =
125                         new FlowCapableNodeConnectorQueueStatisticsBuilder(queueStatistics).build();
126
127                 final FlowCapableNodeConnectorQueueStatisticsDataBuilder statBuild =
128                         new FlowCapableNodeConnectorQueueStatisticsDataBuilder()
129                                 .setFlowCapableNodeConnectorQueueStatistics(statChild);
130
131                 final InstanceIdentifier<Queue> queueStatisticsPath = nodePath
132                         .child(NodeConnector.class, new NodeConnectorKey(queueStatistics.getNodeConnectorId()))
133                         .augmentation(FlowCapableNodeConnector.class)
134                         .child(Queue.class, qKey);
135
136                 final Queue stats = new QueueBuilder()
137                         .setKey(qKey)
138                         .setQueueId(queueStatistics.getQueueId())
139                         .addAugmentation(FlowCapableNodeConnectorQueueStatisticsData.class, statBuild.build()).build();
140
141                 getTxFacade().writeToTransactionWithParentsSlow(LogicalDatastoreType.OPERATIONAL, queueStatisticsPath, stats);
142             }
143         }
144     }
145 }