[AAI-178 Amsterdam] Make Edge Properties to be

[aai/aai-common.git] / aai-core / src / main / java / org / openecomp / aai / dbgen / SchemaGenerator.java

diff --git a/aai-core/src/main/java/org/openecomp/aai/dbgen/SchemaGenerator.java b/aai-core/src/main/java/org/openecomp/aai/dbgen/SchemaGenerator.java
index 9f85d02..165c702 100644 (file)
--- a/aai-core/src/main/java/org/openecomp/aai/dbgen/SchemaGenerator.java
+++ b/aai-core/src/main/java/org/openecomp/aai/dbgen/SchemaGenerator.java
@@ -21,12 +21,14 @@
 package org.openecomp.aai.dbgen;
 
 
-import java.util.HashMap;
-import java.util.HashSet;
-import java.util.Map;
-import java.util.Optional;
-import java.util.Set;
-
+import com.att.eelf.configuration.EELFLogger;
+import com.att.eelf.configuration.EELFManager;
+import com.google.common.collect.Multimap;
+import com.thinkaurelius.titan.core.Cardinality;
+import com.thinkaurelius.titan.core.Multiplicity;
+import com.thinkaurelius.titan.core.PropertyKey;
+import com.thinkaurelius.titan.core.TitanGraph;
+import com.thinkaurelius.titan.core.schema.TitanManagement;
 import org.apache.tinkerpop.gremlin.structure.Vertex;
 import org.openecomp.aai.db.props.AAIProperties;
 import org.openecomp.aai.introspection.Introspector;
@@ -38,15 +40,7 @@ import org.openecomp.aai.serialization.db.EdgeRule;
 import org.openecomp.aai.serialization.db.EdgeRules;
 import org.openecomp.aai.util.AAIConfig;
 
-import com.att.eelf.configuration.EELFLogger;
-import com.att.eelf.configuration.EELFManager;
-import com.google.common.collect.Multimap;
-import com.thinkaurelius.titan.core.Cardinality;
-import com.thinkaurelius.titan.core.Multiplicity;
-import com.thinkaurelius.titan.core.PropertyKey;
-import com.thinkaurelius.titan.core.TitanGraph;
-import com.thinkaurelius.titan.core.schema.TitanManagement;
-
+import java.util.*;
 
 
 public class SchemaGenerator{
@@ -186,6 +180,28 @@ public class SchemaGenerator{
        LOGGER.info(imsg);
        
         graphMgmt.commit();
+        if (addDefaultCR) {
+               if (!graph.traversal().V().has("cloud-owner", "att-aic").has("cloud-region-id", "AAIAIC25").hasNext()) {
+               imsg = "Adding default cloud region to graph...";
+               System.out.println(imsg);
+               LOGGER.info(imsg);
+                       final Vertex cloudRegion = graph.addVertex();
+       
+                       final String ts = String.valueOf(System.currentTimeMillis() / 1000L);
+       
+                       cloudRegion.property("aai-node-type", "cloud-region");
+                       cloudRegion.property("cloud-owner", "att-aic");
+                       cloudRegion.property("cloud-region-id", "AAIAIC25");
+                       cloudRegion.property("cloud-region-version", "2.5");
+                       cloudRegion.property("complex-name", "AAIAIC25");
+                       cloudRegion.property("aai-created-ts", ts);
+                       cloudRegion.property("resource-version", ts);
+                       cloudRegion.property("source-of-truth", "aai-schema-loader");
+                       cloudRegion.property("last-mod-source-of-truth", "aai-schema-loader");
+                       cloudRegion.property(AAIProperties.AAI_URI, "/cloud-infrastructure/cloud-regions/cloud-region/att-aic/AAIAIC25");
+                       graph.tx().commit();
+               }
+               }
     }// End of loadSchemaIntoTitan()
 
 }