schema-service processing for v19

Issue-ID: AAI-2864
Change-Id: I41a430ec9c9fd69be75bd9d01693249895e0928b
Signed-off-by: LaMont, William(wl2432) <wl2432@att.com>

3 files changed, 319 insertions, 0 deletions

diff --git a/aai-queries/src/main/java/org/onap/aai/queries/CustomQueryConfig.java b/aai-queries/src/main/java/org/onap/aai/queries/CustomQueryConfig.java
new file mode 100644
index 0000000..5a58923
--- /dev/null
+++ b/aai-queries/src/main/java/org/onap/aai/queries/CustomQueryConfig.java
@@ -0,0 +1,53 @@
+/**
+ * ============LICENSE_START=======================================================
+ * org.onap.aai
+ * ================================================================================
+ * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+package org.onap.aai.queries;
+
+import java.util.List;
+
+public class CustomQueryConfig {
+	public CustomQueryConfig() {
+		// used by GetCustomQueryConfig
+	}
+
+
+	private String query;
+	private List<String> queryOptionalProperties;
+	private List<String> queryRequiredProperties;
+
+	public void setQuery(String query) {
+		this.query = query;
+	}
+	public String getQuery() {
+		return this.query;
+	}
+
+	public void setQueryOptionalProperties( List<String> queryOptionalProperties) {
+		this.queryOptionalProperties = queryOptionalProperties;
+	}
+	public List<String> getQueryOptionalProperties( ) {
+		return queryOptionalProperties;
+	}
+	public void setQueryRequiredProperties( List<String> queryRequiredProperties) {
+		this.queryRequiredProperties = queryRequiredProperties;
+	}
+	public List<String> getQueryRequiredProperties( ) {
+		return queryRequiredProperties;
+	}
+}
diff --git a/aai-queries/src/main/java/org/onap/aai/queries/GetCustomQueryConfig.java b/aai-queries/src/main/java/org/onap/aai/queries/GetCustomQueryConfig.java
new file mode 100644
index 0000000..ed4ff99
--- /dev/null
+++ b/aai-queries/src/main/java/org/onap/aai/queries/GetCustomQueryConfig.java
@@ -0,0 +1,149 @@
+/**
+ * ============LICENSE_START=======================================================
+ * org.onap.aai
+ * ================================================================================
+ * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+package org.onap.aai.queries;
+
+import com.google.gson.*;
+import com.google.gson.reflect.TypeToken;
+
+import java.lang.reflect.Type;
+import java.util.ArrayList;
+import java.util.List;
+
+public class GetCustomQueryConfig {
+
+	private JsonArray storedQueries = null;
+	private CustomQueryConfig customQueryConfig;
+
+
+	private final static String QUERY_CONFIG = "query";
+	private final static String REQUIRED_CONFIG = "required-properties";
+	private final static String OPTIONAL_CONFIG = "optional-properties";
+	private final static String STORED_QUERIES_CONFIG = "stored-queries";
+	private final static String STORED_QUERY_CONFIG = "stored-query";
+
+//	public static final String AAI_HOME_ETC_QUERY_JSON = AAIConstants.AAI_HOME_ETC + "query" + AAIConstants.AAI_FILESEP + "stored-queries.json";
+
+	public GetCustomQueryConfig(String customQueryJson ) {
+		init(customQueryJson);
+	}
+
+	private void init( String customQueryJson) {
+		JsonParser parser = new JsonParser();
+		JsonObject queriesObject = parser.parse(customQueryJson).getAsJsonObject();
+		if (queriesObject.has(STORED_QUERIES_CONFIG)) {
+
+			storedQueries = queriesObject.getAsJsonArray(STORED_QUERIES_CONFIG);
+		}
+	}
+
+	private List<String> toStringList(JsonArray array) {
+	   Gson converter = new Gson();
+	   Type listType = new TypeToken<List<String>>() {}.getType();
+	   return converter.fromJson(array, listType);
+	}
+
+	private List<String> getPropertyList(JsonObject configObject, String config ) {
+		JsonElement subqueryConfig;
+		JsonArray props;
+
+		if ( configObject.has(config)) {
+			subqueryConfig = configObject.get(config);
+			if ( subqueryConfig != null && !subqueryConfig.isJsonNull() ) {
+				props = subqueryConfig.getAsJsonArray();
+				if ( props != null ) {
+					return toStringList(props);
+				}
+			}
+		}
+		return toStringList(null);
+	}
+
+	private String getPropertyString(JsonObject configObject, String config) {
+		JsonElement subqueryConfig;
+
+		if ( configObject.has(config)) {
+			subqueryConfig = configObject.get(config);
+			if ( subqueryConfig != null && !subqueryConfig.isJsonNull() ) {
+				return subqueryConfig.getAsString();
+			}
+		}
+		return null;
+	}
+
+	private void getStoredQueryBlock( JsonObject configObject, String config ) {
+		if ( !configObject.has(config)) {
+			customQueryConfig.setQueryRequiredProperties( new ArrayList<String>() );
+			customQueryConfig.setQueryOptionalProperties( new ArrayList<String>()  );
+			return;
+		}
+
+		JsonElement queryConfig;
+		JsonObject subObject;
+		String multipleStartNodes;
+		List<String> propertyList;
+
+		queryConfig = configObject.get(config);
+		subObject = queryConfig.getAsJsonObject();
+		propertyList = getPropertyList(subObject, REQUIRED_CONFIG);
+		if ( propertyList == null ) {
+			propertyList = new ArrayList<String>();
+		}
+		customQueryConfig.setQueryRequiredProperties( propertyList );
+		propertyList = getPropertyList(subObject, OPTIONAL_CONFIG);
+		if ( propertyList == null ) {
+			propertyList = new ArrayList<String>();
+		}
+		customQueryConfig.setQueryOptionalProperties( propertyList );
+
+	}
+
+
+	public CustomQueryConfig getStoredQuery(String queryName ) {
+
+		customQueryConfig = null;
+		JsonObject configObject;
+		JsonElement query;
+		JsonElement queryConfig;
+		String queryString;
+
+		for (JsonElement storedQuery : storedQueries) {
+			if (storedQuery.isJsonObject()) {
+				JsonObject queryObject = storedQuery.getAsJsonObject();
+				query = queryObject.get(queryName);
+				if ( query != null ) {
+					customQueryConfig = new CustomQueryConfig();
+					configObject = query.getAsJsonObject();
+					getStoredQueryBlock(configObject, QUERY_CONFIG);
+					if ( configObject.has(STORED_QUERY_CONFIG)) {
+						queryConfig = configObject.get(STORED_QUERY_CONFIG);
+						customQueryConfig.setQuery(queryConfig.getAsString());
+					}
+					break;
+				}
+			}
+		}
+
+		return customQueryConfig;
+
+	}
+
+
+
+}
diff --git a/aai-queries/src/main/java/org/onap/aai/queries/GremlinServerSingleton.java b/aai-queries/src/main/java/org/onap/aai/queries/GremlinServerSingleton.java
new file mode 100644
index 0000000..3b4a406
--- /dev/null
+++ b/aai-queries/src/main/java/org/onap/aai/queries/GremlinServerSingleton.java
@@ -0,0 +1,117 @@
+/**
+ * ============LICENSE_START=======================================================
+ * org.onap.aai
+ * ================================================================================
+ * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+package org.onap.aai.queries;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import org.onap.aai.aaf.auth.FileWatcher;
+import org.onap.aai.logging.LogFormatTools;
+import org.onap.aai.util.AAIConstants;
+
+import org.springframework.beans.factory.annotation.Value;
+import javax.annotation.PostConstruct;
+
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.Date;
+import java.util.Timer;
+import java.util.TimerTask;
+
+public class GremlinServerSingleton {
+
+    private static Logger logger = LoggerFactory.getLogger(GremlinServerSingleton.class);
+
+    private boolean timerSet;
+    private Timer timer;
+
+    private GetCustomQueryConfig queryConfig;
+
+    @Value("${schema.queries.location}")
+    private String storedQueriesLocation;
+
+    /**
+     * Initializes the gremlin server singleton
+     * Loads the configuration of the gremlin server and creates a cluster
+     * Loads the gremlin query file into the properties object
+     * Then creates a file watcher to watch the file every ten seconds
+     * and if there is a change in the file, then reloads the file into
+     * the properties object
+     *
+     */
+    @PostConstruct
+    public void init() {
+
+		try {
+			String filepath = storedQueriesLocation + AAIConstants.AAI_FILESEP + "stored-queries.json";
+			Path path = Paths.get(filepath);
+			String customQueryConfigJson = new String(Files.readAllBytes(path));
+
+
+			queryConfig = new GetCustomQueryConfig(customQueryConfigJson);
+		} catch (IOException e) {
+			logger.error("Error occurred during the processing of query json file: " + LogFormatTools.getStackTop(e));
+		}
+
+
+        TimerTask task = new FileWatcher(new File(storedQueriesLocation)) {
+            @Override
+            protected void onChange(File file) {
+        		try {
+        			String filepath = storedQueriesLocation;
+        			Path path = Paths.get(filepath);
+        			String customQueryConfigJson = new String(Files.readAllBytes(path));
+        			queryConfig = new GetCustomQueryConfig(customQueryConfigJson);
+        		} catch (IOException e) {
+        			logger.error("Error occurred during the processing of query json file: " + LogFormatTools.getStackTop(e));
+        		}
+            }
+        };
+
+        if (!timerSet) {
+            timerSet = true;
+            timer = new Timer();
+            timer.schedule( task , new Date(), 10000 );
+        }
+
+    }
+
+    /**
+     * Gets the query using CustomQueryConfig
+     * @param key
+     * @return
+     */
+    public String getStoredQueryFromConfig(String key){
+    	CustomQueryConfig customQueryConfig = queryConfig.getStoredQuery(key);
+    	if ( customQueryConfig == null ) {
+    		return null;
+    	}
+    	return customQueryConfig.getQuery();
+    }
+
+    public CustomQueryConfig getCustomQueryConfig(String key) {
+    	return queryConfig.getStoredQuery(key);
+    }
+
+
+}