package org.opendaylight.openflowplugin.impl.statistics;
+import com.google.common.util.concurrent.FutureCallback;
+import com.google.common.util.concurrent.Futures;
import com.google.common.util.concurrent.ListenableFuture;
import com.google.common.util.concurrent.SettableFuture;
import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Collection;
+import java.util.HashSet;
import java.util.List;
-import java.util.concurrent.Future;
+import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionContext;
import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
+import org.opendaylight.openflowplugin.api.openflow.device.DeviceState;
import org.opendaylight.openflowplugin.api.openflow.device.RequestContext;
import org.opendaylight.openflowplugin.api.openflow.statistics.StatisticsContext;
import org.opendaylight.openflowplugin.impl.rpc.RequestContextImpl;
-import org.opendaylight.openflowplugin.impl.statistics.services.OpendaylightFlowStatisticsServiceImpl;
-import org.opendaylight.openflowplugin.impl.statistics.services.OpendaylightFlowTableStatisticsServiceImpl;
-import org.opendaylight.openflowplugin.impl.statistics.services.OpendaylightGroupStatisticsServiceImpl;
-import org.opendaylight.openflowplugin.impl.statistics.services.OpendaylightMeterStatisticsServiceImpl;
-import org.opendaylight.openflowplugin.impl.statistics.services.OpendaylightPortStatisticsServiceImpl;
-import org.opendaylight.openflowplugin.impl.statistics.services.OpendaylightQueueStatisticsServiceImpl;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAllFlowsStatisticsFromAllFlowTablesInputBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.GetAllFlowsStatisticsFromAllFlowTablesOutput;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.OpendaylightFlowStatisticsService;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.OpendaylightFlowTableStatisticsService;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.group.statistics.rev131111.OpendaylightGroupStatisticsService;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.OpendaylightMeterStatisticsService;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.OpendaylightPortStatisticsService;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.queue.statistics.rev131216.OpendaylightQueueStatisticsService;
-import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
-import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
+import org.opendaylight.openflowplugin.impl.services.RequestContextUtil;
+import org.opendaylight.openflowplugin.impl.statistics.services.dedicated.StatisticsGatheringService;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MultipartType;
import org.opendaylight.yangtools.yang.common.RpcResult;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
/**
* Created by Martin Bobak <mbobak@cisco.com> on 1.4.2015.
*/
public class StatisticsContextImpl implements StatisticsContext {
- private final List<RequestContext> requestContexts = new ArrayList();
+ private static final Logger LOG = LoggerFactory.getLogger(StatisticsContextImpl.class);
+ public static final String CONNECTION_CLOSED = "Connection closed.";
+ private final Collection<RequestContext> requestContexts = new HashSet<>();
private final DeviceContext deviceContext;
- private final RequestContext requestContext;
- private final OpendaylightFlowStatisticsService flowStatisticsService;
- private final OpendaylightFlowTableStatisticsService flowTableStatisticsService;
- private final OpendaylightGroupStatisticsService groupStatisticsService;
- private final OpendaylightMeterStatisticsService meterStatisticsService;
- private final OpendaylightPortStatisticsService portStatisticsService;
- private final OpendaylightQueueStatisticsService queueStatisticsService;
+ private final StatisticsGatheringService statisticsGatheringService;
-
- /**
- * FIXME : Why we need RequestContext
- * @param deviceContext
- * @param requestContext
- */
- public StatisticsContextImpl(final DeviceContext deviceContext, final RequestContext requestContext) {
+ public StatisticsContextImpl(final DeviceContext deviceContext) {
this.deviceContext = deviceContext;
- this.requestContext = requestContext;
-
- flowStatisticsService = new OpendaylightFlowStatisticsServiceImpl(this, deviceContext);
- flowTableStatisticsService = new OpendaylightFlowTableStatisticsServiceImpl(this, deviceContext);
- groupStatisticsService = new OpendaylightGroupStatisticsServiceImpl(this, deviceContext);
- meterStatisticsService = new OpendaylightMeterStatisticsServiceImpl(this, deviceContext);
- portStatisticsService = new OpendaylightPortStatisticsServiceImpl(this, deviceContext);
- queueStatisticsService = new OpendaylightQueueStatisticsServiceImpl(this, deviceContext);
+ statisticsGatheringService = new StatisticsGatheringService(this, deviceContext);
}
- private void pollFlowStatistics() {
- final KeyedInstanceIdentifier<Node, NodeKey> nodeII = InstanceIdentifier.create(Nodes.class).child(Node.class, new NodeKey(deviceContext.getPrimaryConnectionContext().getNodeId()));
- final NodeRef nodeRef = new NodeRef(nodeII);
- final GetAllFlowsStatisticsFromAllFlowTablesInputBuilder builder =
- new GetAllFlowsStatisticsFromAllFlowTablesInputBuilder();
- builder.setNode(nodeRef);
- final Future<RpcResult<GetAllFlowsStatisticsFromAllFlowTablesOutput>> flowStatisticsResult = flowStatisticsService.getAllFlowsStatisticsFromAllFlowTables(builder.build());
- //TODO : process data from result
- }
-
@Override
- public ListenableFuture<Void> gatherDynamicData() {
- final ListenableFuture<Boolean> flowStatistics = StatisticsGatheringUtils.gatherFlowStatistics(flowStatisticsService, deviceContext);
- final ListenableFuture<Boolean> tableStatistics = StatisticsGatheringUtils.gatherTableStatistics(flowTableStatisticsService, deviceContext);
- final ListenableFuture<Boolean> groupStatistics = StatisticsGatheringUtils.gatherGroupStatistics(groupStatisticsService, deviceContext);
- final ListenableFuture<Boolean> meterStatistics = StatisticsGatheringUtils.gatherMeterStatistics(meterStatisticsService, deviceContext);
- final ListenableFuture<Boolean> portStatistics = StatisticsGatheringUtils.gatherPortStatistics(portStatisticsService, deviceContext);
- final ListenableFuture<Boolean> queueStatistics = StatisticsGatheringUtils.gatherQueueStatistics(queueStatisticsService, deviceContext);
- return null;
+ public ListenableFuture<Boolean> gatherDynamicData() {
+
+ final SettableFuture settableResultingFuture = SettableFuture.create();
+ ListenableFuture<Boolean> resultingFuture = settableResultingFuture;
+
+
+ if (ConnectionContext.CONNECTION_STATE.WORKING.equals(deviceContext.getPrimaryConnectionContext().getConnectionState())) {
+ final DeviceState devState = deviceContext.getDeviceState();
+ final ListenableFuture<Boolean> emptyFuture = Futures.immediateFuture(new Boolean(false));
+ final ListenableFuture<Boolean> flowStatistics = devState.isFlowStatisticsAvailable() ? StatisticsGatheringUtils.gatherStatistics(statisticsGatheringService, deviceContext, MultipartType.OFPMPFLOW) : emptyFuture;
+
+ final ListenableFuture<Boolean> tableStatistics = devState.isTableStatisticsAvailable() ? StatisticsGatheringUtils.gatherStatistics(statisticsGatheringService, deviceContext, MultipartType.OFPMPTABLE) : emptyFuture;
+
+ final ListenableFuture<Boolean> portStatistics = devState.isPortStatisticsAvailable() ? StatisticsGatheringUtils.gatherStatistics(statisticsGatheringService, deviceContext, MultipartType.OFPMPPORTSTATS) : emptyFuture;
+
+ final ListenableFuture<Boolean> queueStatistics = devState.isQueueStatisticsAvailable() ? StatisticsGatheringUtils.gatherStatistics(statisticsGatheringService, deviceContext, MultipartType.OFPMPQUEUE) : emptyFuture;
+
+ final ListenableFuture<Boolean> groupDescStatistics = devState.isGroupAvailable() ? StatisticsGatheringUtils.gatherStatistics(statisticsGatheringService, deviceContext, MultipartType.OFPMPGROUPDESC) : emptyFuture;
+ final ListenableFuture<Boolean> groupStatistics = devState.isGroupAvailable() ? StatisticsGatheringUtils.gatherStatistics(statisticsGatheringService, deviceContext, MultipartType.OFPMPGROUP) : emptyFuture;
+
+ final ListenableFuture<Boolean> meterConfigStatistics = devState.isMetersAvailable() ? StatisticsGatheringUtils.gatherStatistics(statisticsGatheringService, deviceContext, MultipartType.OFPMPMETERCONFIG) : emptyFuture;
+ final ListenableFuture<Boolean> meterStatistics = devState.isMetersAvailable() ? StatisticsGatheringUtils.gatherStatistics(statisticsGatheringService, deviceContext, MultipartType.OFPMPMETER) : emptyFuture;
+
+
+ final ListenableFuture<List<Boolean>> allFutures = Futures.allAsList(Arrays.asList(flowStatistics, tableStatistics, groupDescStatistics, groupStatistics, meterConfigStatistics, meterStatistics, portStatistics, queueStatistics));
+ Futures.addCallback(allFutures, new FutureCallback<List<Boolean>>() {
+ @Override
+ public void onSuccess(final List<Boolean> booleans) {
+ boolean atLeastOneSuccess = false;
+ for (final Boolean bool : booleans) {
+ atLeastOneSuccess |= bool.booleanValue();
+ }
+ settableResultingFuture.set(new Boolean(atLeastOneSuccess));
+ }
+
+ @Override
+ public void onFailure(final Throwable throwable) {
+ settableResultingFuture.setException(throwable);
+ }
+ });
+ } else {
+ switch (deviceContext.getPrimaryConnectionContext().getConnectionState()) {
+ case RIP:
+ resultingFuture = Futures.immediateFailedFuture(new Throwable(String.format("Device connection doesn't exist anymore. Primary connection status : %s", deviceContext.getPrimaryConnectionContext().getConnectionState())));
+ break;
+ default:
+ resultingFuture = Futures.immediateCheckedFuture(Boolean.TRUE);
+ break;
+ }
+
+
+ }
+ return resultingFuture;
}
@Override
public <T> void forgetRequestContext(final RequestContext<T> requestContext) {
- requestContexts.remove(requestContexts);
+ requestContexts.remove(requestContext);
}
@Override
public <T> RequestContext<T> createRequestContext() {
return new RequestContextImpl<>(this);
}
+
+ @Override
+ public void close() throws Exception {
+ for (final RequestContext requestContext : requestContexts) {
+ RequestContextUtil.closeRequestContextWithRpcError(requestContext, CONNECTION_CLOSED);
+ }
+ }
}
Code Review / openflowplugin.git / blobdiff