From a9631bf0505ac256d276b060dc05d99ea971d877 Mon Sep 17 00:00:00 2001 From: david-PC\david <kimi42345@gmail.com> Date: 星期二, 12 六月 2018 17:13:50 +0800 Subject: [PATCH] 修改了Engine的流程(第一次) --- source/my/frame/src/frame/file/office/Engine.java | 114 +++++++++++++++++++++++++++++++++++++++++++++++++------- 1 files changed, 99 insertions(+), 15 deletions(-) diff --git a/source/my/frame/src/frame/file/office/Engine.java b/source/my/frame/src/frame/file/office/Engine.java index b2268f8..7eafd16 100644 --- a/source/my/frame/src/frame/file/office/Engine.java +++ b/source/my/frame/src/frame/file/office/Engine.java @@ -1,10 +1,15 @@ package frame.file.office; import java.io.Console; +import java.io.File; +import java.io.IOException; +import java.io.PrintWriter; import java.net.URL; import java.net.URLEncoder; import java.util.List; +import java.util.Map; +import javax.servlet.ServletOutputStream; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; @@ -14,10 +19,25 @@ import com.sun.corba.se.impl.protocol.JIDLLocalCRDImpl; +import frame.call.writer.JsonWriter; +import frame.config.Configer; import frame.data.DataContext; +import frame.file.FileIO; import frame.file.FileIOContainer; +import frame.file.FileIOErrorType; +import frame.file.FileIOException; +import frame.file.FileIOItem; import frame.file.UploadResult; +import frame.file.db.DataBaseTranslator; +import frame.file.office.excel.DownloadResult; +import frame.file.office.excel.ExcelReader; +import frame.file.office.excel.ExcelReaders; +import frame.file.office.excel.ExcelWriter; import frame.file.processor.Monitor; +import frame.upload.FileItem; +import frame.upload.FileUploadException; +import frame.upload.disk.DiskFileItemFactory; +import frame.upload.servlet.ServletFileUpload; import frame.util.Util; public class Engine { @@ -25,11 +45,14 @@ private static Engine instance; private static Object locker = new Object(); private Logger logger; - private ExcelLoader excelLoader; - + private ExcelReader excelLoader; + + private static int MAX_CACHE_SIZE = 30 * 1024 * 1024; + private static int MAX_FILE_SIZE = 100 * 1024 * 1024; + private Engine() { logger = LoggerFactory.getLogger(this.getClass()); - excelLoader = new ExcelLoader(logger); + excelLoader = new ExcelReader(logger); } @@ -43,29 +66,34 @@ } } - public static void download(HttpServletResponse response, String iocode, DataContext dataContext, Map<String, String> parameterProvider) { + public static void download(HttpServletResponse response, String iocode, DataContext dataContext, Map<String, String> parameterProvider) throws IOException { getInstance(); StringBuilder log = new StringBuilder(); try { - new JIDLLocalCRDImpl(orb, scid, ior) FileIOContainer.load(); - ExcelWriter excelWriter = new ExcelWriter(Monitor.getDownloadProgressor(), parameterProvider); +// ExcelWriter excelWriter = new ExcelWriter(Monitor.getDownloadProgressor(), parameterProvider); + ExcelWriter excelWriter = new ExcelWriter(Monitor.getDownloadProgressor()); DownloadResult result = excelWriter.writeOneFile(iocode, dataContext); - + String filename = result.getFilename(); response.addHeader("Content-Type", "application/octet-stream"); filename = URLEncoder.encode(filename, "utf-8"); response.addHeader("Content-Disposition", "attachment;filename=" + filename); - response.TransmitFile(result.getPath()); + ServletOutputStream outputStream = response.getOutputStream(); + String path = result.getPath(); + File file = new File(path); + //TODO output +// outputStream.write(b) +// response.TransmitFile(result.getPath()); } catch (Exception e) { - JsonBuilder json = new JsonBuilder(); + JsonWriter json = new JsonWriter(); json.beginObject(); json.addValue("log", log.toString()); json.addValue("error", e.getMessage()); - + json.beginArray("detail"); List<String> details = Monitor.getDownloadProgressor().getMessageList(); @@ -75,19 +103,33 @@ json.endArray(); json.endObject(); - - response.Write(json.ToString()); + PrintWriter writer = response.getWriter(); + writer.write(json.toString()); } } - public static UploadResult upload(String iocode, FileInfo file, HttpServletRequest request) { + public static UploadResult upload(String ioCode, File file, HttpServletRequest request) throws Exception { UploadResult result = new UploadResult(request); result.writeBegin(); try { + //1. prepare getInstance(); FileIOContainer.load(); - instance.excelLoader.loadOneFile(result, iocode, file); + + //2. + if (Util.isEmptyStr(ioCode)) { + throw new FileIOException(FileIOErrorType.FileIOCodeIsNull, "file io code is null"); + } + + FileIO fileIO = FileIOContainer.get(ioCode); + + if (fileIO == null) { + throw new FileIOException(FileIOErrorType.FileIONotExists, "fileio not exists: " + ioCode); + } + + //3. + instance.doUpload(fileIO); } catch (Exception e){ result.fail("error", e.getMessage()); @@ -99,7 +141,49 @@ return result; } - public static UploadResult testDataProcess(String iocode, HttpServletRequest request) { + private void doUpload(FileIO fileIO) { + ExcelReaders excelLoaders = new ExcelReaders(request); + + //1. fire begin event + fileIO.execHandlers(result, "begin"); + + //2. file to tempt table + List<FileIOItem> importList = fileIO.getItemList(FileIOTypeCode.Import); + + for (FileIOItem fileIOItem : importList) { + fileIOItem.execHandlers(result, "begin"); + excelLoaders.loadOneItem(fileIOItem); + fileIOItem.execHandlers(result, "end"); + } + + //3. fire after tempt tables + fileIO.execHandlers(result, "after-temp"); + + //4. tempt table to standard table + List<FileIOItem> dbList = fileIO.getItemList(FileIOTypeCode.Db); + + for (FileIOItem fileIOItem : dbList) { + fileIOItem.execHandlers(result, "begin"); + DataBaseTranslator.translate(result, fileIOItem); + fileIOItem.execHandlers(result, "end"); + } + + //5. fire end event + fileIO.execHandlers(result, "end"); + } + + private static List<FileItem> loadUploadFile(HttpServletRequest request) throws FileUploadException { + + DiskFileItemFactory diskFileFactory = new DiskFileItemFactory(); + diskFileFactory.setSizeThreshold(MAX_CACHE_SIZE); + diskFileFactory.setRepository(new File(Configer.getParam("repository"))); + + ServletFileUpload fileUpload = new ServletFileUpload(diskFileFactory); + fileUpload.setSizeMax(MAX_FILE_SIZE); + return fileUpload.parseRequest(request); + } + + public static UploadResult testDataProcess(String iocode, HttpServletRequest request) throws Exception { UploadResult result = new UploadResult(request); result.writeBegin(); -- Gitblit v1.8.0