X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Forg%2Fonap%2Faai%2Fdatarouter%2Futil%2FSearchServiceAgent.java;h=bbdb4c8edd38135f1c417d5bcd37fc8e107a31f5;hb=9939a1d4a4d390cfb89829ad253d94f8f4102ac5;hp=fe1df3bdd4433fd51f5747ef1be2394b81a174cf;hpb=1aae9de4d596a1108818c2d035225a3ba2737501;p=aai%2Fdata-router.git

diff --git a/src/main/java/org/onap/aai/datarouter/util/SearchServiceAgent.java b/src/main/java/org/onap/aai/datarouter/util/SearchServiceAgent.java
index fe1df3b..bbdb4c8 100644
--- a/src/main/java/org/onap/aai/datarouter/util/SearchServiceAgent.java
+++ b/src/main/java/org/onap/aai/datarouter/util/SearchServiceAgent.java
@@ -127,6 +127,14 @@ public class SearchServiceAgent {
     createIndex(index, schemaLocation);
   }
   
+  public void createSearchIndex(String index, String schemaLocation, String endUrl) {
+    
+    // Create a mapping of the index name to schema location 
+    indexSchemaMapping.put(index, schemaLocation);
+    
+    // Now, create the index.
+    createIndex(index, schemaLocation, endUrl);
+  }
   
   /**
    * This method performs the actual work of creating a search index.
@@ -160,6 +168,30 @@ public class SearchServiceAgent {
     }
   }
   
+  private void createIndex(String index, String schemaLocation, String endUrl) {
+    
+    logger.debug("Creating search index, index name: = " + index + ", schemaLocation = " + schemaLocation);
+    
+    MultivaluedMap<String, String> headers = new MultivaluedMapImpl();
+    headers.put("Accept", Arrays.asList("application/json"));
+    headers.put(Headers.FROM_APP_ID, Arrays.asList("DL"));
+    headers.put(Headers.TRANSACTION_ID, Arrays.asList(UUID.randomUUID().toString()));
+      
+    String url = concatSubUri(searchUrl, endUrl, index);
+    try {
+
+      OperationResult result = searchClient.put(url, loadFileData(schemaLocation), headers,
+                                                MediaType.APPLICATION_JSON_TYPE, null);
+      if (!HttpUtil.isHttpResponseClassSuccess(result.getResultCode())) {
+        logger.error(DataRouterMsgs.FAIL_TO_CREATE_SEARCH_INDEX, index, result.getFailureCause());
+      } else {
+        logger.info(DataRouterMsgs.SEARCH_INDEX_CREATE_SUCCESS, index);
+      }
+
+    } catch (Exception e) {
+      logger.error(DataRouterMsgs.FAIL_TO_CREATE_SEARCH_INDEX, index, e.getLocalizedMessage());
+    }
+  }
   
   /**
    * Retrieves a document from the search service.
@@ -288,8 +320,7 @@ public class SearchServiceAgent {
    * Removes a document from the Search Service.
    * 
    * @param index   - The index to create the document in.
-   * @param id      - The identifier for the document.
-   * @param payload - The document contents.
+   * @param documentId      - The identifier for the document.
    * @param headers - HTTP headers.
    */
   public void deleteDocument(String index, String documentId, Map<String, List<String>> headers) {
@@ -308,12 +339,13 @@ public class SearchServiceAgent {
    */
   protected String loadFileData(String filename) throws Exception {
     StringBuilder data = new StringBuilder();
-    try {
-      BufferedReader in = new BufferedReader(new InputStreamReader(
-          EntityEventPolicy.class.getClassLoader().getResourceAsStream("/" + filename),
-          StandardCharsets.UTF_8));
-      String line;
 
+    try (InputStreamReader inputStreamReader = new InputStreamReader(EntityEventPolicy.class.getClassLoader()
+        .getResourceAsStream("/" + filename), StandardCharsets.UTF_8); BufferedReader in = new BufferedReader(
+        inputStreamReader)
+    ) {
+
+      String line;
       while ((line = in.readLine()) != null) {
         data.append(line);
       }