diff --git a/src/main/java/com/xqopen/kehui/exception/ErrorCode.java b/src/main/java/com/xqopen/kehui/exception/ErrorCode.java index 4c109dd..679a09c 100644 --- a/src/main/java/com/xqopen/kehui/exception/ErrorCode.java +++ b/src/main/java/com/xqopen/kehui/exception/ErrorCode.java @@ -103,7 +103,7 @@ public class ErrorCode { public static final int UPLOAD_FILE_EMPTY = 4501; public static final int UPLOAD_FILE_ERORR = 4502; - + public static final int UPLOAD_FILE_ERORR_FORMAT = 4503; //后勤管理[5000,5500) diff --git a/src/main/java/com/xqopen/kehui/util/FileUploadUtil.java b/src/main/java/com/xqopen/kehui/util/FileUploadUtil.java index 8889695..2b582fd 100644 --- a/src/main/java/com/xqopen/kehui/util/FileUploadUtil.java +++ b/src/main/java/com/xqopen/kehui/util/FileUploadUtil.java @@ -20,11 +20,17 @@ public class FileUploadUtil { if (file.isEmpty()) { throw new XQException(ErrorCode.UPLOAD_FILE_EMPTY, null); } - String filename = file.getOriginalFilename(); + String fileName = file.getOriginalFilename(); +// String[] fileNameList={".jpg",".png",".jpeg",".pdf",".docx",".doc",".txt",".xlsx",".xls",".rar",".zip",".7z"}; //后缀名固定 - if(!(filename.endsWith(".jpg")||filename.endsWith(".png")||filename.endsWith(".jpeg")||filename.endsWith(".pdf")||filename.endsWith(".docx")||filename.endsWith(".doc"))){ - throw new XQException(ErrorCode.UPLOAD_FILE_ERORR, null); + if(!(fileName.endsWith(".jpg")||fileName.endsWith(".png")||fileName.endsWith(".jpeg") + ||fileName.endsWith(".docx")||fileName.endsWith(".doc")||fileName.endsWith(".txt")||fileName.endsWith(".pdf") + ||fileName.endsWith(".wps")||fileName.endsWith(".wpt")||fileName.endsWith(".dot")||fileName.endsWith(".rtf") + ||fileName.endsWith(".xlsx")|fileName.endsWith(".xls")) + ||fileName.endsWith(".rar")||fileName.endsWith("zip")||fileName.endsWith(".7z")){ + throw new XQException(ErrorCode.UPLOAD_FILE_ERORR_FORMAT, null); } + StringBuffer sb = new StringBuffer(); String originalFileName = file.getOriginalFilename(); diff --git a/src/main/resources/i18n/messages_zh_CN.properties b/src/main/resources/i18n/messages_zh_CN.properties index d479005..a6d5bd5 100644 --- a/src/main/resources/i18n/messages_zh_CN.properties +++ b/src/main/resources/i18n/messages_zh_CN.properties @@ -44,6 +44,7 @@ #\u6587\u4EF6\u4E0A\u4F20[4500,5000) 4501=\u4E0A\u4F20\u6587\u4EF6\u4E3A\u7A7A 4502=\u4E0A\u4F20\u6587\u4EF6\u51FA\u9519 +4503=\u6587\u4ef6\u7c7b\u578b\u4e0d\u88ab\u5141\u8bb8 #\u540E\u52E4:\u63A5\u5F85\u7BA1\u7406 [5000,5100) 5000=\u5185\u5BB9\u4E0D\u80FD\u4E3A\u7A7A 5001=\u63A5\u5F85\u6E05\u5355\u4E0D\u80FD\u4E3A\u7A7A