Bump MRI upstreams
[openflowplugin.git] / openflowplugin-impl / src / main / java / org / opendaylight / openflowplugin / impl / statistics / services / direct / multilayer / FlowDirectStatisticsService.java
1 /*
2  * Copyright (c) 2017 Pantheon Technologies s.r.o. 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 package org.opendaylight.openflowplugin.impl.statistics.services.direct.multilayer;
9
10 import java.util.ArrayList;
11 import java.util.List;
12 import java.util.Optional;
13 import org.opendaylight.openflowplugin.api.OFConstants;
14 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
15 import org.opendaylight.openflowplugin.api.openflow.device.RequestContextStack;
16 import org.opendaylight.openflowplugin.api.openflow.device.Xid;
17 import org.opendaylight.openflowplugin.extension.api.path.MatchPath;
18 import org.opendaylight.openflowplugin.impl.datastore.MultipartWriterProvider;
19 import org.opendaylight.openflowplugin.impl.services.util.RequestInputUtils;
20 import org.opendaylight.openflowplugin.impl.statistics.services.direct.AbstractFlowDirectStatisticsService;
21 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorExecutor;
22 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.data.FlowStatsResponseConvertorData;
23 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.match.MatchInjector;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.direct.statistics.rev160511.GetFlowStatisticsInput;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.direct.statistics.rev160511.GetFlowStatisticsOutput;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.direct.statistics.rev160511.GetFlowStatisticsOutputBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.FlowId;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.flow.and.statistics.map.list.FlowAndStatisticsMapList;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.flow.and.statistics.map.list.FlowAndStatisticsMapListBuilder;
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.OfHeader;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyFlowCase;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.flow._case.MultipartReplyFlow;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.request.multipart.request.body.MultipartRequestFlowCaseBuilder;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.request.multipart.request.body.multipart.request.flow._case.MultipartRequestFlowBuilder;
36 import org.opendaylight.yangtools.yang.common.Uint32;
37
38 public final class FlowDirectStatisticsService extends AbstractFlowDirectStatisticsService<MultipartReply> {
39     private final FlowStatsResponseConvertorData data;
40
41     public FlowDirectStatisticsService(final RequestContextStack requestContextStack,
42                                        final DeviceContext deviceContext,
43                                        final ConvertorExecutor convertorExecutor,
44                                        final MultipartWriterProvider statisticsWriterProvider) {
45         super(requestContextStack, deviceContext, convertorExecutor, statisticsWriterProvider);
46         data = new FlowStatsResponseConvertorData(getVersion());
47         data.setDatapathId(getDatapathId());
48         data.setMatchPath(MatchPath.FLOWS_STATISTICS_RPC_MATCH);
49     }
50
51     @Override
52     protected GetFlowStatisticsOutput buildReply(final List<MultipartReply> input, final boolean success) {
53         final List<FlowAndStatisticsMapList> statsList = new ArrayList<>();
54
55         if (success) {
56             for (final MultipartReply mpReply : input) {
57                 final MultipartReplyFlowCase caseBody = (MultipartReplyFlowCase) mpReply.getMultipartReplyBody();
58                 final MultipartReplyFlow replyBody = caseBody.getMultipartReplyFlow();
59                 final Optional<List<FlowAndStatisticsMapList>> statsListPart = getConvertorExecutor().convert(
60                     replyBody.getFlowStats(), data);
61
62                 statsListPart.ifPresent(flowAndStatisticsMapLists -> {
63                     for (final FlowAndStatisticsMapList part : flowAndStatisticsMapLists) {
64                         final FlowId flowId = new FlowId(generateFlowId(part).getValue());
65                         statsList.add(new FlowAndStatisticsMapListBuilder(part)
66                                 .setFlowId(flowId)
67                                 .build());
68                     }
69                 });
70             }
71         }
72
73         return new GetFlowStatisticsOutputBuilder()
74             .setFlowAndStatisticsMapList(statsList)
75             .build();
76     }
77
78     @Override
79     protected OfHeader buildRequest(final Xid xid, final GetFlowStatisticsInput input) {
80         final MultipartRequestFlowBuilder mprFlowRequestBuilder = new MultipartRequestFlowBuilder();
81
82         if (input.getTableId() != null) {
83             mprFlowRequestBuilder.setTableId(input.getTableId());
84         } else {
85             mprFlowRequestBuilder.setTableId(OFConstants.OFPTT_ALL);
86         }
87
88         if (input.getOutPort() != null) {
89             mprFlowRequestBuilder.setOutPort(Uint32.valueOf(input.getOutPort().longValue()));
90         } else {
91             mprFlowRequestBuilder.setOutPort(OFConstants.OFPP_ANY);
92         }
93
94         if (input.getOutGroup() != null) {
95             mprFlowRequestBuilder.setOutGroup(input.getOutGroup());
96         } else {
97             mprFlowRequestBuilder.setOutGroup(OFConstants.OFPG_ANY);
98         }
99
100         if (input.getCookie() != null) {
101             mprFlowRequestBuilder.setCookie(input.getCookie().getValue());
102         } else {
103             mprFlowRequestBuilder.setCookie(OFConstants.DEFAULT_COOKIE);
104         }
105
106         if (input.getCookieMask() != null) {
107             mprFlowRequestBuilder.setCookieMask(input.getCookieMask().getValue());
108         } else {
109             mprFlowRequestBuilder.setCookieMask(OFConstants.DEFAULT_COOKIE_MASK);
110         }
111
112         // convert and inject match
113         final Optional<Object> conversionMatch = getConvertorExecutor().convert(input.getMatch(), data);
114         MatchInjector.inject(conversionMatch, mprFlowRequestBuilder, data.getVersion());
115
116         return RequestInputUtils.createMultipartHeader(getMultipartType(), xid.getValue(), getVersion())
117             .setMultipartRequestBody(new MultipartRequestFlowCaseBuilder()
118                 .setMultipartRequestFlow(mprFlowRequestBuilder.build())
119                 .build())
120             .build();
121     }
122 }