From a7ce98e167ae05c2ee00e39733b35521353218cd Mon Sep 17 00:00:00 2001
From: luliqiang <kidgrow>
Date: Fri, 26 Feb 2021 15:23:22 +0800
Subject: [PATCH] 增加模板读取的错误处理和生成SQL文件时的错误处理

---
 kidgrow-business/kidgrow-opration-center/kidgrow-opration-center-server/src/main/java/com/kidgrow/oprationcenter/controller/PartnerDockingInfoController.java |   56 ++++++++++++++++++++++++++++++--------------------------
 1 files changed, 30 insertions(+), 26 deletions(-)

diff --git a/kidgrow-business/kidgrow-opration-center/kidgrow-opration-center-server/src/main/java/com/kidgrow/oprationcenter/controller/PartnerDockingInfoController.java b/kidgrow-business/kidgrow-opration-center/kidgrow-opration-center-server/src/main/java/com/kidgrow/oprationcenter/controller/PartnerDockingInfoController.java
index b2c532a..89ce998 100644
--- a/kidgrow-business/kidgrow-opration-center/kidgrow-opration-center-server/src/main/java/com/kidgrow/oprationcenter/controller/PartnerDockingInfoController.java
+++ b/kidgrow-business/kidgrow-opration-center/kidgrow-opration-center-server/src/main/java/com/kidgrow/oprationcenter/controller/PartnerDockingInfoController.java
@@ -155,38 +155,42 @@
     public void downLoadDataBase(@RequestBody PartnerDockingInfo partnerDockingInfo, HttpServletResponse response) {
         String dbSql = partnerDockingInfoService.createSQL(partnerDockingInfo);
 
-        OutputStream out = null;
-        InputStream fileInputStream = null;
-        try {
-            //BASE64Decoder dencoder = new BASE64Decoder();
-            byte[] arr = dbSql.getBytes();
+        if ((dbSql  == null) ||(dbSql.isEmpty())) {
+            log.error("创建私有云数据库脚本出错!");
+        } else {
 
-            // 设置输出的格式
-            response.reset();
-            response.setContentType("application/x-msdownload");
-            response.addHeader("Content-Length", "" + arr.length);
-            response.addHeader("Content-Disposition", "attachment; filename=\"" + partnerDockingInfo.getHospitalName()+partnerDockingInfo.getDepartmentName() + ".sql\"");
+            OutputStream out = null;
+            InputStream fileInputStream = null;
+            try {
+                //BASE64Decoder dencoder = new BASE64Decoder();
+                byte[] arr = dbSql.getBytes();
+
+                // 设置输出的格式
+                response.reset();
+                response.setContentType("application/x-msdownload");
+                response.addHeader("Content-Length", "" + arr.length);
+                response.addHeader("Content-Disposition", "attachment; filename=\"" + partnerDockingInfo.getHospitalName() + partnerDockingInfo.getDepartmentName() + ".sql\"");
 
 
+                out = response.getOutputStream();
+                fileInputStream = new ByteArrayInputStream(arr);
+                byte[] buffer = new byte[1024 * 10];
+                int len = 0;
+                while ((len = fileInputStream.read(buffer)) > 0) {
+                    out.write(buffer, 0, len);
+                }
 
-            out = response.getOutputStream();
-            fileInputStream = new ByteArrayInputStream(arr);
-            byte[] buffer = new byte[1024 * 10] ;
-            int len = 0;
-            while ((len = fileInputStream.read(buffer)) > 0) {
-                out.write(buffer, 0, len);
-            }
+            } catch (Exception e) {
+                e.printStackTrace();
 
-        } catch (Exception e) {
-            e.printStackTrace();
+            } finally {
 
-        } finally {
-
-            if (fileInputStream != null) {
-                try {
-                    fileInputStream.close();
-                } catch (IOException e) {
-                    e.printStackTrace();
+                if (fileInputStream != null) {
+                    try {
+                        fileInputStream.close();
+                    } catch (IOException e) {
+                        e.printStackTrace();
+                    }
                 }
             }
         }

--
Gitblit v1.8.0