Decompose RPC implementation classes
[openflowplugin.git] / openflowplugin-impl / src / main / java / org / opendaylight / openflowplugin / impl / statistics / services / direct / multilayer / MultiGetGroupStatistics.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.List;
11 import java.util.Optional;
12 import org.opendaylight.openflowplugin.api.OFConstants;
13 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
14 import org.opendaylight.openflowplugin.api.openflow.device.RequestContextStack;
15 import org.opendaylight.openflowplugin.api.openflow.device.Xid;
16 import org.opendaylight.openflowplugin.impl.datastore.MultipartWriterProvider;
17 import org.opendaylight.openflowplugin.impl.services.util.RequestInputUtils;
18 import org.opendaylight.openflowplugin.impl.statistics.services.direct.AbstractGetGroupStatistics;
19 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorExecutor;
20 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.data.VersionConvertorData;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.direct.statistics.rev160511.GetGroupStatisticsInput;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.direct.statistics.rev160511.GetGroupStatisticsOutput;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.direct.statistics.rev160511.GetGroupStatisticsOutputBuilder;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.group.statistics.reply.GroupStats;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.group.statistics.reply.GroupStatsKey;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.GroupId;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReply;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.OfHeader;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyGroupCase;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.request.multipart.request.body.MultipartRequestGroupCaseBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.request.multipart.request.body.multipart.request.group._case.MultipartRequestGroupBuilder;
32 import org.opendaylight.yangtools.yang.binding.util.BindingMap;
33
34 public final class MultiGetGroupStatistics extends AbstractGetGroupStatistics<MultipartReply> {
35     private final VersionConvertorData data;
36
37     public MultiGetGroupStatistics(final RequestContextStack requestContextStack, final DeviceContext deviceContext,
38             final ConvertorExecutor convertorExecutor, final MultipartWriterProvider statisticsWriterProvider) {
39         super(requestContextStack, deviceContext, convertorExecutor, statisticsWriterProvider);
40         data = new VersionConvertorData(getVersion());
41     }
42
43     @Override
44     protected GetGroupStatisticsOutput buildReply(final List<MultipartReply> input, final boolean success) {
45         if (!success) {
46             return new GetGroupStatisticsOutputBuilder().build();
47         }
48
49         final var groupStats = BindingMap.<GroupStatsKey, GroupStats>orderedBuilder();
50         for (var mpReply : input) {
51             final var caseBody = (MultipartReplyGroupCase) mpReply.getMultipartReplyBody();
52             final var replyBody = caseBody.getMultipartReplyGroup();
53             final Optional<List<GroupStats>> groupStatsList = getConvertorExecutor().convert(
54                 replyBody.getGroupStats(), data);
55
56             groupStatsList.ifPresent(groupStats::addAll);
57         }
58
59         return new GetGroupStatisticsOutputBuilder()
60             .setGroupStats(groupStats.build())
61             .build();
62     }
63
64     @Override
65     protected OfHeader buildRequest(final Xid xid, final GetGroupStatisticsInput input) {
66         final var mprGroupBuild = new MultipartRequestGroupBuilder();
67
68         if (input.getGroupId() != null) {
69             mprGroupBuild.setGroupId(new GroupId(input.getGroupId().getValue()));
70         } else {
71             mprGroupBuild.setGroupId(new GroupId(OFConstants.OFPG_ALL));
72         }
73
74         return RequestInputUtils.createMultipartHeader(getMultipartType(), xid.getValue(), getVersion())
75             .setMultipartRequestBody(new MultipartRequestGroupCaseBuilder()
76                 .setMultipartRequestGroup(mprGroupBuild.build())
77                 .build())
78             .build();
79     }
80
81 }