管道基础大数据平台系统开发-【后端】-Server
13693261870
2023-01-08 00d5cf2fc670965fc13fbb01ff1dfbd339130ced
src/main/java/com/lf/server/controller/data/upload/CheckController.java
@@ -5,6 +5,7 @@
import com.lf.server.entity.all.ResponseMsg;
import com.lf.server.entity.all.StaticData;
import com.lf.server.entity.ctrl.FmeReqEntity;
import com.lf.server.helper.PathHelper;
import com.lf.server.helper.StringHelper;
import com.lf.server.service.data.FmeService;
import io.swagger.annotations.ApiImplicitParam;
@@ -15,6 +16,7 @@
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.List;
@@ -24,6 +26,9 @@
 * @author WWW
 */
public class CheckController extends BaseController {
    @Autowired
    protected PathHelper pathHelper;
    @Autowired
    protected FmeService fmeService;
@@ -39,9 +44,7 @@
                return fail("id不能为空");
            }
            Object obj = fmeService.getTaskStatus(id, req);
            return obj;
            return fmeService.getTaskStatus(id, req);
        } catch (Exception ex) {
            return fail(ex.getMessage(), null);
        }
@@ -72,6 +75,20 @@
            if (StringHelper.isEmpty(entity.names)) {
                return fail("任务名称不能为空");
            }
            if (StringHelper.isEmpty(entity.zipPath)) {
                return fail("请选择待质检的zip文件");
            }
            entity.zipPath = getFullPath(entity.zipPath);
            if (!isZipFile(entity.zipPath)) {
                return fail("待质检的zip文件不存在");
            }
            if (entity.names.contains(StaticData.CHECK_MAIN)) {
                entity.wbsPath = getFullPath(entity.wbsPath);
                if (!isXlsFile(entity.wbsPath)) {
                    return fail("待质检的WBS文件不存在");
                }
            }
            List<String> list = new ArrayList<>();
            for (String name : entity.names.split(StaticData.COMMA)) {
@@ -85,6 +102,33 @@
        }
    }
    private String getFullPath(String filePath) {
        return null == filePath ? null : pathHelper.getConfig().getTempPath() + File.separator + filePath;
    }
    private boolean isZipFile(String filePath) {
        if (null == filePath || !filePath.toLowerCase().endsWith(StaticData.ZIP)) {
            return false;
        }
        File zipFile = new File(filePath);
        return zipFile.exists() && !zipFile.isDirectory();
    }
    private boolean isXlsFile(String filePath) {
        if (null == filePath) {
            return false;
        }
        if (!(filePath.toLowerCase().endsWith(StaticData.XLS) || filePath.toLowerCase().endsWith(StaticData.XLSX))) {
            return false;
        }
        File zipFile = new File(filePath);
        return zipFile.exists() && !zipFile.isDirectory();
    }
    /**
     * 方法调用
     */