1

src/main/java/com/fdkankan/fusion/service/impl/ModelServiceImpl.java

@@ -2,6 +2,7 @@ package com.fdkankan.fusion.service.impl;
 import java.io.File;
 import java.io.IOException;
 import java.math.BigDecimal;
+import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
 import java.util.stream.Collectors;
@@ -114,6 +115,16 @@ public class ModelServiceImpl extends ServiceImpl<IModelMapper, Model> implement
             if(!objPathFile.isDirectory()){
                 throw new BusinessException(ResultCode.UPLOAD_FILE_TYPE_ERROR);
             }
+            File[] files = objPathFile.listFiles();
+            List<String> canRunList = new ArrayList<>();
+            for (File file1 : files) {
+                if(file1.getName().contains("obj") || file1.getName().contains("las") || file1.getName().contains("ply")){
+                    canRunList.add(file1.getName());
+                }
+            }
+            if(canRunList.size() != 1){
+                throw new BusinessException(ResultCode.UPLOAD_FILE_TYPE_ERROR);
+            }
             File file1 = FileWriterUtil.getObjLasPlyFile(objPathFile);
             if(file1 == null){
                 throw new BusinessException(ResultCode.UPLOAD_FILE_TYPE_ERROR);