Sync Common folder

[transportpce.git] / common / src / main / java / org / opendaylight / transportpce / common / network / RequestProcessor.java

diff --git a/common/src/main/java/org/opendaylight/transportpce/common/network/RequestProcessor.java b/common/src/main/java/org/opendaylight/transportpce/common/network/RequestProcessor.java
new file mode 100644 (file)
index 0000000..f64c181
--- /dev/null
+++ b/common/src/main/java/org/opendaylight/transportpce/common/network/RequestProcessor.java
@@ -0,0 +1,153 @@
+/*
+ * Copyright © 2017 Orange, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+package org.opendaylight.transportpce.common.network;
+
+import com.google.common.base.Optional;
+import com.google.common.util.concurrent.CheckedFuture;
+import com.google.common.util.concurrent.ListenableFuture;
+
+import java.util.concurrent.locks.ReentrantReadWriteLock;
+
+import org.opendaylight.controller.md.sal.binding.api.DataBroker;
+import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
+import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
+
+import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
+import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
+
+import org.opendaylight.yangtools.yang.binding.DataObject;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+
+
+public class RequestProcessor {
+
+    private static final Logger LOG = LoggerFactory.getLogger(RequestProcessor.class);
+
+    private final DataBroker dataBroker;
+    private ReadWriteTransaction rwTx;
+    private ReadOnlyTransaction readTx;
+    private ReentrantReadWriteLock lock;
+
+
+
+    public RequestProcessor(DataBroker dataBroker) {
+        this.dataBroker = dataBroker;
+        rwTx = dataBroker.newReadWriteTransaction();
+        readTx = dataBroker.newReadOnlyTransaction();
+        lock = new ReentrantReadWriteLock();
+        LOG.info("RequestProcessor instantiated");
+
+    }
+
+    public <T extends DataObject> CheckedFuture<Optional<T>,
+        ReadFailedException> read(LogicalDatastoreType store,InstanceIdentifier<T> path) {
+
+        CheckedFuture<Optional<T>, ReadFailedException> result = null;
+        acquireReadLock();
+        LOG.debug("Number of threads in queue to read " + lock.getQueueLength());
+        result = rwTx.read(store, path);
+
+        releaseReadLock();
+        return result;
+    }
+
+    public <T extends DataObject> void delete(LogicalDatastoreType store, InstanceIdentifier<?> path) {
+
+        acquireLock();
+        LOG.info("Number of delete requests waiting in queue :" + lock.getQueueLength());
+        rwTx.delete(store, path);
+    }
+
+    public <T extends DataObject> void put(LogicalDatastoreType store,
+        InstanceIdentifier<T> path, T data, boolean createMissingParents) {
+
+        acquireLock();
+        LOG.debug("Number of put requests waiting in queue :" + lock.getQueueLength());
+        rwTx.put(store, path, data, createMissingParents);
+    }
+
+    public <T extends DataObject> void put(LogicalDatastoreType store,
+        InstanceIdentifier<T> path, T data) {
+
+        acquireLock();
+        LOG.debug("Number of put requests waiting in queue :" + lock.getQueueLength());
+        rwTx.put(store, path, data);
+    }
+
+
+    public <T extends DataObject> void merge(LogicalDatastoreType store,
+        InstanceIdentifier<T> path, T data, boolean createMissingParents) {
+
+        acquireLock();
+        LOG.debug("Number of merge requests waiting in queue :" + lock.getQueueLength());
+        rwTx.merge(store, path, data, createMissingParents);
+    }
+
+    public <T extends DataObject> void merge(LogicalDatastoreType store,
+        InstanceIdentifier<T> path, T data) {
+
+        acquireLock();
+        LOG.debug("Number of merge requests waiting in queue :" + lock.getQueueLength());
+        rwTx.merge(store, path, data);
+    }
+
+    public ListenableFuture<Void> submit() {
+        acquireLock();
+        ListenableFuture<Void> future = null;
+        future = rwTx.submit();
+        releaseLock();
+        resetRwTx();
+        return future;
+    }
+
+    public void close() {
+        releaseLock();
+    }
+
+    private void acquireLock() {
+        if (!lock.writeLock().isHeldByCurrentThread()) {
+            lock.writeLock().lock();
+            LOG.debug("Number of write lock requests waiting in queue :" + lock.getQueueLength());
+            LOG.info("Write Lock acquired by : " + Thread.currentThread().getName());
+            rwTx = resetRwTx();
+        } else {
+            LOG.debug("Lock already acquired by : " + Thread.currentThread().getName());
+        }
+    }
+
+    private void acquireReadLock() {
+        if (lock.getReadHoldCount() > 0) {
+            LOG.info("Read Lock already acquired by : " + Thread.currentThread().getName());
+        } else {
+            lock.readLock().lock();
+            rwTx = resetRwTx();
+            LOG.info("Read Lock acquired by : " + Thread.currentThread().getName());
+        }
+    }
+
+    private void releaseLock() {
+        if (lock.writeLock().isHeldByCurrentThread()) {
+            LOG.info("Write Lock released by : " + Thread.currentThread().getName());
+            lock.writeLock().unlock();
+        }
+    }
+
+    private void releaseReadLock() {
+        LOG.info("Read Lock released by : " + Thread.currentThread().getName());
+        lock.readLock().unlock();
+    }
+
+    private ReadWriteTransaction resetRwTx() {
+        LOG.info("Resetting the read write transaction .....");
+        rwTx = dataBroker.newReadWriteTransaction();
+        return rwTx;
+    }
+}