openflowplugin-impl/src/main/java/org/opendaylight/openflowplugin/impl/services/multilayer/MultiLayerAggregateFlowMultipartService.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.services.multilayer;
  10
  11 import com.google.common.base.Function;
  12 import com.google.common.base.MoreObjects;
  13 import com.google.common.base.Preconditions;
  14 import com.google.common.util.concurrent.Futures;
  15 import com.google.common.util.concurrent.ListenableFuture;
  16 import com.google.common.util.concurrent.MoreExecutors;
  17 import java.util.List;
  18 import java.util.Optional;
  19 import java.util.stream.Collectors;
  20 import org.opendaylight.openflowplugin.api.OFConstants;
  21 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
  22 import org.opendaylight.openflowplugin.api.openflow.device.MessageTranslator;
  23 import org.opendaylight.openflowplugin.api.openflow.device.RequestContextStack;
  24 import org.opendaylight.openflowplugin.api.openflow.device.TranslatorLibrary;
  25 import org.opendaylight.openflowplugin.api.openflow.device.Xid;
  26 import org.opendaylight.openflowplugin.api.openflow.md.core.TranslatorKey;
  27 import org.opendaylight.openflowplugin.impl.services.AbstractAggregateFlowMultipartService;
  28 import org.opendaylight.openflowplugin.impl.services.util.RequestInputUtils;
  29 import org.opendaylight.openflowplugin.impl.services.util.ServiceException;
  30 import org.opendaylight.openflowplugin.impl.util.FlowCreatorUtil;
  31 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorExecutor;
  32 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.data.VersionConvertorData;
  33 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.match.MatchInjector;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAggregateFlowStatisticsFromFlowTableForGivenMatchInput;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAggregateFlowStatisticsFromFlowTableForGivenMatchOutput;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAggregateFlowStatisticsFromFlowTableForGivenMatchOutputBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.get.aggregate.flow.statistics.from.flow.table._for.given.match.output.AggregatedFlowStatistics;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MultipartType;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReply;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartRequestInputBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.OfHeader;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyAggregateCase;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.request.multipart.request.body.MultipartRequestAggregateCaseBuilder;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.request.multipart.request.body.multipart.request.aggregate._case.MultipartRequestAggregateBuilder;
  45 import org.opendaylight.yangtools.yang.common.RpcResult;
  46 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
  47
  48 public class MultiLayerAggregateFlowMultipartService extends AbstractAggregateFlowMultipartService<MultipartReply> {
  49
  50     private final TranslatorLibrary translatorLibrary;
  51     private final ConvertorExecutor convertorExecutor;
  52     private final VersionConvertorData data;
  53
  54     public MultiLayerAggregateFlowMultipartService(final RequestContextStack requestContextStack,
  55                                                    final DeviceContext deviceContext,
  56                                                    final ConvertorExecutor convertorExecutor,
  57                                                    final TranslatorLibrary translatorLibrary) {
  58         super(requestContextStack, deviceContext);
  59         this.convertorExecutor = convertorExecutor;
  60         this.translatorLibrary = translatorLibrary;
  61         this.data = new VersionConvertorData(getVersion());
  62     }
  63
  64
  65     @Override
  66     protected OfHeader buildRequest(final Xid xid,
  67                                     final GetAggregateFlowStatisticsFromFlowTableForGivenMatchInput input) throws
  68             ServiceException {
  69         final MultipartRequestAggregateCaseBuilder multipartRequestAggregateCaseBuilder
  70                 = new MultipartRequestAggregateCaseBuilder();
  71         final MultipartRequestAggregateBuilder mprAggregateRequestBuilder = new MultipartRequestAggregateBuilder();
  72         final short tableId = MoreObjects.firstNonNull(input.getTableId(), OFConstants.OFPTT_ALL);
  73         mprAggregateRequestBuilder.setTableId(tableId);
  74         long outputPortValue = MoreObjects.firstNonNull(input.getOutPort(), OFConstants.OFPP_ANY).longValue();
  75         mprAggregateRequestBuilder.setOutPort(outputPortValue);
  76
  77         final short version = getVersion();
  78         if (version == OFConstants.OFP_VERSION_1_3) {
  79
  80             if (input.getCookie() == null) {
  81                 mprAggregateRequestBuilder.setCookie(OFConstants.DEFAULT_COOKIE);
  82             } else {
  83                 mprAggregateRequestBuilder
  84                         .setCookie(MoreObjects.firstNonNull(input.getCookie().getValue(), OFConstants.DEFAULT_COOKIE));
  85             }
  86
  87             if (input.getCookieMask() == null) {
  88                 mprAggregateRequestBuilder.setCookieMask(OFConstants.DEFAULT_COOKIE_MASK);
  89             } else {
  90                 mprAggregateRequestBuilder.setCookieMask(
  91                         MoreObjects.firstNonNull(input.getCookieMask().getValue(), OFConstants.DEFAULT_COOKIE_MASK));
  92             }
  93             long outGroup = MoreObjects.firstNonNull(input.getOutGroup(), OFConstants.OFPG_ANY);
  94             mprAggregateRequestBuilder.setOutGroup(outGroup);
  95         } else {
  96             mprAggregateRequestBuilder.setOutGroup(OFConstants.OFPG_ANY);
  97             mprAggregateRequestBuilder.setCookie(OFConstants.DEFAULT_COOKIE);
  98             mprAggregateRequestBuilder.setCookieMask(OFConstants.DEFAULT_COOKIE_MASK);
  99         }
 100
 101         // convert and inject match
 102         final Optional<Object> conversionMatch = convertorExecutor.convert(input.getMatch(), data);
 103         MatchInjector.inject(conversionMatch, mprAggregateRequestBuilder, data.getVersion());
 104
 105         FlowCreatorUtil.setWildcardedFlowMatch(version, mprAggregateRequestBuilder);
 106
 107         // Set request body to main multipart request
 108         multipartRequestAggregateCaseBuilder.setMultipartRequestAggregate(mprAggregateRequestBuilder.build());
 109
 110         final MultipartRequestInputBuilder mprInput = RequestInputUtils
 111                 .createMultipartHeader(MultipartType.OFPMPAGGREGATE, xid.getValue(), version);
 112
 113         mprInput.setMultipartRequestBody(multipartRequestAggregateCaseBuilder.build());
 114
 115         return mprInput.build();
 116     }
 117
 118     @Override
 119     public ListenableFuture<RpcResult<GetAggregateFlowStatisticsFromFlowTableForGivenMatchOutput>> handleAndReply(
 120             final GetAggregateFlowStatisticsFromFlowTableForGivenMatchInput input) {
 121         return Futures.transform(handleServiceCall(input),
 122                      (Function<RpcResult<List<MultipartReply>>,
 123                              RpcResult<GetAggregateFlowStatisticsFromFlowTableForGivenMatchOutput>>)
 124             result -> {
 125                 if (Preconditions.checkNotNull(result).isSuccessful()) {
 126                     final MessageTranslator<MultipartReply, AggregatedFlowStatistics>
 127                              messageTranslator = translatorLibrary.lookupTranslator(
 128                              new TranslatorKey(getVersion(),
 129                                                MultipartReplyAggregateCase.class.getName()));
 130
 131                     return RpcResultBuilder.success(
 132                              new GetAggregateFlowStatisticsFromFlowTableForGivenMatchOutputBuilder()
 133                                      .setAggregatedFlowStatistics(result.getResult().stream()
 134                                                                           .map(multipartReply ->
 135                                                                                        messageTranslator
 136                                                                                   .translate(
 137                                                                                           multipartReply,
 138                                                                                           getDeviceInfo(),
 139                                                                                           null))
 140                                                                           .collect(Collectors
 141                                                                                            .toList())))
 142                              .build();
 143                 }
 144
 145                 return RpcResultBuilder
 146                              .<GetAggregateFlowStatisticsFromFlowTableForGivenMatchOutput>failed()
 147                              .withRpcErrors(result.getErrors()).build();
 148             }, MoreExecutors.directExecutor());
 149     }
 150 }