fix: 数据导入目录浏览/批量导入数量上限，防浏览器渲染与请求爆炸

xyc

4 天以前 72a8ea8b3df5f8ff2ccd49bd951ded5deb2e1593

 traffic-audit-server/src/main/java/com/trafficaudit/dataimport/controller/DataImportController.java

@@ -2,8 +2,12 @@

import com.trafficaudit.common.Result;
import com.trafficaudit.dataimport.dto.ImportResult;
import com.trafficaudit.dataimport.entity.ImportBatch;
import com.trafficaudit.dataimport.mapper.ImportBatchMapper;
import com.trafficaudit.dataimport.service.DataImportService;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
@@ -11,6 +15,9 @@
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

@Slf4j
@RestController
@@ -19,6 +26,8 @@

    @Resource
    private DataImportService importService;
    @Resource
    private ImportBatchMapper importBatchMapper;

    /** H203-2表 道路货物运输月度生产情况 */
    @PostMapping("/h2032")
@@ -50,6 +59,239 @@
    }



    /** H204 道路货运车辆能源消耗情况（季度能耗明细 + 运政车辆信息） */
    @PostMapping("/energyMonthly")
    public Result<String> importEnergyMonthly(@RequestParam("file") MultipartFile file,
                                              @RequestParam("period") String period) {
        try {
            ImportResult result = importService.importEnergyMonthly(file, period);
            String msg = "H204能耗季报导入完成：成功 " + result.getSuccess() + " 条，失败 " + result.getFail() + " 条（含同报表期运政车辆信息）";
            if (result.getFail() > 0) {
                java.util.List<String> details = result.getFailDetails();
                int shown = Math.min(details.size(), 100);
                for (int i = 0; i < shown; i++) {
                    msg += "\n" + details.get(i);
                }
                if (details.size() > shown) {
                    msg += "\n……共 " + details.size() + " 行失败";
                }
            }
            return Result.ok(msg);
        } catch (Exception e) {
            log.error("energyMonthly import error", e);
            return Result.error("导入失败: " + e.getMessage());
        }
    }

    /** 能耗车辆运政信息（独立模板，也可随 H204 季报第二个 sheet 一并导入） */
    @PostMapping("/energyAuth")
    public Result<String> importEnergyAuth(@RequestParam("file") MultipartFile file,
                                           @RequestParam("period") String period) {
        try {
            ImportResult result = importService.importEnergyAuth(file, period);
            String msg = "能耗车辆运政信息导入完成：成功 " + result.getSuccess() + " 条，失败 " + result.getFail() + " 条";
            if (result.getFail() > 0) {
                java.util.List<String> details = result.getFailDetails();
                int shown = Math.min(details.size(), 100);
                for (int i = 0; i < shown; i++) {
                    msg += "\n" + details.get(i);
                }
                if (details.size() > shown) {
                    msg += "\n……共 " + details.size() + " 行失败";
                }
            }
            return Result.ok(msg);
        } catch (Exception e) {
            log.error("energyAuth import error", e);
            return Result.error("导入失败: " + e.getMessage());
        }
    }

    /** H203-1表 公路旅客运输月度生产情况（含运政车辆数sheet） */
    @PostMapping("/passengerMonthly")
    public Result<String> importPassengerMonthly(@RequestParam("file") MultipartFile file,
                                                 @RequestParam("period") String period) {
        try {
            ImportResult result = importService.importPassengerMonthly(file, period);
            String msg = "H2031旅客月报导入完成：成功 " + result.getSuccess() + " 条，失败 " + result.getFail() + " 条（含同报表期运政车辆数据）";
            if (result.getFail() > 0) {
                java.util.List<String> details = result.getFailDetails();
                int shown = Math.min(details.size(), 100);
                for (int i = 0; i < shown; i++) {
                    msg += "\n" + details.get(i);
                }
                if (details.size() > shown) {
                    msg += "\n……共 " + details.size() + " 行失败";
                }
            }
            return Result.ok(msg);
        } catch (Exception e) {
            log.error("passengerMonthly import error", e);
            return Result.error("导入失败: " + e.getMessage());
        }
    }


    /** 公路旅客个体客运量/周转量（企业线下汇报，导入后供分市州表个体行填充） */
    @PostMapping("/passengerIndividual")
    public Result<String> importPassengerIndividual(@RequestParam("file") MultipartFile file,
                                                    @RequestParam("period") String period) {
        try {
            ImportResult result = importService.importPassengerIndividual(file, period);
            String msg = "个体客运量导入完成：成功 " + result.getSuccess() + " 条，失败 " + result.getFail() + " 条";
            if (result.getFail() > 0) {
                java.util.List<String> details = result.getFailDetails();
                int shown = Math.min(details.size(), 100);
                for (int i = 0; i < shown; i++) {
                    msg += "\n" + details.get(i);
                }
                if (details.size() > shown) {
                    msg += "\n……共 " + details.size() + " 行失败";
                }
            }
            return Result.ok(msg);
        } catch (Exception e) {
            log.error("passengerIndividual import error", e);
            return Result.error("导入失败: " + e.getMessage());
        }
    }

    /** 公路旅客运政车辆数（独立模板，也可随 H203-1 月报第二个 sheet 一并导入） */
    @PostMapping("/passengerAuth")
    public Result<String> importPassengerAuth(@RequestParam("file") MultipartFile file,
                                              @RequestParam("period") String period) {
        try {
            ImportResult result = importService.importPassengerAuth(file, period);
            String msg = "旅客运政车辆数导入完成：成功 " + result.getSuccess() + " 条，失败 " + result.getFail() + " 条";
            if (result.getFail() > 0) {
                java.util.List<String> details = result.getFailDetails();
                int shown = Math.min(details.size(), 100);
                for (int i = 0; i < shown; i++) {
                    msg += "\n" + details.get(i);
                }
                if (details.size() > shown) {
                    msg += "\n……共 " + details.size() + " 行失败";
                }
            }
            return Result.ok(msg);
        } catch (Exception e) {
            log.error("passengerAuth import error", e);
            return Result.error("导入失败: " + e.getMessage());
        }
    }

    /** 城市公交月报（城市公共交通月度运营情况，企业级明细） */
    @PostMapping("/cityBus")
    public Result<String> importCityBus(@RequestParam("file") MultipartFile file,
                                        @RequestParam("period") String period) {
        try {
            ImportResult result = importService.importCityBus(file, period);
            String msg = "城市公交月报导入完成：成功 " + result.getSuccess() + " 条，失败 " + result.getFail() + " 条";
            if (result.getFail() > 0) {
                java.util.List<String> details = result.getFailDetails();
                int shown = Math.min(details.size(), 100);
                for (int i = 0; i < shown; i++) {
                    msg += "\n" + details.get(i);
                }
                if (details.size() > shown) {
                    msg += "\n……共 " + details.size() + " 行失败";
                }
            }
            return Result.ok(msg);
        } catch (Exception e) {
            log.error("cityBus import error", e);
            return Result.error("导入失败: " + e.getMessage());
        }
    }

    /** 巡游出租汽车运营服务情况月报（市州级） */
    @PostMapping("/cityTaxi")
    public Result<String> importCityTaxi(@RequestParam("file") MultipartFile file,
                                         @RequestParam("period") String period) {
        try {
            ImportResult result = importService.importCityTaxi(file, period);
            String msg = "巡游出租月报导入完成：成功 " + result.getSuccess() + " 条，失败 " + result.getFail() + " 条";
            if (result.getFail() > 0) {
                java.util.List<String> details = result.getFailDetails();
                int shown = Math.min(details.size(), 100);
                for (int i = 0; i < shown; i++) {
                    msg += "\n" + details.get(i);
                }
                if (details.size() > shown) {
                    msg += "\n……共 " + details.size() + " 行失败";
                }
            }
            return Result.ok(msg);
        } catch (Exception e) {
            log.error("cityTaxi import error", e);
            return Result.error("导入失败: " + e.getMessage());
        }
    }

    /** 出租车运政车辆信息（车辆级明细，用于运营车数对比） */
    @PostMapping("/cityTaxiAuth")
    public Result<String> importCityTaxiAuth(@RequestParam("file") MultipartFile file,
                                             @RequestParam("period") String period) {
        try {
            int count = importService.importCityTaxiAuth(file, period);
            return Result.ok("出租车运政车辆导入成功, 共 " + count + " 条记录");
        } catch (Exception e) {
            log.error("cityTaxiAuth import error", e);
            return Result.error("导入失败: " + e.getMessage());
        }
    }

    /** 投资月报（全省汇总大表）：category=客运站场/物流园区 */
    @PostMapping("/investment")
    public Result<String> importInvestment(@RequestParam("file") MultipartFile file,
                                           @RequestParam("period") String period,
                                           @RequestParam(value = "category", defaultValue = "客运站场") String category) {
        try {
            ImportResult result = importService.importInvestment(file, period, category);
            String msg = "投资月报导入完成（" + category + "）：成功 " + result.getSuccess() + " 条，失败 " + result.getFail() + " 条";
            if (result.getFail() > 0) {
                java.util.List<String> details = result.getFailDetails();
                int shown = Math.min(details.size(), 100);
                for (int i = 0; i < shown; i++) {
                    msg += "\n" + details.get(i);
                }
                if (details.size() > shown) {
                    msg += "\n……共 " + details.size() + " 行失败";
                }
            }
            return Result.ok(msg);
        } catch (Exception e) {
            log.error("investment import error", e);
            return Result.error("导入失败: " + e.getMessage());
        }
    }

    /** 投资市州单表批量导入（M2）：扫描 docs/投资/输入/{类别}/{X月} 目录全部市州报表，category=客运站场/物流园区 */
    @PostMapping("/investCityBatch")
    public Result<Object> importInvestCityBatch(@RequestParam("period") String period,
                                                @RequestParam(value = "category", defaultValue = "客运站场") String category) {
        try {
            Map<String, Object> result = importService.importInvestCityDir(period, category);
            return Result.ok(result);
        } catch (Exception e) {
            log.error("investCityBatch import error", e);
            return Result.error("导入失败: " + e.getMessage());
        }
    }

    /** 投资系统导出（审核比对基准） */
    @PostMapping("/investmentSystem")
    public Result<String> importInvestmentSystem(@RequestParam("file") MultipartFile file,
                                                 @RequestParam("period") String period) {
        try {
            int count = importService.importInvestmentSystem(file, period);
            return Result.ok("投资系统导出导入成功, 共 " + count + " 条记录");
        } catch (Exception e) {
            log.error("investmentSystem import error", e);
            return Result.error("导入失败: " + e.getMessage());
        }
    }
    /** 运政车辆 */
    @PostMapping("/transportAuth")
    public Result<String> importTransportAuth(@RequestParam("file") MultipartFile file,
@@ -101,4 +343,63 @@
            return Result.error("导入失败: " + e.getMessage());
        }
    }
    /** 目录浏览器：path 为空返回项目 docs 数据目录，返回子目录与当前目录 Excel 文件 */
    @GetMapping("/dirs")
    public Result<Map<String, Object>> dirs(@RequestParam(value = "path", required = false) String path) {
        try {
            Map<String, Object> result = importService.listDirs(path);
            return Result.ok(result);
        } catch (Exception e) {
            log.error("list dirs error", e);
            return Result.error("目录浏览失败: " + e.getMessage());
        }
    }

    /** 按路径导入单个 Excel 文件（前端逐文件导入显示进度）；报表期优先从文件名识别 */
    @PostMapping("/fromDirFile")
    public Result<Map<String, Object>> importFromDirFile(@RequestParam("type") String type,
                                                         @RequestParam(value = "period", required = false) String period,
                                                         @RequestParam(value = "skipImported", defaultValue = "true") boolean skipImported,
                                                         @RequestParam("path") String path) {
        try {
            Map<String, Object> result = importService.importDirFileByPath(type, period, skipImported, path);
            return Result.ok(result);
        } catch (Exception e) {
            log.error("fromDirFile import error", e);
            return Result.error("导入失败: " + e.getMessage());
        }
    }

    /** 从服务端文件夹批量导入：扫描 dir 下全部 .xlsx/.xls，按 type 逐个导入；报表期优先从文件名识别 */
    @PostMapping("/fromDir")
    public Result<Object> importFromDir(@RequestParam("type") String type,
                                        @RequestParam(value = "period", required = false) String period,
                                        @RequestParam(value = "skipImported", defaultValue = "true") boolean skipImported,
                                        @RequestParam("dir") String dir) {
        try {
            Map<String, Object> result = importService.importFromDir(type, period, skipImported, dir);
            return Result.ok(result);
        } catch (Exception e) {
            log.error("fromDir import error", e);
            return Result.error("导入失败: " + e.getMessage());
        }
    }

    /** 导入批次历史（数据导入页展示最近导入结果） */
    @GetMapping("/batches")
    public Result<Map<String, Object>> batches(@RequestParam(value = "type", required = false) String type,
                                               @RequestParam(value = "period", required = false) String period,
                                               @RequestParam(value = "limit", defaultValue = "20") int limit) {
        int safeLimit = Math.min(Math.max(limit, 1), 200);
        LambdaQueryWrapper<ImportBatch> w = new LambdaQueryWrapper<ImportBatch>()
            .eq(type != null && !type.trim().isEmpty(), ImportBatch::getImportType, type)
            .eq(period != null && !period.trim().isEmpty(), ImportBatch::getReportPeriod, period)
            .orderByDesc(ImportBatch::getId)
            .last("limit " + safeLimit);
        List<ImportBatch> list = importBatchMapper.selectList(w);
        Map<String, Object> result = new LinkedHashMap<>();
        result.put("records", list);
        return Result.ok(result);
    }

}