package org.opendaylight.openflowplugin.impl.statistics;
+import com.google.common.annotations.VisibleForTesting;
+import com.google.common.base.Optional;
import com.google.common.util.concurrent.FutureCallback;
import com.google.common.util.concurrent.Futures;
import com.google.common.util.concurrent.ListenableFuture;
+import io.netty.util.HashedWheelTimer;
+import io.netty.util.Timeout;
+import io.netty.util.TimerTask;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.Future;
+import java.util.concurrent.Semaphore;
+import java.util.concurrent.TimeUnit;
+import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
+import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
+import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionContext;
import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
import org.opendaylight.openflowplugin.api.openflow.device.handlers.DeviceInitializationPhaseHandler;
+import org.opendaylight.openflowplugin.api.openflow.rpc.ItemLifeCycleSource;
import org.opendaylight.openflowplugin.api.openflow.statistics.StatisticsContext;
import org.opendaylight.openflowplugin.api.openflow.statistics.StatisticsManager;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflowplugin.sm.control.rev150812.ChangeStatisticsWorkModeInput;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflowplugin.sm.control.rev150812.GetStatisticsWorkModeOutput;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflowplugin.sm.control.rev150812.GetStatisticsWorkModeOutputBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflowplugin.sm.control.rev150812.StatisticsManagerControlService;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflowplugin.sm.control.rev150812.StatisticsWorkMode;
+import org.opendaylight.yangtools.yang.common.RpcError;
+import org.opendaylight.yangtools.yang.common.RpcResult;
+import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.role.service.rev150727.OfpRole;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* Created by Martin Bobak <mbobak@cisco.com> on 1.4.2015.
*/
-public class StatisticsManagerImpl implements StatisticsManager {
+public class StatisticsManagerImpl implements StatisticsManager, StatisticsManagerControlService {
private static final Logger LOG = LoggerFactory.getLogger(StatisticsManagerImpl.class);
+ private final RpcProviderRegistry rpcProviderRegistry;
private DeviceInitializationPhaseHandler deviceInitPhaseHandler;
+ private HashedWheelTimer hashedWheelTimer;
+
+ private final ConcurrentHashMap<DeviceContext, StatisticsContext> contexts = new ConcurrentHashMap<>();
+
+ private static final long basicTimerDelay = 3000;
+ private static long currentTimerDelay = basicTimerDelay;
+ private static long maximumTimerDelay = 900000; //wait max 15 minutes for next statistics
+
+ private StatisticsWorkMode workMode = StatisticsWorkMode.COLLECTALL;
+ private Semaphore workModeGuard = new Semaphore(1, true);
+ private boolean shuttingDownStatisticsPolling;
+ private BindingAwareBroker.RpcRegistration<StatisticsManagerControlService> controlServiceRegistration;
+
@Override
public void setDeviceInitializationPhaseHandler(final DeviceInitializationPhaseHandler handler) {
deviceInitPhaseHandler = handler;
}
+ public StatisticsManagerImpl(RpcProviderRegistry rpcProviderRegistry) {
+ this.rpcProviderRegistry = rpcProviderRegistry;
+ controlServiceRegistration = rpcProviderRegistry.addRpcImplementation(StatisticsManagerControlService.class, this);
+ }
+
+ public StatisticsManagerImpl(RpcProviderRegistry rpcProviderRegistry, final boolean shuttingDownStatisticsPolling) {
+ this(rpcProviderRegistry);
+ this.shuttingDownStatisticsPolling = shuttingDownStatisticsPolling;
+ }
+
@Override
public void onDeviceContextLevelUp(final DeviceContext deviceContext) {
+ LOG.debug("Node:{}, deviceContext.getDeviceState().getRole():{}", deviceContext.getDeviceState().getNodeId(),
+ deviceContext.getDeviceState().getRole());
+ if (deviceContext.getDeviceState().getRole() == OfpRole.BECOMESLAVE) {
+ // if slave, we dont poll for statistics and jump to rpc initialization
+ LOG.info("Skipping Statistics for slave role for node:{}", deviceContext.getDeviceState().getNodeId());
+ deviceInitPhaseHandler.onDeviceContextLevelUp(deviceContext);
+ return;
+ }
+
+ if (null == hashedWheelTimer) {
+ LOG.trace("This is first device that delivered timer. Starting statistics polling immediately.");
+ hashedWheelTimer = deviceContext.getTimer();
+ }
+
+ LOG.info("Starting Statistics for master role for node:{}", deviceContext.getDeviceState().getNodeId());
+
final StatisticsContext statisticsContext = new StatisticsContextImpl(deviceContext);
- final ListenableFuture<Void> weHaveDynamicData = statisticsContext.gatherDynamicData();
- Futures.addCallback(weHaveDynamicData, new FutureCallback<Void>() {
+ deviceContext.addDeviceContextClosedHandler(this);
+ final ListenableFuture<Boolean> weHaveDynamicData = statisticsContext.gatherDynamicData();
+ Futures.addCallback(weHaveDynamicData, new FutureCallback<Boolean>() {
@Override
- public void onSuccess(final Void aVoid) {
- // wake up RPC registration
- deviceInitPhaseHandler.onDeviceContextLevelUp(deviceContext);
+ public void onSuccess(final Boolean statisticsGathered) {
+ if (statisticsGathered) {
+ //there are some statistics on device worth gathering
+ contexts.put(deviceContext, statisticsContext);
+ final TimeCounter timeCounter = new TimeCounter();
+ scheduleNextPolling(deviceContext, statisticsContext, timeCounter);
+ LOG.trace("Device dynamic info collecting done. Going to announce raise to next level.");
+ deviceInitPhaseHandler.onDeviceContextLevelUp(deviceContext);
+ deviceContext.getDeviceState().setDeviceSynchronized(true);
+ } else {
+ final String deviceAdress = deviceContext.getPrimaryConnectionContext().getConnectionAdapter().getRemoteAddress().toString();
+ try {
+ deviceContext.close();
+ } catch (Exception e) {
+ LOG.info("Statistics for device {} could not be gathered. Closing its device context.", deviceAdress);
+ }
+ }
}
@Override
public void onFailure(final Throwable throwable) {
- LOG.error("Statistics manager was not able to collect dynamic info for device {}", deviceContext.getDeviceState().getNodeId());
+ LOG.warn("Statistics manager was not able to collect dynamic info for device.", deviceContext.getDeviceState().getNodeId(), throwable);
+ try {
+ deviceContext.close();
+ } catch (Exception e) {
+ LOG.warn("Error closing device context.", e);
+ }
}
});
}
+
+ private void pollStatistics(final DeviceContext deviceContext,
+ final StatisticsContext statisticsContext,
+ final TimeCounter timeCounter) {
+ timeCounter.markStart();
+ ListenableFuture<Boolean> deviceStatisticsCollectionFuture = statisticsContext.gatherDynamicData();
+ Futures.addCallback(deviceStatisticsCollectionFuture, new FutureCallback<Boolean>() {
+ @Override
+ public void onSuccess(final Boolean o) {
+ timeCounter.addTimeMark();
+ calculateTimerDelay(timeCounter);
+ scheduleNextPolling(deviceContext, statisticsContext, timeCounter);
+ }
+
+ @Override
+ public void onFailure(final Throwable throwable) {
+ timeCounter.addTimeMark();
+ LOG.info("Statistics gathering for single node was not successful: {}", throwable.getMessage());
+ LOG.debug("Statistics gathering for single node was not successful.. ", throwable);
+ if (ConnectionContext.CONNECTION_STATE.WORKING.equals(deviceContext.getPrimaryConnectionContext().getConnectionState())) {
+ calculateTimerDelay(timeCounter);
+ scheduleNextPolling(deviceContext, statisticsContext, timeCounter);
+ }
+ }
+ });
+ }
+
+ private void scheduleNextPolling(final DeviceContext deviceContext,
+ final StatisticsContext statisticsContext,
+ final TimeCounter timeCounter) {
+ if (null != hashedWheelTimer) {
+ if (!shuttingDownStatisticsPolling) {
+ Timeout pollTimeout = hashedWheelTimer.newTimeout(new TimerTask() {
+ @Override
+ public void run(final Timeout timeout) throws Exception {
+ pollStatistics(deviceContext, statisticsContext, timeCounter);
+ }
+ }, currentTimerDelay, TimeUnit.MILLISECONDS);
+ statisticsContext.setPollTimeout(pollTimeout);
+ }
+ }
+ }
+
+ @VisibleForTesting
+ protected void calculateTimerDelay(final TimeCounter timeCounter) {
+ // TODO: move into TimeCounter
+ long averageStatisticsGatheringTime = timeCounter.getAverageTimeBetweenMarks();
+ if (averageStatisticsGatheringTime > currentTimerDelay) {
+ currentTimerDelay *= 2;
+ if (currentTimerDelay > maximumTimerDelay) {
+ currentTimerDelay = maximumTimerDelay;
+ }
+ } else {
+ if (currentTimerDelay > basicTimerDelay) {
+ currentTimerDelay /= 2;
+ } else {
+ currentTimerDelay = basicTimerDelay;
+ }
+ }
+ }
+
+ @VisibleForTesting
+ protected static long getCurrentTimerDelay() {
+ return currentTimerDelay;
+ }
+
+ @Override
+ public void onDeviceContextClosed(final DeviceContext deviceContext) {
+ StatisticsContext statisticsContext = contexts.remove(deviceContext);
+ if (null != statisticsContext) {
+ LOG.trace("Removing device context from stack. No more statistics gathering for node {}", deviceContext.getDeviceState().getNodeId());
+ try {
+ statisticsContext.close();
+ } catch (Exception e) {
+ LOG.debug("Error closing statistic context for node {}.", deviceContext.getDeviceState().getNodeId());
+ }
+ }
+ }
+
+ @Override
+ public Future<RpcResult<GetStatisticsWorkModeOutput>> getStatisticsWorkMode() {
+ GetStatisticsWorkModeOutputBuilder smModeOutputBld = new GetStatisticsWorkModeOutputBuilder();
+ smModeOutputBld.setMode(workMode);
+ return RpcResultBuilder.success(smModeOutputBld.build()).buildFuture();
+ }
+
+ @Override
+ public Future<RpcResult<Void>> changeStatisticsWorkMode(ChangeStatisticsWorkModeInput input) {
+ final Future<RpcResult<Void>> result;
+ // acquire exclusive access
+ if (workModeGuard.tryAcquire()) {
+ final StatisticsWorkMode targetWorkMode = input.getMode();
+ if (!workMode.equals(targetWorkMode)) {
+ shuttingDownStatisticsPolling = StatisticsWorkMode.FULLYDISABLED.equals(targetWorkMode);
+ // iterate through stats-ctx: propagate mode
+ for (Map.Entry<DeviceContext, StatisticsContext> contextEntry : contexts.entrySet()) {
+ final DeviceContext deviceContext = contextEntry.getKey();
+ final StatisticsContext statisticsContext = contextEntry.getValue();
+ switch (targetWorkMode) {
+ case COLLECTALL:
+ scheduleNextPolling(deviceContext, statisticsContext, new TimeCounter());
+ for (ItemLifeCycleSource lifeCycleSource : deviceContext.getItemLifeCycleSourceRegistry().getLifeCycleSources()) {
+ lifeCycleSource.setItemLifecycleListener(null);
+ }
+ break;
+ case FULLYDISABLED:
+ final Optional<Timeout> pollTimeout = statisticsContext.getPollTimeout();
+ if (pollTimeout.isPresent()) {
+ pollTimeout.get().cancel();
+ }
+ for (ItemLifeCycleSource lifeCycleSource : deviceContext.getItemLifeCycleSourceRegistry().getLifeCycleSources()) {
+ lifeCycleSource.setItemLifecycleListener(statisticsContext.getItemLifeCycleListener());
+ }
+ break;
+ default:
+ LOG.warn("statistics work mode not supported: {}", targetWorkMode);
+ }
+ }
+ workMode = targetWorkMode;
+ }
+ workModeGuard.release();
+ result = RpcResultBuilder.<Void>success().buildFuture();
+ } else {
+ result = RpcResultBuilder.<Void>failed()
+ .withError(RpcError.ErrorType.APPLICATION, "mode change already in progress")
+ .buildFuture();
+ }
+ return result;
+ }
+
+ @Override
+ public void close() {
+ if (controlServiceRegistration != null) {
+ controlServiceRegistration.close();
+ controlServiceRegistration = null;
+ }
+ }
}