소스 검색

添加 上传前文件名校验

wuweihao 5 년 전
부모
커밋
acb7fd5cde
2개의 변경된 파일10개의 추가작업 그리고 0개의 파일을 삭제
  1. 5 0
      src/main/java/com/fd/controller/FdModelController.java
  2. 5 0
      src/main/java/com/fd/controller/RasterController.java

+ 5 - 0
src/main/java/com/fd/controller/FdModelController.java

@@ -129,6 +129,11 @@ public class FdModelController {
             return new R(51005, MsgCode.E51005);
         }
 
+        String s = StringUtils.substringAfterLast(fileName, ".");
+        if (!"zip".equals(s)) {
+            return new R(50007, MsgCode.E50007);
+        }
+
         List<FileEntity> list = modelServer.findByFileName(fileName);
         if (list.size() > 0) {
             return new R(51006, MsgCode.E51006);

+ 5 - 0
src/main/java/com/fd/controller/RasterController.java

@@ -87,6 +87,11 @@ public class RasterController {
             return new R(51005, MsgCode.E51005);
         }
 
+        String s = StringUtils.substringAfterLast(fileName, ".");
+        if (!"tif".equals(s)) {
+            return new R(50008,MsgCode.E50008);
+        }
+
         List<FileEntity> list = rasterServer.findByFileName(fileName);
         if (list.size() > 0) {
             return new R(51006, MsgCode.E51006);