re base code

[sdc.git] / asdctool / src / main / java / org / openecomp / sdc / asdctool / impl / TitanGraphInitializer.java

diff --git a/asdctool/src/main/java/org/openecomp/sdc/asdctool/impl/TitanGraphInitializer.java b/asdctool/src/main/java/org/openecomp/sdc/asdctool/impl/TitanGraphInitializer.java
index ae226f1..28a5bbd 100644 (file)
--- a/asdctool/src/main/java/org/openecomp/sdc/asdctool/impl/TitanGraphInitializer.java
+++ b/asdctool/src/main/java/org/openecomp/sdc/asdctool/impl/TitanGraphInitializer.java
@@ -20,10 +20,10 @@
 
 package org.openecomp.sdc.asdctool.impl;
 
-import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.Map;
-
+import com.thinkaurelius.titan.core.*;
+import com.thinkaurelius.titan.core.schema.ConsistencyModifier;
+import com.thinkaurelius.titan.core.schema.TitanGraphIndex;
+import com.thinkaurelius.titan.core.schema.TitanManagement;
 import org.apache.tinkerpop.gremlin.structure.Edge;
 import org.apache.tinkerpop.gremlin.structure.Vertex;
 import org.openecomp.sdc.be.dao.graph.datatype.ActionEnum;
@@ -39,14 +39,9 @@ import org.openecomp.sdc.be.resources.data.UserData;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-import com.thinkaurelius.titan.core.PropertyKey;
-import com.thinkaurelius.titan.core.TitanException;
-import com.thinkaurelius.titan.core.TitanFactory;
-import com.thinkaurelius.titan.core.TitanGraph;
-import com.thinkaurelius.titan.core.TitanGraphQuery;
-import com.thinkaurelius.titan.core.schema.ConsistencyModifier;
-import com.thinkaurelius.titan.core.schema.TitanGraphIndex;
-import com.thinkaurelius.titan.core.schema.TitanManagement;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.Map;
 
 public class TitanGraphInitializer {
 
@@ -90,6 +85,10 @@ public class TitanGraphInitializer {
                return false;
        }
 
+       private static boolean isVertexNotExist(Map<String, Object> properties) {
+               return !isVertexExist(properties);
+       }
+
        private static void createDefaultAdminUser() {
                createUser(getDefaultUserAdmin());
                graph.tx().commit();
@@ -187,13 +186,19 @@ public class TitanGraphInitializer {
                createVertexIndixes();
                createEdgeIndixes();
                createDefaultAdminUser();
-               createRootCatalogVertex();
+               createRootVertex(VertexTypeEnum.CATALOG_ROOT);
+               createRootVertex(VertexTypeEnum.ARCHIVE_ROOT);
        }
 
-       private static void createRootCatalogVertex() {
-               Vertex vertex = graph.addVertex();
-               vertex.property(GraphPropertyEnum.UNIQUE_ID.getProperty(), IdBuilderUtils.generateUniqueId());
-               vertex.property(GraphPropertyEnum.LABEL.getProperty(), VertexTypeEnum.CATALOG_ROOT.getName());
-               graph.tx().commit();
+       private static void createRootVertex(VertexTypeEnum vertexTypeEnum) {
+               Map<String, Object> checkedProperties = new HashMap<>();
+               checkedProperties.put(GraphPropertiesDictionary.LABEL.getProperty(), vertexTypeEnum.getName());
+               if (isVertexNotExist(checkedProperties)) {
+                       Vertex vertex = graph.addVertex();
+                       vertex.property(GraphPropertyEnum.UNIQUE_ID.getProperty(), IdBuilderUtils.generateUniqueId());
+                       vertex.property(GraphPropertyEnum.LABEL.getProperty(), vertexTypeEnum.getName());
+                       graph.tx().commit();
+               }
        }
+
 }