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