1

src/main/java/com/fdkankan/manage_jp/common/ResultCode.java

@@ -53,6 +53,7 @@ public enum ResultCode  {
     SCENE_NOT_GPS(5031, "此场景无位置信息，无法在地图显示。"),
     PLATFORM_NOT_EXIT(5032, "平台不存在。"),
     UPLOAD_ERROR(5033, "上传失败。"),
+    UPLOAD_ERROR2(5034, "上传失败。文件格式不对"),
 
 
     ;

src/main/java/com/fdkankan/manage_jp/controller/UploadController.java

@@ -47,10 +47,12 @@ public class UploadController extends BaseController{
     FdKKClient fdKKClient;
     @PostMapping("/uploadE57")
     public Result uploadE57(@RequestParam("file") MultipartFile file,Integer isObj) {
+        String originalFilename = file.getOriginalFilename();
+        String suffix = originalFilename.substring(originalFilename.lastIndexOf("."));
+        if(!suffix.equals(".e57")){
+            throw new BusinessException(ResultCode.UPLOAD_ERROR2);
+        }
         try {
-            String originalFilename = file.getOriginalFilename();
-            String suffix = originalFilename.substring(originalFilename.lastIndexOf("."));
-
             String newFileName = UUID.randomUUID().toString().replace("-","");
             String filePath = Constant.MANAGE_PATH + "e57" + File.separator + newFileName +suffix;
             File targetFile = new File(filePath);