openflowplugin-impl/src/main/java/org/opendaylight/openflowplugin/impl/statistics/StatisticsManagerImpl.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
   9 package org.opendaylight.openflowplugin.impl.statistics;
  10
  11 import com.google.common.base.Preconditions;
  12 import com.google.common.util.concurrent.ListeningExecutorService;
  13 import java.util.concurrent.ConcurrentHashMap;
  14 import java.util.concurrent.ConcurrentMap;
  15 import java.util.concurrent.Future;
  16 import java.util.concurrent.Semaphore;
  17 import javax.annotation.Nonnull;
  18 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
  19 import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
  20 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
  21 import org.opendaylight.openflowplugin.api.openflow.device.DeviceInfo;
  22 import org.opendaylight.openflowplugin.api.openflow.statistics.StatisticsContext;
  23 import org.opendaylight.openflowplugin.api.openflow.statistics.StatisticsManager;
  24 import org.opendaylight.openflowplugin.impl.datastore.MultipartWriterProvider;
  25 import org.opendaylight.openflowplugin.impl.datastore.MultipartWriterProviderFactory;
  26 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorExecutor;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.multipart.types.rev170112.MultipartReply;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflow.provider.config.rev160510.OpenflowProviderConfig;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflowplugin.sm.control.rev150812.ChangeStatisticsWorkModeInput;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflowplugin.sm.control.rev150812.GetStatisticsWorkModeOutput;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflowplugin.sm.control.rev150812.GetStatisticsWorkModeOutputBuilder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflowplugin.sm.control.rev150812.StatisticsManagerControlService;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflowplugin.sm.control.rev150812.StatisticsWorkMode;
  34 import org.opendaylight.yangtools.yang.common.RpcError;
  35 import org.opendaylight.yangtools.yang.common.RpcResult;
  36 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
  37 import org.slf4j.Logger;
  38 import org.slf4j.LoggerFactory;
  39
  40 public class StatisticsManagerImpl implements StatisticsManager, StatisticsManagerControlService {
  41
  42     private static final Logger LOG = LoggerFactory.getLogger(StatisticsManagerImpl.class);
  43
  44     private final OpenflowProviderConfig config;
  45     private final ConvertorExecutor converterExecutor;
  46     private final ConcurrentMap<DeviceInfo, StatisticsContext> contexts = new ConcurrentHashMap<>();
  47     private final Semaphore workModeGuard = new Semaphore(1, true);
  48     private final BindingAwareBroker.RpcRegistration<StatisticsManagerControlService> controlServiceRegistration;
  49     private final ListeningExecutorService executorService;
  50     private StatisticsWorkMode workMode = StatisticsWorkMode.COLLECTALL;
  51     private boolean isStatisticsFullyDisabled;
  52
  53     public StatisticsManagerImpl(@Nonnull final OpenflowProviderConfig config,
  54                                  @Nonnull final RpcProviderRegistry rpcProviderRegistry,
  55                                  final ConvertorExecutor convertorExecutor,
  56                                  @Nonnull final ListeningExecutorService executorService) {
  57         this.config = config;
  58         this.converterExecutor = convertorExecutor;
  59         this.controlServiceRegistration = Preconditions.checkNotNull(rpcProviderRegistry
  60                 .addRpcImplementation(StatisticsManagerControlService.class, this));
  61         this.executorService = executorService;
  62     }
  63
  64     @Override
  65     public Future<RpcResult<GetStatisticsWorkModeOutput>> getStatisticsWorkMode() {
  66         return RpcResultBuilder.success(new GetStatisticsWorkModeOutputBuilder()
  67                 .setMode(workMode)
  68                 .build()).buildFuture();
  69     }
  70
  71     @Override
  72     public Future<RpcResult<Void>> changeStatisticsWorkMode(ChangeStatisticsWorkModeInput input) {
  73         if (workModeGuard.tryAcquire()) {
  74             final StatisticsWorkMode targetWorkMode = input.getMode();
  75             isStatisticsFullyDisabled = StatisticsWorkMode.FULLYDISABLED.equals(targetWorkMode);
  76
  77             contexts.values().forEach(context -> {
  78                 switch (targetWorkMode) {
  79                     case COLLECTALL:
  80                         context.enableGathering();
  81                         // FIXME: is it a genuine fall through or an error?
  82                     case FULLYDISABLED:
  83                         context.disableGathering();
  84                         break;
  85                     default:
  86                         LOG.warn("Statistics work mode not supported: {}", targetWorkMode);
  87                 }
  88             });
  89
  90             workModeGuard.release();
  91             return RpcResultBuilder.<Void>success().buildFuture();
  92         }
  93
  94         return RpcResultBuilder.<Void>failed()
  95                 .withError(RpcError.ErrorType.APPLICATION,
  96                         "Statistics work mode change is already in progress")
  97                 .buildFuture();
  98     }
  99
 100     @Override
 101     public StatisticsContext createContext(@Nonnull final DeviceContext deviceContext,
 102                                            final boolean useReconciliationFramework) {
 103         final MultipartWriterProvider statisticsWriterProvider = MultipartWriterProviderFactory
 104                 .createDefaultProvider(deviceContext);
 105
 106         final StatisticsContext statisticsContext =
 107                 deviceContext.canUseSingleLayerSerialization()
 108                         ? new StatisticsContextImpl<MultipartReply>(
 109                                 deviceContext,
 110                                 converterExecutor,
 111                                 statisticsWriterProvider,
 112                                 executorService,
 113                                 !isStatisticsFullyDisabled && config.isIsStatisticsPollingOn(),
 114                                 useReconciliationFramework,
 115                                 config.getBasicTimerDelay().getValue(),
 116                                 config.getMaximumTimerDelay().getValue()) :
 117                         new StatisticsContextImpl<org.opendaylight.yang.gen.v1.urn.opendaylight.openflow
 118                                 .protocol.rev130731.MultipartReply>(
 119                                 deviceContext,
 120                                 converterExecutor,
 121                                 statisticsWriterProvider,
 122                                 executorService,
 123                                 !isStatisticsFullyDisabled && config.isIsStatisticsPollingOn(),
 124                                 useReconciliationFramework,
 125                                 config.getBasicTimerDelay().getValue(),
 126                                 config.getMaximumTimerDelay().getValue());
 127
 128         contexts.put(deviceContext.getDeviceInfo(), statisticsContext);
 129         return statisticsContext;
 130     }
 131
 132     @Override
 133     public void onDeviceRemoved(final DeviceInfo deviceInfo) {
 134         contexts.remove(deviceInfo);
 135         LOG.debug("Statistics context removed for node {}", deviceInfo);
 136     }
 137
 138     @Override
 139     public void close() {
 140         isStatisticsFullyDisabled = true;
 141         controlServiceRegistration.close();
 142
 143         for (StatisticsContext context : contexts.values()) {
 144             context.close();
 145         }
 146
 147         contexts.clear();
 148     }
 149 }