1 files changed, 206 insertions, 0 deletions
diff --git a/ecomp-sdk/epsdk-analytics/src/main/java/org/openecomp/portalsdk/analytics/system/fusion/controller/FileServletController.java b/ecomp-sdk/epsdk-analytics/src/main/java/org/openecomp/portalsdk/analytics/system/fusion/controller/FileServletController.java
new file mode 100644
index 00000000..203ee9cc
--- /dev/null
+++ b/ecomp-sdk/epsdk-analytics/src/main/java/org/openecomp/portalsdk/analytics/system/fusion/controller/FileServletController.java
@@ -0,0 +1,206 @@
+/*-
+ * ================================================================================
+ * eCOMP Portal SDK
+ * ================================================================================
+ * Copyright (C) 2017 AT&T Intellectual Property
+ * ================================================================================
+ * 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.
+ * ================================================================================
+ */
+package org.openecomp.portalsdk.analytics.system.fusion.controller;
+/**
+ * Raptor Blob Extract Servlet
+ * 
+ */
+
+import java.io.BufferedInputStream;
+import java.io.BufferedOutputStream;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.sql.Blob;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+import org.openecomp.portalsdk.core.logging.logic.EELFLoggerDelegate;
+import org.openecomp.portalsdk.core.service.DataAccessService;
+import org.springframework.web.servlet.ModelAndView;;
+
+
+public class FileServletController  {
+
+	private DataAccessService dataAccessService;
+
+	EELFLoggerDelegate logger = EELFLoggerDelegate.getLogger(FileServletController.class);
+
+
+	public ModelAndView handleRequestInternal(HttpServletRequest request,
+			HttpServletResponse response) throws Exception {
+		logger.debug(EELFLoggerDelegate.debugLogger, ("FileServletController:: f=" + request.getParameter("f")));
+
+		String fname = request.getParameter("f");
+
+		try {
+			Map params = new HashMap();
+			params.put("fname", fname);
+
+			logger.debug(EELFLoggerDelegate.debugLogger, ("executing query: select file_blob from cr_report_file_history where file_name = :"
+					+ fname));
+
+			List<Object> fileFromDB = (List<Object>) getDataAccessService().executeNamedQuery("getFileWithName", params, null);
+
+			byte[] allBytesInBlob = null;
+
+			if (fileFromDB != null && fileFromDB.size() > 0) {
+
+				logger.debug(EELFLoggerDelegate.debugLogger, ("reading file blob from DB..."));
+				try {
+					
+		            /*for weblogic setup
+		             * if(Globals.isWeblogicServer()) {
+		            	weblogic.jdbc.vendor.oracle.OracleThinBlob aBlob = (weblogic.jdbc.vendor.oracle.OracleThinBlob) ((org.hibernate.lob.SerializableBlob) fileFromDB
+								.get(0)).getWrappedBlob();
+		            	InputStream inBlob = ((java.sql.Blob) aBlob).getBinaryStream();
+		            	ByteArrayOutputStream baos = new ByteArrayOutputStream();
+		            	byte[] buf = new byte[1024];
+		            	int n = 0;
+		            	while ((n=inBlob.read(buf))>=0) {
+		            		baos.write(buf, 0, n);
+		            	}
+		            	inBlob.close();
+		            	allBytesInBlob = baos.toByteArray();
+		             } else { */
+					/* works in Hinernate3	[	oracle.sql.BLOB aBlob = (oracle.sql.BLOB) ((org.hibernate.lob.SerializableBlob) fileFromDB
+									.get(0)).getWrappedBlob();
+							allBytesInBlob = aBlob.getBytes(1, (int) aBlob.length()); ] */
+		            // }
+					
+					Object fileFromDBType = fileFromDB.get(0);
+					if(fileFromDBType instanceof byte[] ) // postgres
+						allBytesInBlob = (byte[]) fileFromDB.get(0);
+					else if (fileFromDBType instanceof Blob ) // oracle
+						allBytesInBlob = ((Blob) fileFromDB.get(0)).getBytes(1, (int) ((Blob) fileFromDB.get(0)).length());
+					
+					
+					
+				} catch (Exception e) {
+					logger.error(EELFLoggerDelegate.debugLogger, ("An exception has occurred: " + e.getMessage()));
+					throw (e);
+				}
+
+			} else {
+				logger.error(EELFLoggerDelegate.debugLogger, ("ERROR: No BLOB returned from DB..."));
+				throw (new Exception("ERROR: No BLOB returned from DB..."));
+			}
+
+			serveFile(response, allBytesInBlob, fname);
+			return null;
+		} catch (Exception e) {
+			logger.error(EELFLoggerDelegate.debugLogger, ("Exception occurred..." + e.getMessage()));
+			Map<String, Object> errView = new HashMap<String, Object>();
+			errView.put("error", "The requested resource was not found.");
+			//return new ModelAndView(getExceptionView(), "model", errView);
+			return null;
+		}
+
+	}
+
+	private void serveFile(HttpServletResponse response, File inFile)
+			throws Exception {
+		OutputStream os = null;
+		InputStream is = null;
+		try {
+			response.reset();
+			is = new BufferedInputStream(new FileInputStream(inFile));
+			os = new BufferedOutputStream(response.getOutputStream());
+			response.setContentLength((int) inFile.length());
+			response.setContentType("application/octet-stream");
+			response.setHeader("Content-disposition", "attachment; filename=\""
+					+ inFile.getName() + "\"");
+			copyStream(is, os);
+			os.flush();
+		} catch (Exception ex) {
+			if (os == null)
+				throw new Exception("Could not open output stream for file ");
+			if (is == null)
+				throw new Exception("Could not open input stream for file ");
+		} finally {
+			if (os != null) {
+				os.close();
+			}
+			if (is != null)
+				is.close();
+		}
+	}
+
+	private void serveFile(HttpServletResponse response, byte[] outStream,
+			String name) throws Exception {
+		OutputStream os = null;
+		InputStream is = null;
+		try {
+			response.reset();
+			response.setContentLength((int) outStream.length);
+			response.setContentType("application/octet-stream");
+			response.setHeader("Content-disposition", "attachment; filename=\""
+					+ name + "\"");
+			copyStream(response, outStream);
+		} catch (Exception ex) {
+			if (os == null)
+				throw new Exception("Could not open output stream for file ");
+			if (is == null)
+				throw new Exception("Could not open input stream for file ");
+		} finally {
+			if (os != null) {
+				os.close();
+			}
+			if (is != null)
+				is.close();
+		}
+	}
+
+	private int copyStream(InputStream in, OutputStream out) throws IOException {
+		int bytes, totalBytes = 0;
+
+		byte[] b = new byte[4096];
+
+		while ((bytes = in.read(b, 0, b.length)) != -1) {
+			totalBytes += bytes;
+			out.write(b, 0, bytes);
+		}
+		return totalBytes;
+	}
+
+	private int copyStream(HttpServletResponse response, byte[] outStream)
+			throws IOException {
+		
+		OutputStream os = new BufferedOutputStream(response.getOutputStream());
+		os.write(outStream);
+		os.flush();
+		return outStream.length;
+	}
+
+	public DataAccessService getDataAccessService() {
+		return dataAccessService;
+	}
+
+	public void setDataAccessService(DataAccessService dataAccessService) {
+		this.dataAccessService = dataAccessService;
+	}
+
+}