Reset all version failure

[sdc.git] / openecomp-be / tools / zusammen-tools / src / main / java / org / openecomp / core / tools / store / ElementHandler.java

diff --git a/openecomp-be/tools/zusammen-tools/src/main/java/org/openecomp/core/tools/store/ElementHandler.java b/openecomp-be/tools/zusammen-tools/src/main/java/org/openecomp/core/tools/store/ElementHandler.java
index 2690130..5c72c95 100644 (file)
--- a/openecomp-be/tools/zusammen-tools/src/main/java/org/openecomp/core/tools/store/ElementHandler.java
+++ b/openecomp-be/tools/zusammen-tools/src/main/java/org/openecomp/core/tools/store/ElementHandler.java
@@ -2,25 +2,32 @@ package org.openecomp.core.tools.store;
 
 import com.amdocs.zusammen.datatypes.Id;
 import com.amdocs.zusammen.datatypes.SessionContext;
+import com.amdocs.zusammen.datatypes.item.ElementContext;
 import com.amdocs.zusammen.plugin.statestore.cassandra.dao.types.ElementEntityContext;
-import org.openecomp.core.zusammen.plugin.dao.impl.CassandraElementRepository;
 import org.openecomp.core.zusammen.plugin.dao.types.ElementEntity;
 
 public class ElementHandler {
 
-  private static final String GLOBAL_USER = "GLOBAL_USER";
+    private static final String GLOBAL_USER = "GLOBAL_USER";
 
-  public static void update(SessionContext context,
+    public static void update(SessionContext context,
+                              String itemId, String versionId, String changeRef,
+                              ElementEntity elementEntity) {
 
-                            String itemId, String versionId,
-                            ElementEntity elementEntity) {
+        ElementEntityContext elementContext;
+        CassandraElementRepository cassandraElementRepository = new CassandraElementRepository();
+        if (changeRef == null) {
 
-    ElementEntityContext elementContext;
-    elementContext = new ElementEntityContext(GLOBAL_USER, new Id(itemId),
-        new Id(versionId));
-    CassandraElementRepository cassandraElementRepository = new CassandraElementRepository();
-    cassandraElementRepository.update(context, elementContext, elementEntity);
+            elementContext = new ElementEntityContext(GLOBAL_USER, new Id(itemId),
+                    new Id(versionId));
 
-  }
+        } else {
+            elementContext = new ElementEntityContext(GLOBAL_USER, new ElementContext(new Id(itemId),
+                    new Id(versionId),
+                    changeRef));
+        }
+        cassandraElementRepository.update(context, elementContext, elementEntity);
+
+    }
 }