2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2017-2018 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=========================================================
20 package org.onap.policy.pap.xacml.rest.service;
22 import java.io.BufferedWriter;
24 import java.io.FileOutputStream;
25 import java.io.IOException;
26 import java.io.InputStream;
27 import java.io.OutputStreamWriter;
28 import java.io.Writer;
29 import java.util.HashMap;
31 import java.util.UUID;
33 import javax.servlet.http.HttpServletRequest;
34 import javax.servlet.http.HttpServletResponse;
36 import org.onap.policy.common.logging.eelf.PolicyLogger;
37 import org.onap.policy.common.logging.flexlogger.FlexLogger;
38 import org.onap.policy.common.logging.flexlogger.Logger;
39 import org.onap.policy.pap.xacml.rest.components.CreateBRMSRuleTemplate;
40 import org.onap.policy.pap.xacml.rest.components.CreateNewMicroServiceModel;
41 import org.onap.policy.pap.xacml.rest.components.CreateNewOptimizationModel;
43 public class ImportService {
44 private static final Logger logger = FlexLogger.getLogger(ImportService.class);
45 private static String errorMessage = "Error in reading in file from API call";
46 private static String errorMsg = "error";
47 private static String operation = "operation";
48 private static String importHeader = "import";
49 private static String service = "service";
50 private static String extractDir = "ExtractDir";
51 private static String successMessage = "success";
52 private static String invalidServiceName = "Invalid ServiceName";
53 private static final String REGEX = "[0-9a-zA-Z._ ]*";
54 private static final String MISSING = "missing";
56 public void doImportMicroServicePut(HttpServletRequest request, HttpServletResponse response) {
57 String importServiceCreation = request.getParameter("importService");
58 String fileName = request.getParameter("fileName");
59 String version = request.getParameter("version");
60 String serviceName = request.getParameter("serviceName");
62 if(serviceName == null || serviceName.isEmpty() || !serviceName.matches(REGEX)){
63 response.setStatus(HttpServletResponse.SC_BAD_REQUEST);
64 response.addHeader(errorMsg, MISSING);
65 response.addHeader(operation, importHeader);
66 response.addHeader(service, invalidServiceName);
70 String description = request.getParameter("description");
71 Map<String, String> successMap = new HashMap<>();
72 if(("BRMSPARAM").equals(importServiceCreation)){
73 StringBuilder builder = new StringBuilder();
76 while((ch = request.getInputStream().read()) != -1){
77 builder.append((char)ch);
79 } catch (IOException e) {
81 PolicyLogger.error(errorMessage);
82 response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
83 response.addHeader(errorMsg, MISSING);
84 response.addHeader(operation, importHeader);
85 response.addHeader(service, serviceName);
87 CreateBRMSRuleTemplate brmsRuleTemplate = new CreateBRMSRuleTemplate();
88 successMap = brmsRuleTemplate.addRule(builder.toString(), serviceName, description, "API");
90 else if(("MICROSERVICE").equals(importServiceCreation)){
91 CreateNewMicroServiceModel newMS = null;
92 String randomID = UUID.randomUUID().toString();
94 if ( fileName != null) {
95 File extracDir = new File(extractDir);
96 if (!extracDir.exists()){
99 if (fileName.contains(".xmi") || fileName.contains(".yml")){
100 if(fileName.contains(".yml")){
103 // get the request content into a String
105 java.util.Scanner scanner;
107 scanner = new java.util.Scanner(request.getInputStream());
108 scanner.useDelimiter("\\A");
109 xmi = scanner.hasNext() ? scanner.next() : "";
111 } catch (IOException e1) {
113 PolicyLogger.error(errorMessage);
116 PolicyLogger.info("Request from API to import new Service");
117 try (Writer writer = new BufferedWriter(new OutputStreamWriter(
118 new FileOutputStream(extractDir + File.separator + randomID+type), "utf-8"))) {
120 } catch (IOException e) {
122 PolicyLogger.error(errorMessage);
126 InputStream inputStream = null;
127 try(FileOutputStream outputStream = new FileOutputStream(extractDir + File.separator + randomID+".zip")) {
128 inputStream = request.getInputStream();
129 byte[] buffer = new byte[4096];
131 while ((bytesRead = inputStream.read(buffer)) != -1) {
132 outputStream.write(buffer, 0, bytesRead) ;
134 } catch (IOException e) {
135 PolicyLogger.error("Error in reading in Zip File from API call"+e);
139 if(inputStream != null){
142 } catch (IOException e) {
143 PolicyLogger.error("Exception Occured while closing the input/output stream"+e);
148 newMS = new CreateNewMicroServiceModel(fileName, serviceName, "API", version, randomID);
150 successMap = newMS.addValuesToNewModel(type);
152 if (successMap.containsKey(successMessage)) {
154 successMap = newMS.saveImportService();
157 } else if(("OPTIMIZATION").equals(importServiceCreation)){
158 CreateNewOptimizationModel newOOF = null;
159 String randomID = UUID.randomUUID().toString();
160 if ( fileName != null) {
161 File extracDir = new File(extractDir);
162 if (!extracDir.exists()){
166 String type = ".yml";
168 // get the request content into a String
170 try (java.util.Scanner scanner = new java.util.Scanner(request.getInputStream());){
171 scanner.useDelimiter("\\A");
172 yml = scanner.hasNext() ? scanner.next() : "";
173 } catch (IOException e1) {
175 PolicyLogger.error(errorMessage);
178 PolicyLogger.info("Request from API to import new Optimization Service Model");
179 try (Writer writer = new BufferedWriter(new OutputStreamWriter(
180 new FileOutputStream(extractDir + File.separator + randomID+type), "utf-8"))) {
182 } catch (IOException e) {
184 PolicyLogger.error(errorMessage);
188 newOOF = new CreateNewOptimizationModel(fileName, serviceName, "API Import Service", version, randomID);
189 successMap = newOOF.addValuesToNewModel();
190 if (successMap.containsKey(successMessage)) {
192 successMap = newOOF.saveImportService();
197 // return a response to the PAP
198 if (successMap.containsKey(successMessage)) {
199 response.setStatus(HttpServletResponse.SC_OK);
200 response.addHeader("successMapKey", successMessage);
201 response.addHeader(operation, importHeader);
202 response.addHeader(service, serviceName);
203 } else if (successMap.containsKey("DBError")) {
204 if (successMap.get("DBError").contains("EXISTS")){
205 response.setStatus(HttpServletResponse.SC_CONFLICT);
206 response.addHeader(service, serviceName);
207 response.addHeader(errorMsg, "modelExistsDB");
209 response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
210 response.addHeader(errorMsg, "importDB");
212 response.addHeader(operation, importHeader);
213 response.addHeader(service, serviceName);
214 }else if (successMap.get(errorMsg).contains("MISSING")){
215 response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
216 response.addHeader(errorMsg, MISSING);
217 response.addHeader(operation, importHeader);
218 response.addHeader(service, serviceName);
219 }else if (successMap.get(errorMsg).contains("VALIDATION")){
220 response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
221 response.addHeader(errorMsg, "validation");
222 response.addHeader(operation, importHeader);
223 response.addHeader(service, serviceName);