1 files changed, 70 insertions, 0 deletions
diff --git a/catalog-dao/src/main/java/org/openecomp/sdc/be/dao/janusgraph/transactions/SimpleJanusGraphTransactionManager.java b/catalog-dao/src/main/java/org/openecomp/sdc/be/dao/janusgraph/transactions/SimpleJanusGraphTransactionManager.java
new file mode 100644
index 0000000000..8c67254da8
--- /dev/null
+++ b/catalog-dao/src/main/java/org/openecomp/sdc/be/dao/janusgraph/transactions/SimpleJanusGraphTransactionManager.java
@@ -0,0 +1,70 @@
+package org.openecomp.sdc.be.dao.janusgraph.transactions;
+
+import org.janusgraph.core.JanusGraphException;
+import org.janusgraph.core.JanusGraph;
+import org.openecomp.sdc.be.dao.janusgraph.JanusGraphClient;
+import org.openecomp.sdc.be.dao.janusgraph.JanusGraphOperationStatus;
+import org.openecomp.sdc.common.log.wrappers.Logger;
+import org.springframework.stereotype.Component;
+import org.springframework.transaction.PlatformTransactionManager;
+import org.springframework.transaction.TransactionDefinition;
+import org.springframework.transaction.TransactionStatus;
+import org.springframework.transaction.TransactionSystemException;
+import org.springframework.transaction.support.SimpleTransactionStatus;
+
+import javax.annotation.PostConstruct;
+
+/**
+ * Simple transaction manager for the janusgraph database.
+ * This manager does not deal with transactions propagation and relies on the fact that transactions are automatically created with the first operation on the graph
+ */
+@Component
+public class SimpleJanusGraphTransactionManager implements PlatformTransactionManager {
+
+    private static final Logger log = Logger.getLogger(SimpleJanusGraphTransactionManager.class.getName());
+    private final JanusGraphClient janusGraphClient;
+    private JanusGraph janusGraph;
+
+    public SimpleJanusGraphTransactionManager(JanusGraphClient janusGraphClient) {
+        this.janusGraphClient = janusGraphClient;
+    }
+
+    @PostConstruct
+    public void onInit() {
+        janusGraph = janusGraphClient.getGraph().left().on(this::onFailingToStartJanusGraph);
+    }
+
+    @Override
+    public TransactionStatus getTransaction(TransactionDefinition transactionDefinition) {
+        log.debug("#getTransaction - returning simple transaction status");
+        return new SimpleTransactionStatus();
+    }
+
+    @Override
+    public void commit(TransactionStatus transactionStatus) {
+        log.debug("#commit - committing transaction");
+        try {
+            janusGraph.tx().commit();
+        } catch (JanusGraphException e) {
+            log.debug("#commit - failed to commit transaction", e);
+            throw new TransactionSystemException("failed to commit transaction", e);
+        }
+    }
+
+    @Override
+    public void rollback(TransactionStatus transactionStatus) {
+        log.debug("#rollback - committing transaction");
+        try {
+            janusGraph.tx().rollback();
+        } catch (JanusGraphException e) {
+            log.debug("#rollback - failed to rollback transaction", e);
+            throw new TransactionSystemException("failed to rollback transaction", e);
+        }
+    }
+
+    private JanusGraph onFailingToStartJanusGraph(JanusGraphOperationStatus err) {
+        log.debug("#onFailingToStartJanusGraph - could not open janusgraph client");
+        throw new IllegalStateException("janusgraph could not be initialized: " + err);
+    }
+
+}