src/main/java/org/onap/aai/sparky/sync/task/PerformActiveInventoryRetrieval.java

   1 /**
   2  * ============LICENSE_START=======================================================
   3  * org.onap.aai
   4  * ================================================================================
   5  * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
   6  * Copyright © 2017-2018 Amdocs
   7  * ================================================================================
   8  * Licensed under the Apache License, Version 2.0 (the "License");
   9  * you may not use this file except in compliance with the License.
  10  * You may obtain a copy of the License at
  11  *
  12  *       http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing, software
  15  * distributed under the License is distributed on an "AS IS" BASIS,
  16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17  * See the License for the specific language governing permissions and
  18  * limitations under the License.
  19  * ============LICENSE_END=========================================================
  20  */
  21 package org.onap.aai.sparky.sync.task;
  22
  23 import java.util.Map;
  24 import java.util.function.Supplier;
  25
  26 import org.onap.aai.cl.api.Logger;
  27 import org.onap.aai.cl.eelf.LoggerFactory;
  28 import org.onap.aai.restclient.client.OperationResult;
  29 import org.onap.aai.sparky.dal.ActiveInventoryAdapter;
  30 import org.onap.aai.sparky.dal.NetworkTransaction;
  31 import org.onap.aai.sparky.logging.AaiUiMsgs;
  32 import org.slf4j.MDC;
  33
  34 /*
  35  * Consider abstraction the tasks into common elemnts, because most of them repeat a generic call
  36  * flow pattern
  37  */
  38
  39 /**
  40  * The Class PerformActiveInventoryRetrieval.
  41  */
  42 public class PerformActiveInventoryRetrieval implements Supplier<NetworkTransaction> {
  43
  44   private static Logger logger = LoggerFactory.getInstance().getLogger(PerformActiveInventoryRetrieval.class);
  45
  46   private NetworkTransaction txn;
  47   private ActiveInventoryAdapter aaiAdapter;
  48   private Map<String, String> contextMap;
  49
  50   /**
  51    * Instantiates a new perform active inventory retrieval.
  52    *
  53    * @param txn the txn
  54    * @param aaiProvider the aai provider
  55    */
  56   public PerformActiveInventoryRetrieval(NetworkTransaction txn,
  57       ActiveInventoryAdapter aaiAdapter) {
  58     this.txn = txn;
  59     this.aaiAdapter = aaiAdapter;
  60     this.contextMap = MDC.getCopyOfContextMap();
  61   }
  62
  63   /* (non-Javadoc)
  64    * @see java.util.function.Supplier#get()
  65    */
  66   @Override
  67   public NetworkTransaction get() {
  68
  69     txn.setTaskAgeInMs();
  70
  71     long startTimeInMs = System.currentTimeMillis();
  72     MDC.setContextMap(contextMap);
  73     OperationResult result = null;
  74     try {
  75
  76       final String absoluteSelfLink = aaiAdapter.repairSelfLink(txn.getLink(), txn.getQueryParameters());
  77       result = aaiAdapter.queryActiveInventoryWithRetries(absoluteSelfLink, "application/json", 5);
  78     } catch (Exception exc) {
  79       logger.error(AaiUiMsgs.ERROR_GENERIC,"Failure to resolve self link from AAI.  Error = " + exc.getMessage());
  80       result = new OperationResult(500,
  81           "Caught an exception while trying to resolve link = " + exc.getMessage());
  82     } finally {
  83       txn.setOperationResult(result);
  84       txn.setOpTimeInMs(System.currentTimeMillis() - startTimeInMs);
  85     }
  86
  87     return txn;
  88   }
  89
  90   protected void setContextMap(Map<String, String> contextMap) {
  91     this.contextMap = contextMap;
  92   }
  93 }