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