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.and.statistics.map.FlowTableAndStatisticsMapBuilder;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.flow.table.statistics.FlowTableStatistics;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.flow.table.statistics.FlowTableStatisticsBuilder;
  27 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  28
  29 final class FlowTableStatsTracker extends AbstractStatsTracker<FlowTableAndStatisticsMap, FlowTableAndStatisticsMap> {
  30     private final Set<TableKey> privateTables = new ConcurrentSkipListSet<>();
  31     private final Set<TableKey> tables = Collections.unmodifiableSet(privateTables);
  32     private final OpendaylightFlowTableStatisticsService flowTableStatsService;
  33
  34     FlowTableStatsTracker(OpendaylightFlowTableStatisticsService flowTableStatsService, final FlowCapableContext context) {
  35         super(context);
  36         this.flowTableStatsService = flowTableStatsService;
  37     }
  38
  39     Set<TableKey> getTables() {
  40         return tables;
  41     }
  42
  43     @Override
  44     protected void cleanupSingleStat(DataModificationTransaction trans, FlowTableAndStatisticsMap item) {
  45         // TODO: do we want to do this?
  46     }
  47
  48     @Override
  49     protected FlowTableAndStatisticsMap updateSingleStat(DataModificationTransaction trans, FlowTableAndStatisticsMap item) {
  50
  51         InstanceIdentifier<Table> tableRef = getNodeIdentifierBuilder()
  52                 .augmentation(FlowCapableNode.class).child(Table.class, new TableKey(item.getTableId().getValue())).build();
  53
  54         FlowTableStatisticsDataBuilder statisticsDataBuilder = new FlowTableStatisticsDataBuilder();
  55         final FlowTableStatistics stats = new FlowTableStatisticsBuilder(item).build();
  56         statisticsDataBuilder.setFlowTableStatistics(stats);
  57
  58         TableBuilder tableBuilder = new TableBuilder();
  59         tableBuilder.setKey(new TableKey(item.getTableId().getValue()));
  60         tableBuilder.addAugmentation(FlowTableStatisticsData.class, statisticsDataBuilder.build());
  61         trans.putOperationalData(tableRef, tableBuilder.build());
  62         return item;
  63     }
  64
  65     @Override
  66     public void request() {
  67         if (flowTableStatsService != null) {
  68             final GetFlowTablesStatisticsInputBuilder input = new GetFlowTablesStatisticsInputBuilder();
  69             input.setNode(getNodeRef());
  70
  71             requestHelper(flowTableStatsService.getFlowTablesStatistics(input.build()));
  72         }
  73     }
  74
  75     @Override
  76     protected FlowTableAndStatisticsMap createInvariantKey(FlowTableAndStatisticsMap item) {
  77         FlowTableAndStatisticsMapBuilder flowTableAndStatisticsMapBuilder = new FlowTableAndStatisticsMapBuilder();
  78         flowTableAndStatisticsMapBuilder.setTableId(item.getTableId());
  79         flowTableAndStatisticsMapBuilder.setKey(item.getKey());
  80         return flowTableAndStatisticsMapBuilder.build();
  81     }
  82 }