update sync queries to use searh data service
[aai/sparky-be.git] / sparkybe-onap-service / src / main / java / org / onap / aai / sparky / sync / task / PerformSearchServiceRetrieval.java
@@ -23,20 +23,18 @@ 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.search.SearchServiceAdapter;
 import org.onap.aai.sparky.dal.NetworkTransaction;
 import org.slf4j.MDC;
 
 /**
  * The Class PerformElasticSearchRetrieval.
  */
-public class PerformElasticSearchRetrieval implements Supplier<NetworkTransaction> {
+public class PerformSearchServiceRetrieval implements Supplier<NetworkTransaction> {
 
   private NetworkTransaction txn;
-  private ElasticSearchAdapter esAdapter;
+  private SearchServiceAdapter searchServiceAdapter;
   private Map<String, String> contextMap;
 
   /**
@@ -45,10 +43,10 @@ public class PerformElasticSearchRetrieval implements Supplier<NetworkTransactio
    * @param elasticSearchTxn the elastic search txn
    * @param restDataProvider the rest data provider
    */
-  public PerformElasticSearchRetrieval(NetworkTransaction elasticSearchTxn,
-      ElasticSearchAdapter esAdapter) {
+  public PerformSearchServiceRetrieval(NetworkTransaction elasticSearchTxn,
+                 SearchServiceAdapter searchServiceAdapter) {
     this.txn = elasticSearchTxn;
-    this.esAdapter = esAdapter;
+    this.searchServiceAdapter = searchServiceAdapter;
     this.contextMap = MDC.getCopyOfContextMap();
   }
 
@@ -59,7 +57,7 @@ public class PerformElasticSearchRetrieval implements Supplier<NetworkTransactio
   public NetworkTransaction get() {
        MDC.setContextMap(contextMap);
        long startTimeInMs = System.currentTimeMillis();
-    OperationResult or = esAdapter.doGet(txn.getLink(), MediaType.APPLICATION_JSON_TYPE);
+    OperationResult or = searchServiceAdapter.doGet(txn.getLink(), "application/json");
     txn.setOperationResult(or);
     txn.setOpTimeInMs(System.currentTimeMillis() - startTimeInMs);
     return txn;