2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2017-2019 AT&T Intellectual Property. All rights reserved.
6 * ================================================================================
7 * Licensed under the Apache License, Version 2.0 (the "License");
8 * you may not use this file except in compliance with the License.
9 * You may obtain a copy of the License at
11 * http://www.apache.org/licenses/LICENSE-2.0
13 * Unless required by applicable law or agreed to in writing, software
14 * distributed under the License is distributed on an "AS IS" BASIS,
15 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * See the License for the specific language governing permissions and
17 * limitations under the License.
18 * ============LICENSE_END=========================================================
21 package org.onap.policy.pap.xacml.rest.service;
23 import java.io.BufferedWriter;
25 import java.io.FileOutputStream;
26 import java.io.IOException;
27 import java.io.InputStream;
28 import java.io.OutputStreamWriter;
29 import java.io.Writer;
30 import java.util.HashMap;
32 import java.util.UUID;
33 import javax.servlet.http.HttpServletRequest;
34 import javax.servlet.http.HttpServletResponse;
35 import org.onap.policy.common.logging.eelf.PolicyLogger;
36 import org.onap.policy.common.logging.flexlogger.FlexLogger;
37 import org.onap.policy.common.logging.flexlogger.Logger;
38 import org.onap.policy.pap.xacml.rest.components.CreateBRMSRuleTemplate;
39 import org.onap.policy.pap.xacml.rest.components.CreateNewMicroServiceModel;
40 import org.onap.policy.pap.xacml.rest.components.CreateNewOptimizationModel;
42 public class ImportService {
43 private static final Logger logger = FlexLogger.getLogger(ImportService.class);
44 private static String errorMessage = "Error in reading in file from API call";
45 private static String errorMsg = "error";
46 private static String operation = "operation";
47 private static String importHeader = "import";
48 private static String service = "service";
49 private static String extractDir = "ExtractDir";
50 private static String successMessage = "success";
51 private static String invalidServiceName = "Invalid ServiceName";
52 private static final String REGEX = "[0-9a-zA-Z._ ]*";
53 private static final String MISSING = "missing";
55 public void doImportMicroServicePut(HttpServletRequest request, HttpServletResponse response) {
56 String importServiceCreation = request.getParameter("importService");
57 String fileName = request.getParameter("fileName");
58 String version = request.getParameter("version");
59 String serviceName = request.getParameter("serviceName");
61 if (serviceName == null || serviceName.isEmpty() || !serviceName.matches(REGEX)) {
62 response.setStatus(HttpServletResponse.SC_BAD_REQUEST);
63 response.addHeader(errorMsg, MISSING);
64 response.addHeader(operation, importHeader);
65 response.addHeader(service, invalidServiceName);
69 String description = request.getParameter("description");
70 Map<String, String> successMap = new HashMap<>();
71 if (("BRMSPARAM").equals(importServiceCreation)) {
72 StringBuilder builder = new StringBuilder();
75 while ((ch = request.getInputStream().read()) != -1) {
76 builder.append((char) ch);
78 } catch (IOException e) {
80 PolicyLogger.error(errorMessage);
81 response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
82 response.addHeader(errorMsg, MISSING);
83 response.addHeader(operation, importHeader);
84 response.addHeader(service, serviceName);
86 CreateBRMSRuleTemplate brmsRuleTemplate = new CreateBRMSRuleTemplate();
88 brmsRuleTemplate.addRule(builder.toString(), serviceName, description, "API");
89 } else if (("MICROSERVICE").equals(importServiceCreation)) {
90 CreateNewMicroServiceModel newMS = null;
91 String randomID = UUID.randomUUID().toString();
93 if (fileName != null) {
94 File extracDir = new File(extractDir);
95 if (!extracDir.exists()) {
98 if (fileName.contains(".xmi") || fileName.contains(".yml")) {
99 if (fileName.contains(".yml")) {
102 // get the request content into a String
104 java.util.Scanner scanner;
106 scanner = new java.util.Scanner(request.getInputStream());
107 scanner.useDelimiter("\\A");
108 xmi = scanner.hasNext() ? scanner.next() : "";
110 } catch (IOException e1) {
112 PolicyLogger.error(errorMessage);
115 PolicyLogger.info("Request from API to import new Service");
116 try (Writer writer = new BufferedWriter(new OutputStreamWriter(
117 new FileOutputStream(extractDir + File.separator + randomID + type),
120 } catch (IOException e) {
122 PolicyLogger.error(errorMessage);
126 InputStream inputStream = null;
127 try (FileOutputStream outputStream =
128 new FileOutputStream(extractDir + File.separator + randomID + ".zip")) {
129 inputStream = request.getInputStream();
130 byte[] buffer = new byte[4096];
132 while ((bytesRead = inputStream.read(buffer)) != -1) {
133 outputStream.write(buffer, 0, bytesRead);
135 } catch (IOException e) {
136 PolicyLogger.error("Error in reading in Zip File from API call" + e);
140 if (inputStream != null) {
143 } catch (IOException e) {
145 "Exception Occured while closing the input/output stream" + e);
150 newMS = new CreateNewMicroServiceModel(fileName, serviceName, "API", version,
153 successMap = newMS.addValuesToNewModel(type);
155 if (successMap.containsKey(successMessage)) {
157 successMap = newMS.saveImportService();
160 } else if (("OPTIMIZATION").equals(importServiceCreation)) {
161 CreateNewOptimizationModel newOOF = null;
162 String randomID = UUID.randomUUID().toString();
163 if (fileName != null) {
164 File extracDir = new File(extractDir);
165 if (!extracDir.exists()) {
169 String type = ".yml";
171 // get the request content into a String
173 try (java.util.Scanner scanner = new java.util.Scanner(request.getInputStream());) {
174 scanner.useDelimiter("\\A");
175 yml = scanner.hasNext() ? scanner.next() : "";
176 } catch (IOException e1) {
178 PolicyLogger.error(errorMessage);
181 PolicyLogger.info("Request from API to import new Optimization Service Model");
182 try (Writer writer = new BufferedWriter(new OutputStreamWriter(
183 new FileOutputStream(extractDir + File.separator + randomID + type),
186 } catch (IOException e) {
188 PolicyLogger.error(errorMessage);
192 newOOF = new CreateNewOptimizationModel(fileName, serviceName, "API Import Service",
194 successMap = newOOF.addValuesToNewModel();
195 if (successMap.containsKey(successMessage)) {
197 successMap = newOOF.saveImportService();
202 // return a response to the PAP
203 if (successMap.containsKey(successMessage)) {
204 response.setStatus(HttpServletResponse.SC_OK);
205 response.addHeader("successMapKey", successMessage);
206 response.addHeader(operation, importHeader);
207 response.addHeader(service, serviceName);
208 } else if (successMap.containsKey("DBError")) {
209 if (successMap.get("DBError").contains("EXISTS")) {
210 response.setStatus(HttpServletResponse.SC_CONFLICT);
211 response.addHeader(service, serviceName);
212 response.addHeader(errorMsg, "modelExistsDB");
214 response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
215 response.addHeader(errorMsg, "importDB");
217 response.addHeader(operation, importHeader);
218 response.addHeader(service, serviceName);
219 } else if (successMap.get(errorMsg).contains("MISSING")) {
220 response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
221 response.addHeader(errorMsg, MISSING);
222 response.addHeader(operation, importHeader);
223 response.addHeader(service, serviceName);
224 } else if (successMap.get(errorMsg).contains("VALIDATION")) {
225 response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
226 response.addHeader(errorMsg, "validation");
227 response.addHeader(operation, importHeader);
228 response.addHeader(service, serviceName);