diff options
Diffstat (limited to 'dcae-analytics/dcae-analytics-tca-web/src/main/java/org/onap/dcae/analytics/tca/web/aai/TcaAaiEnrichmentContextImpl.java')
-rw-r--r-- | dcae-analytics/dcae-analytics-tca-web/src/main/java/org/onap/dcae/analytics/tca/web/aai/TcaAaiEnrichmentContextImpl.java | 20 |
1 files changed, 15 insertions, 5 deletions
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); } } |