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
8 package org.opendaylight.openflowplugin.impl.statistics;
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;
43 public final class StatisticsManagerImpl implements StatisticsManager {
44 private static final Logger LOG = LoggerFactory.getLogger(StatisticsManagerImpl.class);
47 final ConcurrentMap<DeviceInfo, StatisticsContext> contexts = new ConcurrentHashMap<>();
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;
57 public StatisticsManagerImpl(@NonNull final OpenflowProviderConfig config,
58 @NonNull final RpcProviderService rpcProviderRegistry,
59 final ConvertorExecutor convertorExecutor,
60 @NonNull final Executor executor) {
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)
72 ListenableFuture<RpcResult<GetStatisticsWorkModeOutput>> getStatisticsWorkMode(
73 final GetStatisticsWorkModeInput input) {
74 return RpcResultBuilder.success(new GetStatisticsWorkModeOutputBuilder()
76 .build()).buildFuture();
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);
86 contexts.values().forEach(context -> {
87 switch (targetWorkMode) {
89 context.enableGathering();
92 context.disableGathering();
95 LOG.warn("Statistics work mode not supported: {}", targetWorkMode);
99 workModeGuard.release();
100 return RpcResultBuilder.<ChangeStatisticsWorkModeOutput>success().buildFuture();
103 return RpcResultBuilder.<ChangeStatisticsWorkModeOutput>failed()
104 .withError(ErrorType.APPLICATION, "Statistics work mode change is already in progress")
109 public StatisticsContext createContext(@NonNull final DeviceContext deviceContext,
110 final boolean useReconciliationFramework) {
111 final MultipartWriterProvider statisticsWriterProvider = MultipartWriterProviderFactory
112 .createDefaultProvider(deviceContext);
114 final StatisticsContext statisticsContext = new StatisticsContextImpl<>(
117 statisticsWriterProvider,
120 !isStatisticsFullyDisabled && config.getIsStatisticsPollingOn(),
121 useReconciliationFramework);
123 contexts.put(deviceContext.getDeviceInfo(), statisticsContext);
124 return statisticsContext;
128 public void onDeviceRemoved(final DeviceInfo deviceInfo) {
129 contexts.remove(deviceInfo);
130 LOG.debug("Statistics context removed for node {}", deviceInfo);
134 public void close() {
135 isStatisticsFullyDisabled = true;
136 controlServiceRegistration.close();
138 for (StatisticsContext context : contexts.values()) {