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