f506ddfe8f73af429a36ba133f6af8014a5abb3b
[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 class FlowDirectStatisticsService extends AbstractFlowDirectStatisticsService<MultipartReply> {
39
40     private final FlowStatsResponseConvertorData data;
41
42     public FlowDirectStatisticsService(final RequestContextStack requestContextStack,
43                                        final DeviceContext deviceContext,
44                                        final ConvertorExecutor convertorExecutor,
45                                        final MultipartWriterProvider statisticsWriterProvider) {
46         super(requestContextStack, deviceContext, convertorExecutor, statisticsWriterProvider);
47         data = new FlowStatsResponseConvertorData(getVersion());
48         data.setDatapathId(getDatapathId());
49         data.setMatchPath(MatchPath.FLOWS_STATISTICS_RPC_MATCH);
50     }
51
52     @Override
53     protected GetFlowStatisticsOutput buildReply(final List<MultipartReply> input, final boolean success) {
54         final List<FlowAndStatisticsMapList> statsList = new ArrayList<>();
55
56         if (success) {
57             for (final MultipartReply mpReply : input) {
58                 final MultipartReplyFlowCase caseBody = (MultipartReplyFlowCase) mpReply.getMultipartReplyBody();
59                 final MultipartReplyFlow replyBody = caseBody.getMultipartReplyFlow();
60                 final Optional<List<FlowAndStatisticsMapList>> statsListPart = getConvertorExecutor().convert(
61                     replyBody.getFlowStats(), data);
62
63                 statsListPart.ifPresent(flowAndStatisticsMapLists -> {
64                     for (final FlowAndStatisticsMapList part : flowAndStatisticsMapLists) {
65                         final FlowId flowId = new FlowId(generateFlowId(part).getValue());
66                         statsList.add(new FlowAndStatisticsMapListBuilder(part)
67                                 .setFlowId(flowId)
68                                 .build());
69                     }
70                 });
71             }
72         }
73
74         return new GetFlowStatisticsOutputBuilder()
75             .setFlowAndStatisticsMapList(statsList)
76             .build();
77     }
78
79     @Override
80     protected OfHeader buildRequest(final Xid xid, final GetFlowStatisticsInput input) {
81         final MultipartRequestFlowBuilder mprFlowRequestBuilder = new MultipartRequestFlowBuilder();
82
83         if (input.getTableId() != null) {
84             mprFlowRequestBuilder.setTableId(input.getTableId());
85         } else {
86             mprFlowRequestBuilder.setTableId(OFConstants.OFPTT_ALL);
87         }
88
89         if (input.getOutPort() != null) {
90             mprFlowRequestBuilder.setOutPort(Uint32.valueOf(input.getOutPort().longValue()));
91         } else {
92             mprFlowRequestBuilder.setOutPort(OFConstants.OFPP_ANY);
93         }
94
95         if (input.getOutGroup() != null) {
96             mprFlowRequestBuilder.setOutGroup(input.getOutGroup());
97         } else {
98             mprFlowRequestBuilder.setOutGroup(OFConstants.OFPG_ANY);
99         }
100
101         if (input.getCookie() != null) {
102             mprFlowRequestBuilder.setCookie(input.getCookie().getValue());
103         } else {
104             mprFlowRequestBuilder.setCookie(OFConstants.DEFAULT_COOKIE);
105         }
106
107         if (input.getCookieMask() != null) {
108             mprFlowRequestBuilder.setCookieMask(input.getCookieMask().getValue());
109         } else {
110             mprFlowRequestBuilder.setCookieMask(OFConstants.DEFAULT_COOKIE_MASK);
111         }
112
113         // convert and inject match
114         final Optional<Object> conversionMatch = getConvertorExecutor().convert(input.getMatch(), data);
115         MatchInjector.inject(conversionMatch, mprFlowRequestBuilder, data.getVersion());
116
117         return RequestInputUtils.createMultipartHeader(getMultipartType(), xid.getValue(), getVersion())
118             .setMultipartRequestBody(new MultipartRequestFlowCaseBuilder()
119                 .setMultipartRequestFlow(mprFlowRequestBuilder.build())
120                 .build())
121             .build();
122     }
123 }