Upgrade SDC from Titan to Janus Graph
[sdc.git] / catalog-be / src / main / java / org / openecomp / sdc / common / transaction / impl / JanusGraphRollbackHandler.java
 
 package org.openecomp.sdc.common.transaction.impl;
 
-import org.openecomp.sdc.be.dao.titan.TitanGenericDao;
-import org.openecomp.sdc.be.dao.titan.TitanOperationStatus;
+import org.openecomp.sdc.be.dao.janusgraph.JanusGraphGenericDao;
+import org.openecomp.sdc.be.dao.janusgraph.JanusGraphOperationStatus;
 import org.openecomp.sdc.common.transaction.api.RollbackHandler;
 import org.openecomp.sdc.common.transaction.api.TransactionUtils.DBActionCodeEnum;
 import org.openecomp.sdc.common.transaction.api.TransactionUtils.DBTypeEnum;
 
-public class TitanRollbackHandler extends RollbackHandler {
+public class JanusGraphRollbackHandler extends RollbackHandler {
 
-    private TitanGenericDao titanGenericDao;
+    private JanusGraphGenericDao janusGraphGenericDao;
 
-    public TitanRollbackHandler(Integer transactionId, String userId, String actionType, TitanGenericDao titanGenericDao) {
+    public JanusGraphRollbackHandler(Integer transactionId, String userId, String actionType, JanusGraphGenericDao janusGraphGenericDao) {
         super(transactionId, userId, actionType);
-        this.titanGenericDao = titanGenericDao;
+        this.janusGraphGenericDao = janusGraphGenericDao;
     }
 
     public DBTypeEnum getDBType() {
-        return DBTypeEnum.TITAN;
+        return DBTypeEnum.JANUSGRAPH;
     }
 
     protected boolean isRollbackForPersistenceData() {
@@ -45,8 +45,8 @@ public class TitanRollbackHandler extends RollbackHandler {
 
     public DBActionCodeEnum doNonPersistenceDataRollback() {
         DBActionCodeEnum result = DBActionCodeEnum.SUCCESS;
-        TitanOperationStatus titanStatus = titanGenericDao.rollback();
-        if (titanStatus != TitanOperationStatus.OK) {
+        JanusGraphOperationStatus janusGraphStatus = janusGraphGenericDao.rollback();
+        if (janusGraphStatus != JanusGraphOperationStatus.OK) {
             result = DBActionCodeEnum.FAIL_GENERAL;
         }
         return result;