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