X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=dcae-analytics%2Fdcae-analytics-tca-web%2Fsrc%2Fmain%2Fjava%2Forg%2Fonap%2Fdcae%2Fanalytics%2Ftca%2Fweb%2Faai%2FTcaAaiEnrichmentContextImpl.java;h=e1444ff415d397bb49f0415072b3fd036c555e17;hb=02b0b3fe6956e1a8b1f7a1f2c87f03443059d741;hp=73a600a0a7156039d6e64654cc56c8a816fb2b72;hpb=5869f5d2556b4fb8e31ac0544080f31eea16717e;p=dcaegen2%2Fanalytics%2Ftca-gen2.git

diff --git a/dcae-analytics/dcae-analytics-tca-web/src/main/java/org/onap/dcae/analytics/tca/web/aai/TcaAaiEnrichmentContextImpl.java b/dcae-analytics/dcae-analytics-tca-web/src/main/java/org/onap/dcae/analytics/tca/web/aai/TcaAaiEnrichmentContextImpl.java
index 73a600a..e1444ff 100644
--- a/dcae-analytics/dcae-analytics-tca-web/src/main/java/org/onap/dcae/analytics/tca/web/aai/TcaAaiEnrichmentContextImpl.java
+++ b/dcae-analytics/dcae-analytics-tca-web/src/main/java/org/onap/dcae/analytics/tca/web/aai/TcaAaiEnrichmentContextImpl.java
@@ -22,6 +22,10 @@ package org.onap.dcae.analytics.tca.web.aai;
 import org.onap.dcae.analytics.tca.core.service.TcaAaiEnrichmentContext;
 import org.onap.dcae.analytics.tca.core.service.TcaAaiEnrichmentService;
 import org.onap.dcae.analytics.tca.web.TcaAppProperties;
+import org.onap.dcae.analytics.tca.web.util.function.TcaAppPropsToAaiRestClientPrefsFunction;
+import org.onap.dcae.analytics.web.http.HttpClientPreferencesCustomizer;
+import org.springframework.boot.web.client.RestTemplateBuilder;
+import org.springframework.web.client.RestTemplate;
 
 /**
  * @author Rajiv Singla
@@ -29,15 +33,14 @@ import org.onap.dcae.analytics.tca.web.TcaAppProperties;
 public class TcaAaiEnrichmentContextImpl implements TcaAaiEnrichmentContext {
 
     private final TcaAppProperties tcaAppProperties;
-    private final TcaAaiEnrichmentService tcaAaiEnrichmentService;
+    private final RestTemplateBuilder restTemplateBuilder;
 
     public TcaAaiEnrichmentContextImpl(final TcaAppProperties tcaAppProperties,
-                                       final TcaAaiEnrichmentService tcaAaiEnrichmentService) {
+            final RestTemplateBuilder restTemplateBuilder) {
         this.tcaAppProperties = tcaAppProperties;
-        this.tcaAaiEnrichmentService = tcaAaiEnrichmentService;
+        this.restTemplateBuilder = restTemplateBuilder;
     }
 
-
     @Override
     public boolean isAaiEnrichmentEnabled() {
         return tcaAppProperties.getTca().getAai().getEnableEnrichment();
@@ -45,6 +48,13 @@ public class TcaAaiEnrichmentContextImpl implements TcaAaiEnrichmentContext {
 
     @Override
     public TcaAaiEnrichmentService getAaiEnrichmentService() {
-        return tcaAaiEnrichmentService;
+        TcaAaiRestClientPreferences aaiRestClientPreferences = new TcaAppPropsToAaiRestClientPrefsFunction()
+                .apply(tcaAppProperties);
+        if (aaiRestClientPreferences == null) {
+            return null;
+        }
+        RestTemplate aaiRestTemplate = restTemplateBuilder
+                .additionalCustomizers(new HttpClientPreferencesCustomizer<>(aaiRestClientPreferences)).build();
+        return new TcaAaiEnrichmentServiceImpl(tcaAppProperties, aaiRestTemplate);
     }
 }