2 * Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
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
9 package org.opendaylight.openflowplugin.impl.statistics;
11 import com.google.common.base.Preconditions;
12 import com.google.common.util.concurrent.ListenableFuture;
13 import com.google.common.util.concurrent.ListeningExecutorService;
14 import java.util.concurrent.ConcurrentHashMap;
15 import java.util.concurrent.ConcurrentMap;
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.params.xml.ns.yang.openflow.provider.config.rev160510.OpenflowProviderConfig;
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.GetStatisticsWorkModeInput;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflowplugin.sm.control.rev150812.GetStatisticsWorkModeOutput;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflowplugin.sm.control.rev150812.GetStatisticsWorkModeOutputBuilder;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflowplugin.sm.control.rev150812.StatisticsManagerControlService;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflowplugin.sm.control.rev150812.StatisticsWorkMode;
35 import org.opendaylight.yangtools.yang.common.RpcError;
36 import org.opendaylight.yangtools.yang.common.RpcResult;
37 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
38 import org.slf4j.Logger;
39 import org.slf4j.LoggerFactory;
41 public class StatisticsManagerImpl implements StatisticsManager, StatisticsManagerControlService {
43 private static final Logger LOG = LoggerFactory.getLogger(StatisticsManagerImpl.class);
45 private final OpenflowProviderConfig config;
46 private final ConvertorExecutor converterExecutor;
47 private final ConcurrentMap<DeviceInfo, StatisticsContext> contexts = new ConcurrentHashMap<>();
48 private final Semaphore workModeGuard = new Semaphore(1, true);
49 private final BindingAwareBroker.RpcRegistration<StatisticsManagerControlService> controlServiceRegistration;
50 private final ListeningExecutorService executorService;
51 private final StatisticsWorkMode workMode = StatisticsWorkMode.COLLECTALL;
52 private boolean isStatisticsFullyDisabled;
54 public StatisticsManagerImpl(@Nonnull final OpenflowProviderConfig config,
55 @Nonnull final RpcProviderRegistry rpcProviderRegistry,
56 final ConvertorExecutor convertorExecutor,
57 @Nonnull final ListeningExecutorService executorService) {
59 this.converterExecutor = convertorExecutor;
60 this.controlServiceRegistration = Preconditions.checkNotNull(rpcProviderRegistry
61 .addRpcImplementation(StatisticsManagerControlService.class, this));
62 this.executorService = executorService;
66 public ListenableFuture<RpcResult<GetStatisticsWorkModeOutput>> getStatisticsWorkMode(
67 GetStatisticsWorkModeInput input) {
68 return RpcResultBuilder.success(new GetStatisticsWorkModeOutputBuilder()
70 .build()).buildFuture();
74 public ListenableFuture<RpcResult<ChangeStatisticsWorkModeOutput>> changeStatisticsWorkMode(
75 ChangeStatisticsWorkModeInput input) {
76 if (workModeGuard.tryAcquire()) {
77 final StatisticsWorkMode targetWorkMode = input.getMode();
78 isStatisticsFullyDisabled = StatisticsWorkMode.FULLYDISABLED.equals(targetWorkMode);
80 contexts.values().forEach(context -> {
81 switch (targetWorkMode) {
83 context.enableGathering();
86 context.disableGathering();
89 LOG.warn("Statistics work mode not supported: {}", targetWorkMode);
93 workModeGuard.release();
94 return RpcResultBuilder.<ChangeStatisticsWorkModeOutput>success().buildFuture();
97 return RpcResultBuilder.<ChangeStatisticsWorkModeOutput>failed()
98 .withError(RpcError.ErrorType.APPLICATION,
99 "Statistics work mode change is already in progress")
104 public StatisticsContext createContext(@Nonnull final DeviceContext deviceContext,
105 final boolean useReconciliationFramework) {
106 final MultipartWriterProvider statisticsWriterProvider = MultipartWriterProviderFactory
107 .createDefaultProvider(deviceContext);
109 final StatisticsContext statisticsContext = new StatisticsContextImpl<>(
112 statisticsWriterProvider,
115 !isStatisticsFullyDisabled && config.isIsStatisticsPollingOn(),
116 useReconciliationFramework);
118 contexts.put(deviceContext.getDeviceInfo(), statisticsContext);
119 return statisticsContext;
123 public void onDeviceRemoved(final DeviceInfo deviceInfo) {
124 contexts.remove(deviceInfo);
125 LOG.debug("Statistics context removed for node {}", deviceInfo);
129 public void close() {
130 isStatisticsFullyDisabled = true;
131 controlServiceRegistration.close();
133 for (StatisticsContext context : contexts.values()) {