dcae-analytics/dcae-analytics-tca-web/src/main/java/org/onap/dcae/analytics/tca/web/aai/TcaAaiEnrichmentContextImpl.java

   1 /*
   2  * ================================================================================
   3  * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
   4  * ================================================================================
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *      http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  * ============LICENSE_END=========================================================
  17  *
  18  */
  19
  20 package org.onap.dcae.analytics.tca.web.aai;
  21
  22 import org.onap.dcae.analytics.tca.core.service.TcaAaiEnrichmentContext;
  23 import org.onap.dcae.analytics.tca.core.service.TcaAaiEnrichmentService;
  24 import org.onap.dcae.analytics.tca.web.TcaAppProperties;
  25 import org.onap.dcae.analytics.tca.web.util.function.TcaAppPropsToAaiRestClientPrefsFunction;
  26 import org.onap.dcae.analytics.web.http.HttpClientPreferencesCustomizer;
  27 import org.springframework.boot.web.client.RestTemplateBuilder;
  28 import org.springframework.web.client.RestTemplate;
  29
  30 /**
  31  * @author Rajiv Singla
  32  */
  33 public class TcaAaiEnrichmentContextImpl implements TcaAaiEnrichmentContext {
  34
  35     private final TcaAppProperties tcaAppProperties;
  36     private final RestTemplateBuilder restTemplateBuilder;
  37
  38     public TcaAaiEnrichmentContextImpl(final TcaAppProperties tcaAppProperties,
  39             final RestTemplateBuilder restTemplateBuilder) {
  40         this.tcaAppProperties = tcaAppProperties;
  41         this.restTemplateBuilder = restTemplateBuilder;
  42     }
  43
  44     @Override
  45     public boolean isAaiEnrichmentEnabled() {
  46         return tcaAppProperties.getTca().getAai().getEnableEnrichment();
  47     }
  48
  49     @Override
  50     public TcaAaiEnrichmentService getAaiEnrichmentService() {
  51         TcaAaiRestClientPreferences aaiRestClientPreferences = new TcaAppPropsToAaiRestClientPrefsFunction()
  52                 .apply(tcaAppProperties);
  53         if (aaiRestClientPreferences == null) {
  54             return null;
  55         }
  56         RestTemplate aaiRestTemplate = restTemplateBuilder
  57                 .additionalCustomizers(new HttpClientPreferencesCustomizer<>(aaiRestClientPreferences)).build();
  58         return new TcaAaiEnrichmentServiceImpl(tcaAppProperties, aaiRestTemplate);
  59     }
  60 }