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