Bug 5540 - MeterStatsResponseConvertor, MeterConfigStatsResponseConvertor
[openflowplugin.git] / openflowplugin-impl / src / main / java / org / opendaylight / openflowplugin / impl / statistics / services / direct / MeterDirectStatisticsService.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 java.util.Optional;
14 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
15 import org.opendaylight.openflowplugin.api.OFConstants;
16 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
17 import org.opendaylight.openflowplugin.api.openflow.device.RequestContextStack;
18 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorManager;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.direct.statistics.rev160511.GetMeterStatisticsInput;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.direct.statistics.rev160511.GetMeterStatisticsOutput;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.direct.statistics.rev160511.GetMeterStatisticsOutputBuilder;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.Meter;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.MeterKey;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.NodeMeterStatistics;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.nodes.node.meter.MeterStatistics;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.nodes.node.meter.MeterStatisticsBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.meter.statistics.reply.MeterStats;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MeterId;
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.MultipartReplyMeterCase;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.meter._case.MultipartReplyMeter;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.request.MultipartRequestBody;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.request.multipart.request.body.MultipartRequestMeterCaseBuilder;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.request.multipart.request.body.multipart.request.meter._case.MultipartRequestMeterBuilder;
37 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
38
39 /**
40  * The Meter direct statistics service.
41  */
42 public class MeterDirectStatisticsService extends AbstractDirectStatisticsService<GetMeterStatisticsInput, GetMeterStatisticsOutput> {
43     /**
44      * Instantiates a new Meter direct statistics service.
45      *
46      * @param requestContextStack the request context stack
47      * @param deviceContext       the device context
48      */
49     public MeterDirectStatisticsService(RequestContextStack requestContextStack, DeviceContext deviceContext) {
50         super(MultipartType.OFPMPMETER, requestContextStack, deviceContext);
51     }
52
53     @Override
54     protected MultipartRequestBody buildRequestBody(GetMeterStatisticsInput input) {
55         final MultipartRequestMeterBuilder mprMeterBuild = new MultipartRequestMeterBuilder();
56
57         if (input.getMeterId() != null) {
58             mprMeterBuild.setMeterId(new MeterId(input.getMeterId().getValue()));
59         } else {
60             mprMeterBuild.setMeterId(new MeterId(OFConstants.OFPM_ALL));
61         }
62
63         return new MultipartRequestMeterCaseBuilder()
64                 .setMultipartRequestMeter(mprMeterBuild.build())
65                 .build();
66     }
67
68     @Override
69     protected GetMeterStatisticsOutput buildReply(List<MultipartReply> input, boolean success) {
70         final List<MeterStats> meterStats = new ArrayList<>();
71
72         if (success) {
73             for (final MultipartReply mpReply : input) {
74                 final MultipartReplyMeterCase caseBody = (MultipartReplyMeterCase) mpReply.getMultipartReplyBody();
75                 final MultipartReplyMeter replyBody = caseBody.getMultipartReplyMeter();
76                 final Optional<List<MeterStats>> meterStatsList = ConvertorManager.getInstance().convert(replyBody.getMeterStats());
77
78                 if (meterStatsList.isPresent()) {
79                     meterStats.addAll(meterStatsList.get());
80                 }
81             }
82         }
83
84         return new GetMeterStatisticsOutputBuilder()
85                 .setMeterStats(meterStats)
86                 .build();
87     }
88
89     @Override
90     protected void storeStatistics(GetMeterStatisticsOutput output) throws Exception {
91         final InstanceIdentifier<FlowCapableNode> nodePath = getDeviceInfo().getNodeInstanceIdentifier().augmentation(FlowCapableNode.class);
92
93         for (final MeterStats meterStatistics : output.getMeterStats()) {
94             final InstanceIdentifier<MeterStatistics> meterPath = nodePath
95                     .child(Meter.class, new MeterKey(meterStatistics.getMeterId()))
96                     .augmentation(NodeMeterStatistics.class)
97                     .child(MeterStatistics.class);
98
99             final MeterStatistics stats = new MeterStatisticsBuilder(meterStatistics).build();
100             getTxFacade().writeToTransactionWithParentsSlow(LogicalDatastoreType.OPERATIONAL, meterPath, stats);
101         }
102     }
103 }