data-change-counter/src/main/java/org/opendaylight/protocol/data/change/counter/TopologyDataChangeCounter.java

   1 /*
   2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  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
   9 package org.opendaylight.protocol.data.change.counter;
  10
  11 import com.google.common.util.concurrent.FutureCallback;
  12 import com.google.common.util.concurrent.MoreExecutors;
  13 import java.util.Collection;
  14 import java.util.concurrent.ExecutionException;
  15 import java.util.concurrent.atomic.AtomicBoolean;
  16 import java.util.concurrent.atomic.LongAdder;
  17 import org.opendaylight.mdsal.binding.api.ClusteredDataTreeChangeListener;
  18 import org.opendaylight.mdsal.binding.api.DataBroker;
  19 import org.opendaylight.mdsal.binding.api.DataTreeIdentifier;
  20 import org.opendaylight.mdsal.binding.api.DataTreeModification;
  21 import org.opendaylight.mdsal.binding.api.Transaction;
  22 import org.opendaylight.mdsal.binding.api.TransactionChain;
  23 import org.opendaylight.mdsal.binding.api.TransactionChainListener;
  24 import org.opendaylight.mdsal.binding.api.WriteTransaction;
  25 import org.opendaylight.mdsal.common.api.CommitInfo;
  26 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.data.change.counter.rev160315.DataChangeCounter;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.data.change.counter.rev160315.data.change.counter.Counter;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.data.change.counter.rev160315.data.change.counter.CounterBuilder;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.data.change.counter.rev160315.data.change.counter.CounterKey;
  31 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
  32 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
  33 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
  34 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
  35 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  36 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  37 import org.opendaylight.yangtools.yang.common.Uint32;
  38 import org.slf4j.Logger;
  39 import org.slf4j.LoggerFactory;
  40
  41 final class TopologyDataChangeCounter implements ClusteredDataTreeChangeListener<Topology>,
  42     TransactionChainListener, AutoCloseable {
  43
  44     private static final Logger LOG = LoggerFactory.getLogger(TopologyDataChangeCounter.class);
  45
  46     private final DataBroker dataBroker;
  47     private final String counterId;
  48     private final InstanceIdentifier<Counter> counterInstanceId;
  49     private final LongAdder count = new LongAdder();
  50     private final ListenerRegistration<TopologyDataChangeCounter> registration;
  51     private final AtomicBoolean closed = new AtomicBoolean(false);
  52     private TransactionChain transactionChain;
  53
  54     TopologyDataChangeCounter(final DataBroker dataBroker, final String counterId, final String topologyName) {
  55         this.dataBroker = dataBroker;
  56         this.transactionChain = this.dataBroker.createMergingTransactionChain(this);
  57         this.counterId = counterId;
  58         this.counterInstanceId = InstanceIdentifier.builder(DataChangeCounter.class)
  59                 .child(Counter.class, new CounterKey(this.counterId)).build();
  60         putCount(this.count.longValue());
  61         final InstanceIdentifier<Topology> topoIId = InstanceIdentifier.builder(NetworkTopology.class)
  62                 .child(Topology.class, new TopologyKey(new TopologyId(topologyName))).build();
  63         this.registration = this.dataBroker.registerDataTreeChangeListener(
  64             DataTreeIdentifier.create(LogicalDatastoreType.OPERATIONAL, topoIId), this);
  65         LOG.debug("Data change counter {} initiated", this.counterId);
  66     }
  67
  68     @Override
  69     public synchronized void onDataTreeChanged(final Collection<DataTreeModification<Topology>> changes) {
  70         this.count.increment();
  71         final long inc = this.count.sum();
  72         LOG.debug("Data change #{} for counter {}", inc, this.counterId);
  73         putCount(inc);
  74     }
  75
  76     @Override
  77     public synchronized void close() {
  78         this.registration.close();
  79         final WriteTransaction wTx = this.dataBroker.newWriteOnlyTransaction();
  80         wTx.delete(LogicalDatastoreType.OPERATIONAL, this.counterInstanceId);
  81         try {
  82             wTx.commit().get();
  83         } catch (final ExecutionException | InterruptedException except) {
  84             LOG.warn("Error on remove data change counter {}", this.counterId, except);
  85         }
  86         this.transactionChain.close();
  87         LOG.debug("Data change counter {} removed", this.counterId);
  88     }
  89
  90     private void putCount(final long totalCount) {
  91         final WriteTransaction wTx = this.transactionChain.newWriteOnlyTransaction();
  92         final Counter counter = new CounterBuilder().setId(this.counterId).setCount(Uint32.valueOf(totalCount)).build();
  93         wTx.put(LogicalDatastoreType.OPERATIONAL, this.counterInstanceId, counter);
  94         wTx.commit().addCallback(new FutureCallback<CommitInfo>() {
  95             @Override
  96             public void onSuccess(final CommitInfo result) {
  97                 LOG.debug("Data change count update stored");
  98             }
  99
 100             @Override
 101             public void onFailure(final Throwable trw) {
 102                 LOG.error("Failed to store Data change count");
 103             }
 104         }, MoreExecutors.directExecutor());
 105     }
 106
 107     @Override
 108     public synchronized void onTransactionChainFailed(final TransactionChain chain, final Transaction transaction,
 109         final Throwable cause) {
 110         LOG.warn("Transaction chain failure. Transaction: {}", transaction, cause);
 111         if (!closed.get()) {
 112             this.transactionChain.close();
 113             this.transactionChain = dataBroker.createMergingTransactionChain(this);
 114         }
 115     }
 116
 117     @Override
 118     public synchronized void onTransactionChainSuccessful(final TransactionChain chain) {
 119         LOG.debug("Transaction chain successful. {}", chain);
 120     }
 121 }