Adding UI extensibility

[aai/sparky-be.git] / src / main / java / org / onap / aai / sparky / sync / task / PerformElasticSearchPut.java

diff --git a/src/main/java/org/onap/aai/sparky/synchronizer/task/PerformElasticSearchPut.java b/src/main/java/org/onap/aai/sparky/sync/task/PerformElasticSearchPut.java
similarity index 58%
rename from src/main/java/org/onap/aai/sparky/synchronizer/task/PerformElasticSearchPut.java
rename to src/main/java/org/onap/aai/sparky/sync/task/PerformElasticSearchPut.java
index e1bbd56..6359dbe 100644 (file)
--- a/src/main/java/org/onap/aai/sparky/synchronizer/task/PerformElasticSearchPut.java
+++ b/src/main/java/org/onap/aai/sparky/sync/task/PerformElasticSearchPut.java
@@ -20,14 +20,16 @@
  *
  * ECOMP is a trademark and service mark of AT&T Intellectual Property.
  */
-package org.onap.aai.sparky.synchronizer.task;
+package org.onap.aai.sparky.sync.task;
 
 import java.util.Map;
 import java.util.function.Supplier;
 
+import javax.ws.rs.core.MediaType;
+
+import org.onap.aai.restclient.client.OperationResult;
+import org.onap.aai.sparky.dal.ElasticSearchAdapter;
 import org.onap.aai.sparky.dal.NetworkTransaction;
-import org.onap.aai.sparky.dal.rest.OperationResult;
-import org.onap.aai.sparky.dal.rest.RestDataProvider;
 import org.slf4j.MDC;
 
 /**
@@ -35,7 +37,7 @@ import org.slf4j.MDC;
  */
 public class PerformElasticSearchPut implements Supplier<NetworkTransaction> {
 
-  private RestDataProvider restDataProvider;
+  private ElasticSearchAdapter esAdapter;
   private String jsonPayload;
   private NetworkTransaction txn;
   private Map<String, String> contextMap;
@@ -48,18 +50,18 @@ public class PerformElasticSearchPut implements Supplier<NetworkTransaction> {
    * @param restDataProvider the rest data provider
    */
   public PerformElasticSearchPut(String jsonPayload, NetworkTransaction txn,
-      RestDataProvider restDataProvider) {
+      ElasticSearchAdapter esAdapter) {
     this.jsonPayload = jsonPayload;
     this.txn = txn;
-    this.restDataProvider = restDataProvider;
+    this.esAdapter = esAdapter;
     this.contextMap = MDC.getCopyOfContextMap();
   }
 
   public PerformElasticSearchPut(String jsonPayload, NetworkTransaction txn,
-      RestDataProvider restDataProvider, Map<String, String> contextMap) {
+      ElasticSearchAdapter esAdapter, Map<String, String> contextMap) {
     this.jsonPayload = jsonPayload;
     this.txn = txn;
-    this.restDataProvider = restDataProvider;
+    this.esAdapter = esAdapter;
     this.contextMap = contextMap;
   }
 
@@ -71,70 +73,16 @@ public class PerformElasticSearchPut implements Supplier<NetworkTransaction> {
   @Override
   public NetworkTransaction get() {
     txn.setTaskAgeInMs();
-    long startTimeInMs = System.currentTimeMillis();
     MDC.setContextMap(contextMap);
 
-    OperationResult or = restDataProvider.doPut(txn.getLink(), jsonPayload, "application/json");
-
-    or.setResponseTimeInMs(System.currentTimeMillis() - startTimeInMs);
-    txn.setOperationResult(or);
-
-    return txn;
-  }
-
-  /**
-   * @return the restDataProvider
-   */
-  public RestDataProvider getRestDataProvider() {
-    return restDataProvider;
-  }
-
-  /**
-   * @param restDataProvider the restDataProvider to set
-   */
-  public void setRestDataProvider(RestDataProvider restDataProvider) {
-    this.restDataProvider = restDataProvider;
-  }
+    long startTimeInMs = System.currentTimeMillis();
 
-  /**
-   * @return the jsonPayload
-   */
-  public String getJsonPayload() {
-    return jsonPayload;
-  }
+    OperationResult or =
+        esAdapter.doPut(txn.getLink(), jsonPayload, MediaType.APPLICATION_JSON_TYPE);
 
-  /**
-   * @param jsonPayload the jsonPayload to set
-   */
-  public void setJsonPayload(String jsonPayload) {
-    this.jsonPayload = jsonPayload;
-  }
+    txn.setOperationResult(or);
+    txn.setOpTimeInMs(System.currentTimeMillis() - startTimeInMs);
 
-  /**
-   * @return the txn
-   */
-  public NetworkTransaction getTxn() {
     return txn;
   }
-
-  /**
-   * @param txn the txn to set
-   */
-  public void setTxn(NetworkTransaction txn) {
-    this.txn = txn;
-  }
-
-  /**
-   * @return the contextMap
-   */
-  public Map<String, String> getContextMap() {
-    return contextMap;
-  }
-
-  /**
-   * @param contextMap the contextMap to set
-   */
-  public void setContextMap(Map<String, String> contextMap) {
-    this.contextMap = contextMap;
-  }
 }