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