1 files changed, 183 insertions, 0 deletions
diff --git a/mod2/catalog-service/src/main/java/org/onap/dcaegen2/platform/mod/web/service/policymodel/PolicyModelDistributionServiceImpl.java b/mod2/catalog-service/src/main/java/org/onap/dcaegen2/platform/mod/web/service/policymodel/PolicyModelDistributionServiceImpl.java
new file mode 100644
index 0000000..d0646d2
--- /dev/null
+++ b/mod2/catalog-service/src/main/java/org/onap/dcaegen2/platform/mod/web/service/policymodel/PolicyModelDistributionServiceImpl.java
@@ -0,0 +1,183 @@
+/*
+ * ============LICENSE_START=======================================================
+ *  org.onap.dcae
+ *  ================================================================================
+ *  Copyright (c) 2020 AT&T Intellectual Property. All rights reserved.
+ *  ================================================================================
+ *  Licensed under the Apache License, Version 2.0 (the "License");
+ *  you may not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ *  ============LICENSE_END=========================================================
+ */
+
+package org.onap.dcaegen2.platform.mod.web.service.policymodel;
+
+import io.netty.handler.ssl.SslContext;
+import io.netty.handler.ssl.SslContextBuilder;
+import io.netty.handler.ssl.util.InsecureTrustManagerFactory;
+import lombok.Setter;
+import lombok.extern.slf4j.Slf4j;
+import org.onap.dcaegen2.platform.mod.model.exceptions.policymodel.PolicyModelDistributionEnvNotFoundException;
+import org.onap.dcaegen2.platform.mod.model.policymodel.DistributionInfo;
+import org.onap.dcaegen2.platform.mod.model.policymodel.PolicyModel;
+import org.onap.dcaegen2.platform.mod.model.policymodel.PolicyModelDistributionEnv;
+import org.onap.dcaegen2.platform.mod.model.policymodel.PolicyModelStatus;
+import org.onap.dcaegen2.platform.mod.util.PolicyModelUtils;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpMethod;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
+import org.springframework.http.client.reactive.ReactorClientHttpConnector;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+import org.springframework.util.Base64Utils;
+import org.springframework.web.client.RestTemplate;
+import org.springframework.web.reactive.function.client.ClientResponse;
+import org.springframework.web.reactive.function.client.ExchangeFilterFunctions;
+import org.springframework.web.reactive.function.client.WebClient;
+import reactor.core.publisher.Mono;
+import reactor.netty.http.client.HttpClient;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.concurrent.atomic.AtomicBoolean;
+
+/**
+ * Policy Model Service implementation
+ */
+
+@Service
+@Setter
+@Slf4j
+public class PolicyModelDistributionServiceImpl implements PolicyModelDistributionService {
+
+    @Autowired
+    PolicyModelServiceImpl policyModelServiceImpl;
+
+    @Autowired
+    PolicyModelGateway policyModelGateway;
+
+    @Autowired
+    PolicyModelUtils policyModelUtils;
+
+    /**
+     * Gets the Policy Model Distribution by Policy Model ID
+     * @param env
+     * @param modelId
+     * @return
+     */
+    @Override
+    public ResponseEntity getPolicyModelDistributionById(String env,String modelId) {
+
+        String responseBody = null;
+        HttpStatus httpStatus;
+        String url = policyModelUtils.getPolicyEngineURL(env);
+        PolicyModel policyModel = policyModelServiceImpl.getPolicyModelById(modelId);
+
+        try{
+            WebClient webClient = policyModelUtils.getWebClient(env);
+            WebClient.RequestHeadersSpec<?> requestHeadersSpec = webClient.method(HttpMethod.GET).uri(url + "/" + policyModel.getName());
+            httpStatus = requestHeadersSpec.exchange().map(response -> response.statusCode()).block();
+            if (httpStatus.is2xxSuccessful()) {
+                responseBody = requestHeadersSpec.retrieve().bodyToMono(String.class).block();
+            } else if(httpStatus.is4xxClientError()){
+                if(httpStatus.value() == 401) {
+                    responseBody = "Authentication Error";
+                } else if(httpStatus.value() == 403) {
+                    responseBody = "Authorization Error";
+                } else if(httpStatus.value() == 404) {
+                    responseBody = "Resource Not Found";
+                }
+            } else if(httpStatus.is5xxServerError()){
+                responseBody = "Internal Server Error";
+            }
+            else {
+                responseBody = "Problem in getting the Policy Model.";
+            }
+        }catch(Exception ex){
+            log.error("Problem in getting the Policy Model.");
+            log.error("error: ", ex);
+            httpStatus = HttpStatus.INTERNAL_SERVER_ERROR;
+            responseBody = "Problem in getting the Policy Model.";
+
+        }
+
+        return ResponseEntity.status(httpStatus).body(responseBody);
+
+    }
+
+    /**
+     * Distributes a Policy Model
+     * @param env
+     * @param modelId
+     * @return
+     */
+    @Override
+    @Transactional
+    public ResponseEntity distributePolicyModel(String env, String modelId) {
+
+        String responseBody = null;
+        HttpStatus httpStatus;
+        String url = policyModelUtils.getPolicyEngineURL(env);
+        PolicyModel policyModel = policyModelServiceImpl.getPolicyModelById(modelId);
+        String content = policyModel.getContent();
+        DistributionInfo distributionInfo = DistributionInfo.builder().url(env).build();
+
+        try{
+            WebClient webClient = policyModelUtils.getWebClient(env);
+            WebClient.RequestHeadersSpec<?> requestHeadersSpec = webClient.method(HttpMethod.POST).uri(url).bodyValue(content.getBytes());
+            httpStatus = requestHeadersSpec.exchange().map(response -> response.statusCode()).block();
+            if (httpStatus.is2xxSuccessful()) {
+                distributionInfo.setStatus(PolicyModelStatus.SUCCESS);
+                responseBody = requestHeadersSpec.retrieve().bodyToMono(String.class).block();
+            } else if(httpStatus.is4xxClientError()){
+                distributionInfo.setStatus(PolicyModelStatus.FAILED);
+                if(httpStatus.value() == 400) {
+                    responseBody = "Invalid Body";
+                } else if(httpStatus.value() == 401) {
+                    responseBody = "Authentication Error";
+                } else if(httpStatus.value() == 403) {
+                    responseBody = "Authorization Error";
+                } else if(httpStatus.value() == 406) {
+                    responseBody = "Not Acceptable Version";
+                } else if(httpStatus.value() == 415) {
+                    responseBody = "UnSupported Media Type";
+                }
+            } else if(httpStatus.is5xxServerError()){
+                responseBody = "Internal Server Error";
+                distributionInfo.setStatus(PolicyModelStatus.FAILED);
+            }
+            else {
+                distributionInfo.setStatus(PolicyModelStatus.FAILED);
+                responseBody = "Problem in Distributing the Policy Model.";
+            }
+        }catch(Exception ex){
+            log.error("Problem in Distributing the Policy Model.");
+            log.error("error: ", ex);
+            distributionInfo.setStatus(PolicyModelStatus.FAILED);
+            httpStatus = HttpStatus.INTERNAL_SERVER_ERROR;
+            responseBody = "Problem in Distributing the Policy Model.";
+
+        }
+        List<DistributionInfo> distributionInfos = new ArrayList<>();
+        distributionInfos.add(distributionInfo);
+        policyModel.setDistributionInfo(distributionInfos);
+
+        policyModelGateway.save(policyModel);
+        return ResponseEntity.status(httpStatus).body(responseBody);
+
+    }
+
+
+
+}