openflowplugin-impl/src/main/java/org/opendaylight/openflowplugin/impl/statistics/services/AggregateFlowsInTableService.java

   1 /**
   2  * Copyright (c) 2015 Cisco Systems, Inc. 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;
   9
  10 import com.google.common.base.Preconditions;
  11 import java.util.List;
  12 import java.util.concurrent.atomic.AtomicLong;
  13 import org.opendaylight.openflowplugin.api.OFConstants;
  14 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
  15 import org.opendaylight.openflowplugin.api.openflow.device.MessageTranslator;
  16 import org.opendaylight.openflowplugin.api.openflow.device.RequestContextStack;
  17 import org.opendaylight.openflowplugin.api.openflow.device.TranslatorLibrary;
  18 import org.opendaylight.openflowplugin.api.openflow.device.Xid;
  19 import org.opendaylight.openflowplugin.api.openflow.md.core.TranslatorKey;
  20 import org.opendaylight.openflowplugin.impl.services.util.RequestInputUtils;
  21 import org.opendaylight.openflowplugin.impl.services.util.ServiceException;
  22 import org.opendaylight.openflowplugin.impl.statistics.services.compatibility.AbstractCompatibleStatService;
  23 import org.opendaylight.openflowplugin.openflow.md.util.FlowCreatorUtil;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.AggregateFlowStatisticsUpdate;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.AggregateFlowStatisticsUpdateBuilder;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAggregateFlowStatisticsFromFlowTableForAllFlowsInput;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAggregateFlowStatisticsFromFlowTableForAllFlowsOutput;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAggregateFlowStatisticsFromFlowTableForAllFlowsOutputBuilder;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.get.aggregate.flow.statistics.from.flow.table._for.given.match.output.AggregatedFlowStatistics;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.transaction.rev150304.TransactionId;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MultipartType;
  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.MultipartRequestInputBuilder;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.OfHeader;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyAggregateCase;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.request.multipart.request.body.MultipartRequestAggregateCaseBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.request.multipart.request.body.multipart.request.aggregate._case.MultipartRequestAggregateBuilder;
  38
  39 public final class AggregateFlowsInTableService extends
  40         AbstractCompatibleStatService<GetAggregateFlowStatisticsFromFlowTableForAllFlowsInput,
  41                                       GetAggregateFlowStatisticsFromFlowTableForAllFlowsOutput,
  42                                       AggregateFlowStatisticsUpdate> {
  43
  44     final TranslatorLibrary translatorLibrary;
  45
  46     public static AggregateFlowsInTableService createWithOook(final RequestContextStack requestContextStack,
  47                                                                                   final DeviceContext deviceContext,
  48                                                                                   AtomicLong compatibilityXidSeed) {
  49         return new AggregateFlowsInTableService(requestContextStack,
  50                                                 deviceContext,
  51                                                 compatibilityXidSeed,
  52                                                 deviceContext.oook());
  53     }
  54
  55     public AggregateFlowsInTableService(final RequestContextStack requestContextStack,
  56                                         final DeviceContext deviceContext,
  57                                         AtomicLong compatibilityXidSeed,
  58                                         TranslatorLibrary translatorLibrary) {
  59         super(requestContextStack, deviceContext, compatibilityXidSeed);
  60
  61         this.translatorLibrary = translatorLibrary;
  62     }
  63
  64     @Override
  65     protected OfHeader buildRequest(final Xid xid, final GetAggregateFlowStatisticsFromFlowTableForAllFlowsInput input)
  66             throws ServiceException {
  67         // Create multipart request body for fetch all the group stats
  68         final MultipartRequestAggregateCaseBuilder multipartRequestAggregateCaseBuilder =
  69                 new MultipartRequestAggregateCaseBuilder();
  70         final MultipartRequestAggregateBuilder mprAggregateRequestBuilder = new MultipartRequestAggregateBuilder();
  71         mprAggregateRequestBuilder.setTableId(input.getTableId().getValue());
  72         mprAggregateRequestBuilder.setOutPort(OFConstants.OFPP_ANY);
  73         mprAggregateRequestBuilder.setOutGroup(OFConstants.OFPG_ANY);
  74         mprAggregateRequestBuilder.setCookie(OFConstants.DEFAULT_COOKIE);
  75         mprAggregateRequestBuilder.setCookieMask(OFConstants.DEFAULT_COOKIE_MASK);
  76         final short version = getVersion();
  77         FlowCreatorUtil.setWildcardedFlowMatch(version, mprAggregateRequestBuilder);
  78
  79         // Set request body to main multipart request
  80         multipartRequestAggregateCaseBuilder.setMultipartRequestAggregate(mprAggregateRequestBuilder
  81                 .build());
  82         final MultipartRequestInputBuilder mprInput = RequestInputUtils.createMultipartHeader(
  83                 MultipartType.OFPMPAGGREGATE, xid.getValue(), version);
  84
  85         mprInput.setMultipartRequestBody(multipartRequestAggregateCaseBuilder.build());
  86
  87         return mprInput.build();
  88     }
  89
  90     @Override
  91     public GetAggregateFlowStatisticsFromFlowTableForAllFlowsOutput buildTxCapableResult(TransactionId emulatedTxId) {
  92         return new GetAggregateFlowStatisticsFromFlowTableForAllFlowsOutputBuilder().setTransactionId(emulatedTxId)
  93                 .build();
  94     }
  95
  96     @Override
  97     public AggregateFlowStatisticsUpdate transformToNotification(List<MultipartReply> result,
  98                                                                  TransactionId emulatedTxId) {
  99         final int mpSize = result.size();
 100         Preconditions.checkArgument(mpSize == 1, "unexpected (!=1) mp-reply size received: {}", mpSize);
 101
 102         MultipartReply mpReply = result.get(0);
 103         final TranslatorKey translatorKey =
 104                 new TranslatorKey(mpReply.getVersion(), MultipartReplyAggregateCase.class.getName());
 105         final MessageTranslator<MultipartReply, AggregatedFlowStatistics> messageTranslator =
 106                 translatorLibrary.lookupTranslator(translatorKey);
 107
 108         final AggregatedFlowStatistics flowStatistics = messageTranslator.translate(mpReply, getDeviceInfo(), null);
 109         final AggregateFlowStatisticsUpdateBuilder notification =
 110                 new AggregateFlowStatisticsUpdateBuilder(flowStatistics)
 111                 .setId(getDeviceInfo().getNodeId())
 112                 .setMoreReplies(Boolean.FALSE)
 113                 .setTransactionId(emulatedTxId);
 114
 115         return notification.build();
 116     }
 117 }