opendaylight/md-sal/statistics-manager/src/main/java/org/opendaylight/controller/md/statistics/manager/FlowTableStatsTracker.java

   1 /*
   2  * Copyright IBM Corporation, 2013.  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.controller.md.statistics.manager;
   9
  10 import java.util.Collections;
  11 import java.util.Set;
  12 import java.util.concurrent.ConcurrentSkipListSet;
  13
  14 import org.opendaylight.controller.sal.binding.api.data.DataModificationTransaction;
  15 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.Table;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.TableBuilder;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.TableKey;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.FlowTableStatisticsData;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.FlowTableStatisticsDataBuilder;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.GetFlowTablesStatisticsInputBuilder;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.OpendaylightFlowTableStatisticsService;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.flow.table.and.statistics.map.FlowTableAndStatisticsMap;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.flow.table.statistics.FlowTableStatistics;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.flow.table.statistics.FlowTableStatisticsBuilder;
  26 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  27
  28 final class FlowTableStatsTracker extends AbstractStatsTracker<FlowTableAndStatisticsMap, FlowTableAndStatisticsMap> {
  29     private final Set<TableKey> privateTables = new ConcurrentSkipListSet<>();
  30     private final Set<TableKey> tables = Collections.unmodifiableSet(privateTables);
  31     private final OpendaylightFlowTableStatisticsService flowTableStatsService;
  32
  33     FlowTableStatsTracker(OpendaylightFlowTableStatisticsService flowTableStatsService, final FlowCapableContext context) {
  34         super(context);
  35         this.flowTableStatsService = flowTableStatsService;
  36     }
  37
  38     Set<TableKey> getTables() {
  39         return tables;
  40     }
  41
  42     @Override
  43     protected void cleanupSingleStat(DataModificationTransaction trans, FlowTableAndStatisticsMap item) {
  44         // TODO: do we want to do this?
  45     }
  46
  47     @Override
  48     protected FlowTableAndStatisticsMap updateSingleStat(DataModificationTransaction trans, FlowTableAndStatisticsMap item) {
  49
  50         InstanceIdentifier<Table> tableRef = getNodeIdentifierBuilder()
  51                 .augmentation(FlowCapableNode.class).child(Table.class, new TableKey(item.getTableId().getValue())).build();
  52
  53         FlowTableStatisticsDataBuilder statisticsDataBuilder = new FlowTableStatisticsDataBuilder();
  54         final FlowTableStatistics stats = new FlowTableStatisticsBuilder(item).build();
  55         statisticsDataBuilder.setFlowTableStatistics(stats);
  56
  57         TableBuilder tableBuilder = new TableBuilder();
  58         tableBuilder.setKey(new TableKey(item.getTableId().getValue()));
  59         tableBuilder.addAugmentation(FlowTableStatisticsData.class, statisticsDataBuilder.build());
  60         trans.putOperationalData(tableRef, tableBuilder.build());
  61         return item;
  62     }
  63
  64     @Override
  65     public void request() {
  66         if (flowTableStatsService != null) {
  67             final GetFlowTablesStatisticsInputBuilder input = new GetFlowTablesStatisticsInputBuilder();
  68             input.setNode(getNodeRef());
  69
  70             requestHelper(flowTableStatsService.getFlowTablesStatistics(input.build()));
  71         }
  72     }
  73 }