2 * Copyright IBM Corporation, 2013. 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.controller.md.statistics.manager;
10 import java.util.Map.Entry;
12 import org.opendaylight.controller.md.sal.common.api.data.DataChangeEvent;
13 import org.opendaylight.controller.sal.binding.api.data.DataBrokerService;
14 import org.opendaylight.controller.sal.binding.api.data.DataModificationTransaction;
15 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNodeConnector;
16 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.port.rev130925.queues.Queue;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.port.rev130925.queues.QueueBuilder;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.port.rev130925.queues.QueueKey;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.queue.rev130925.QueueId;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorId;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnector;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorKey;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.queue.statistics.rev131216.FlowCapableNodeConnectorQueueStatisticsData;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.queue.statistics.rev131216.FlowCapableNodeConnectorQueueStatisticsDataBuilder;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.queue.statistics.rev131216.GetAllQueuesStatisticsFromAllPortsInputBuilder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.queue.statistics.rev131216.GetQueueStatisticsFromGivenPortInputBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.queue.statistics.rev131216.OpendaylightQueueStatisticsService;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.queue.statistics.rev131216.flow.capable.node.connector.queue.statistics.FlowCapableNodeConnectorQueueStatisticsBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.queue.statistics.rev131216.queue.id.and.statistics.map.QueueIdAndStatisticsMap;
30 import org.opendaylight.yangtools.yang.binding.DataObject;
31 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
32 import org.slf4j.Logger;
33 import org.slf4j.LoggerFactory;
35 final class QueueStatsTracker extends AbstractListeningStatsTracker<QueueIdAndStatisticsMap, QueueStatsEntry> {
36 private static final Logger logger = LoggerFactory.getLogger(QueueStatsTracker.class);
37 private final OpendaylightQueueStatisticsService queueStatsService;
39 QueueStatsTracker(OpendaylightQueueStatisticsService queueStatsService, final FlowCapableContext context) {
41 this.queueStatsService = queueStatsService;
45 protected void cleanupSingleStat(DataModificationTransaction trans, QueueStatsEntry item) {
46 InstanceIdentifier<?> queueRef
47 = getNodeIdentifierBuilder().child(NodeConnector.class, new NodeConnectorKey(item.getNodeConnectorId()))
48 .augmentation(FlowCapableNodeConnector.class)
49 .child(Queue.class, new QueueKey(item.getQueueId()))
50 .augmentation(FlowCapableNodeConnectorQueueStatisticsData.class).build();
51 trans.removeOperationalData(queueRef);
55 protected QueueStatsEntry updateSingleStat(DataModificationTransaction trans, QueueIdAndStatisticsMap item) {
57 QueueStatsEntry queueEntry = new QueueStatsEntry(item.getNodeConnectorId(), item.getQueueId());
59 FlowCapableNodeConnectorQueueStatisticsDataBuilder queueStatisticsDataBuilder = new FlowCapableNodeConnectorQueueStatisticsDataBuilder();
61 FlowCapableNodeConnectorQueueStatisticsBuilder queueStatisticsBuilder = new FlowCapableNodeConnectorQueueStatisticsBuilder();
63 queueStatisticsBuilder.fieldsFrom(item);
65 queueStatisticsDataBuilder.setFlowCapableNodeConnectorQueueStatistics(queueStatisticsBuilder.build());
67 InstanceIdentifier<Queue> queueRef = getNodeIdentifierBuilder().child(NodeConnector.class, new NodeConnectorKey(item.getNodeConnectorId()))
68 .augmentation(FlowCapableNodeConnector.class)
69 .child(Queue.class, new QueueKey(item.getQueueId())).toInstance();
71 QueueBuilder queueBuilder = new QueueBuilder();
72 FlowCapableNodeConnectorQueueStatisticsData qsd = queueStatisticsDataBuilder.build();
73 queueBuilder.addAugmentation(FlowCapableNodeConnectorQueueStatisticsData.class, qsd);
74 queueBuilder.setKey(new QueueKey(item.getQueueId()));
76 logger.debug("Augmenting queue statistics {} of queue {} to port {}",
79 item.getNodeConnectorId());
81 trans.putOperationalData(queueRef, queueBuilder.build());
86 public void request() {
87 if (queueStatsService != null) {
88 GetAllQueuesStatisticsFromAllPortsInputBuilder input = new GetAllQueuesStatisticsFromAllPortsInputBuilder();
89 input.setNode(getNodeRef());
91 requestHelper(queueStatsService.getAllQueuesStatisticsFromAllPorts(input.build()));
95 public void request(NodeConnectorId nodeConnectorId, QueueId queueId) {
96 if (queueStatsService != null) {
97 GetQueueStatisticsFromGivenPortInputBuilder input = new GetQueueStatisticsFromGivenPortInputBuilder();
99 input.setNode(getNodeRef());
100 input.setNodeConnectorId(nodeConnectorId);
101 input.setQueueId(queueId);
103 requestHelper(queueStatsService.getQueueStatisticsFromGivenPort(input.build()));
108 public void onDataChanged(DataChangeEvent<InstanceIdentifier<?>, DataObject> change) {
109 for (Entry<InstanceIdentifier<?>, DataObject> e : change.getCreatedConfigurationData().entrySet()) {
110 if (Queue.class.equals(e.getKey().getTargetType())) {
111 final Queue queue = (Queue) e.getValue();
112 final NodeConnectorKey key = e.getKey().firstKeyOf(NodeConnector.class, NodeConnectorKey.class);
113 logger.debug("Key {} triggered request for connector {} queue {}", key.getId(), queue.getQueueId());
114 request(key.getId(), queue.getQueueId());
116 logger.debug("Ignoring key {}", e.getKey());
120 final DataModificationTransaction trans = startTransaction();
121 for (InstanceIdentifier<?> key : change.getRemovedConfigurationData()) {
122 if (Queue.class.equals(key.getTargetType())) {
123 @SuppressWarnings("unchecked")
124 final InstanceIdentifier<Queue> queue = (InstanceIdentifier<Queue>)key;
125 final InstanceIdentifier<?> del = queue
126 .augmentation(FlowCapableNodeConnectorQueueStatisticsData.class);
127 logger.debug("Key {} triggered remove of augmentation {}", key, del);
129 trans.removeOperationalData(del);
136 protected InstanceIdentifier<?> listenPath() {
137 return getNodeIdentifierBuilder().child(NodeConnector.class)
138 .augmentation(FlowCapableNodeConnector.class).child(Queue.class).build();
142 protected String statName() {
147 public void start(final DataBrokerService dbs) {
148 if (queueStatsService == null) {
149 logger.debug("No Queue Statistics service, not subscribing to queues on node {}", getNodeIdentifier());