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 java.util.concurrent.atomic.AtomicLong;
  12 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  13 import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
  14 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  15 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
  16 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.data.change.counter.rev140815.DataChangeCounter;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.data.change.counter.rev140815.DataChangeCounterBuilder;
  19 import org.opendaylight.yangtools.yang.binding.DataObject;
  20 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  21 import org.slf4j.Logger;
  22 import org.slf4j.LoggerFactory;
  23
  24 public class TopologyDataChangeCounter implements DataChangeListener {
  25
  26     private static final Logger LOG = LoggerFactory.getLogger(TopologyDataChangeCounter.class);
  27
  28     protected static final InstanceIdentifier<DataChangeCounter> IID = InstanceIdentifier
  29             .builder(DataChangeCounter.class).build();
  30
  31     private final DataBroker dataBroker;
  32     private AtomicLong count;
  33
  34     public TopologyDataChangeCounter(final DataBroker dataBroker) {
  35         this.dataBroker = dataBroker;
  36         this.count = new AtomicLong(0);
  37         putCount(this.count.get());
  38         LOG.debug("Data change counter initiated");
  39     }
  40
  41     @Override
  42     public void onDataChanged(AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> change) {
  43         putCount(this.count.incrementAndGet());
  44         LOG.debug("Data change #{}", this.count.get());
  45     }
  46
  47     public void close() {
  48         final WriteTransaction wTx = this.dataBroker.newWriteOnlyTransaction();
  49         wTx.delete(LogicalDatastoreType.OPERATIONAL, IID);
  50         wTx.submit();
  51         LOG.debug("Data change counter removed");
  52     }
  53
  54     private void putCount(final long count) {
  55         final WriteTransaction wTx = this.dataBroker.newWriteOnlyTransaction();
  56         wTx.put(LogicalDatastoreType.OPERATIONAL, IID, new DataChangeCounterBuilder().setCount(count)
  57                 .build());
  58         wTx.submit();
  59     }
  60
  61 }