code optimization

Crab2died · May 2, 2018 · a23fcbc · a23fcbc
1 parent d20df95
commit a23fcbc
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/main/java/com/github/crab2died/handler/SheetTemplateHandler.java b/src/main/java/com/github/crab2died/handler/SheetTemplateHandler.java
@@ -13,7 +13,8 @@
 public class SheetTemplateHandler {
 
     // 构建SheetTemplate
-    public static SheetTemplate sheetTemplateBuilder(String templatePath) throws Excel4JException {
+    public static SheetTemplate sheetTemplateBuilder(String templatePath)
+            throws Excel4JException {
         SheetTemplate sheetTemplate = new SheetTemplate();
         try {
             // 读取模板文件
@@ -31,11 +32,16 @@ public static SheetTemplate sheetTemplateBuilder(String templatePath) throws Exc
         return sheetTemplate;
     }
 
-    public static SheetTemplate sheetTemplateBuilder(InputStream is) throws Exception {
+    public static SheetTemplate sheetTemplateBuilder(InputStream is)
+            throws Excel4JException {
         SheetTemplate sheetTemplate = new SheetTemplate();
 
         // 读取模板文件
-        sheetTemplate.workbook = WorkbookFactory.create(is);
+        try {
+            sheetTemplate.workbook = WorkbookFactory.create(is);
+        } catch (IOException | InvalidFormatException e) {
+            throw new Excel4JException(e);
+        }
         return sheetTemplate;
     }