re base code

[sdc.git] / catalog-dao / src / main / java / org / openecomp / sdc / be / dao / config / TitanSpringConfig.java

diff --git a/catalog-dao/src/main/java/org/openecomp/sdc/be/dao/config/TitanSpringConfig.java b/catalog-dao/src/main/java/org/openecomp/sdc/be/dao/config/TitanSpringConfig.java
index 111c63e..6360af8 100644 (file)
--- a/catalog-dao/src/main/java/org/openecomp/sdc/be/dao/config/TitanSpringConfig.java
+++ b/catalog-dao/src/main/java/org/openecomp/sdc/be/dao/config/TitanSpringConfig.java
@@ -4,16 +4,20 @@ import org.openecomp.sdc.be.dao.DAOTitanStrategy;
 import org.openecomp.sdc.be.dao.TitanClientStrategy;
 import org.openecomp.sdc.be.dao.titan.TitanGenericDao;
 import org.openecomp.sdc.be.dao.titan.TitanGraphClient;
+import org.openecomp.sdc.be.dao.titan.transactions.SimpleTitanTransactionManager;
 import org.springframework.beans.factory.annotation.Qualifier;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.ComponentScan;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.context.annotation.Primary;
+import org.springframework.transaction.PlatformTransactionManager;
+import org.springframework.transaction.annotation.EnableTransactionManagement;
 
 @Configuration
 @ComponentScan({
         "org.openecomp.sdc.be.dao.jsongraph",
 })
+@EnableTransactionManagement
 public class TitanSpringConfig {
 
     @Bean(name = "titan-generic-dao")
@@ -32,4 +36,9 @@ public class TitanSpringConfig {
     public TitanClientStrategy titanClientStrategy() {
         return new DAOTitanStrategy();
     }
+
+    @Bean
+    public PlatformTransactionManager txManager() {
+        return new SimpleTitanTransactionManager(titanGraphClient(titanClientStrategy()));
+    }
 }