1 files changed, 62 insertions, 9 deletions

diff --git a/rest-services/cbs-client/src/main/java/org/onap/dcaegen2/services/sdk/rest/services/cbs/client/impl/CbsClientConfigMap.java b/rest-services/cbs-client/src/main/java/org/onap/dcaegen2/services/sdk/rest/services/cbs/client/impl/CbsClientConfigMap.java
index 42f53616..75577318 100644
--- a/rest-services/cbs-client/src/main/java/org/onap/dcaegen2/services/sdk/rest/services/cbs/client/impl/CbsClientConfigMap.java
+++ b/rest-services/cbs-client/src/main/java/org/onap/dcaegen2/services/sdk/rest/services/cbs/client/impl/CbsClientConfigMap.java
@@ -3,6 +3,7 @@
  * DCAEGEN2-SERVICES-SDK
  * ================================================================================
  * Copyright (C) 2021 Nokia. All rights reserved.
+ * Copyright (C) 2021 Wipro Limited.
  * ================================================================================
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -17,11 +18,15 @@
  * limitations under the License.
  * ============LICENSE_END=========================================================
  */
+
 package org.onap.dcaegen2.services.sdk.rest.services.cbs.client.impl;
 
 import com.google.gson.Gson;
 import com.google.gson.GsonBuilder;
 import com.google.gson.JsonObject;
+
+import java.util.LinkedHashMap;
+
 import org.jetbrains.annotations.NotNull;
 import org.onap.dcaegen2.services.sdk.rest.services.cbs.client.api.CbsClient;
 import org.onap.dcaegen2.services.sdk.rest.services.cbs.client.model.CbsRequest;
@@ -30,23 +35,33 @@ import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.yaml.snakeyaml.Yaml;
 import reactor.core.publisher.Mono;
-import java.util.LinkedHashMap;
 
 public class CbsClientConfigMap implements CbsClient {
 
     private static final Logger LOGGER = LoggerFactory.getLogger(CbsClientConfigMap.class);
     private final String configMapFilePath;
+    private String policySyncFilePath = "";
+    private String appName = "";
 
+    public CbsClientConfigMap(String configMapFilePath) {
+        this.configMapFilePath = configMapFilePath;
+    }
 
-    public CbsClientConfigMap (String configMapFilePath) {
+    public CbsClientConfigMap(String configMapFilePath, String policySyncFilePath, String appName) {
         this.configMapFilePath = configMapFilePath;
+        this.policySyncFilePath = policySyncFilePath;
+        this.appName = appName;
     }
 
     @Override
     public @NotNull Mono<JsonObject> get(CbsRequest request) {
-        return Mono.just(this.loadConfigMapFile())
-                .map(CbsClientEnvironmentParsing::processEnvironmentVariables)
-                .doOnNext(this::logConfigMapOutput);
+        Mono<JsonObject> configJsonMono =
+                Mono.just(this.loadConfigMapFile()).map(CbsClientEnvironmentParsing::processEnvironmentVariables);
+        if (this.shouldReadPolicySyncFile(request)) {
+
+            return configJsonMono.map(this::loadPolicySyncFile).doOnNext(this::logConfigMapOutput);
+        }
+        return configJsonMono.doOnNext(this::logConfigMapOutput);
     }
 
     public boolean verifyConfigMapFile() {
@@ -54,7 +69,7 @@ public class CbsClientConfigMap implements CbsClient {
             LOGGER.info("Trying to load configuration from configMap file: {}", configMapFilePath);
             this.loadConfigMapFile().isJsonObject();
             return true;
-        } catch(Exception ex) {
+        } catch (Exception ex) {
             this.logConfigMapError(ex);
             return false;
         }
@@ -69,8 +84,6 @@ public class CbsClientConfigMap implements CbsClient {
         return new Yaml().load(new FileReader(configMapFilePath).getContent());
     }
 
-
-
     private void logConfigMapOutput(JsonObject jsonObject) {
         LOGGER.info("Got successful output from ConfigMap file");
         LOGGER.debug("ConfigMap output: {}", jsonObject);
@@ -79,4 +92,44 @@ public class CbsClientConfigMap implements CbsClient {
     private void logConfigMapError(Exception ex) {
         LOGGER.error("Error loading configuration from configMap file: {}", ex.getMessage());
     }
-}
\ No newline at end of file
+
+    private boolean shouldReadPolicySyncFile(CbsRequest request) {
+        try {
+            return request.requestPath()
+                          .getForService(appName)
+                          .contains("service_component_all");
+        } catch (Exception ex) {
+            LOGGER.error("Error finding requestPath", ex.getMessage());
+            return false;
+        }
+    }
+
+    private JsonObject loadPolicySyncFile(JsonObject configJsonObject) {
+
+        try {
+
+            if (new FileReader(policySyncFilePath).doesFileExists()) {
+                LOGGER.info("PolicySync file is present");
+                Gson gson = new GsonBuilder().create();
+                JsonObject policyJsonObject = gson.fromJson(this.loadJsonStringPolicySyncFile(), JsonObject.class);
+                policyJsonObject.add("config", configJsonObject);
+                return policyJsonObject;
+            }
+            LOGGER.info("PolicySync file does not exist");
+            JsonObject policyJsonObject = new JsonObject();
+            policyJsonObject.add("config", configJsonObject);
+            return policyJsonObject;
+
+        } catch (Exception ex) {
+            LOGGER.error("PolicySync file does not contain a valid json");
+            JsonObject policyJsonObject = new JsonObject();
+            policyJsonObject.add("config", configJsonObject);
+            return policyJsonObject;
+        }
+
+    }
+
+    private String loadJsonStringPolicySyncFile() {
+        return new FileReader(policySyncFilePath).getContent();
+    }
+}