导入功能

YHYZT/server/src/main/java/com/zjugis/yzt/controller/StNzydkController.java

@@ -4,7 +4,6 @@ import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.zjugis.yzt.beans.entity.StNzydk;
 import com.zjugis.yzt.service.StNzydkService;
-import com.zjugis.yzt.service.StNzydkUploadService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;

YHYZT/server/src/main/java/com/zjugis/yzt/service/StNzydkUploadService.java

@@ -1,13 +0,0 @@
-package com.zjugis.yzt.service;
-
-import org.springframework.web.multipart.MultipartFile;
-
-/**
- * @program: yh_yzt
- * @description: 农转用数据上传Service
- * @author: chen
- * @create: 2024-10-11 14:45
- **/
-public interface StNzydkUploadService {
-    boolean uploadNzydkData(MultipartFile file);
-} 

YHYZT/server/src/main/java/com/zjugis/yzt/service/impl/StNzydkUploadServiceImpl.java

@@ -1,216 +0,0 @@
-package com.zjugis.yzt.service.impl;
-
-import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
-import com.zjugis.yzt.beans.entity.StNzydk;
-import com.zjugis.yzt.dao.StNzydkMapper;
-import com.zjugis.yzt.service.StNzydkService;
-import com.zjugis.yzt.service.StNzydkUploadService;
-import com.zjugis.yzt.utils.geocomm.TxtReader;
-import com.zjugis.yzt.utils.geocomm.ParseResult;
-import com.zjugis.yzt.utils.ExcelUtils;
-import lombok.extern.slf4j.Slf4j;
-import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.stereotype.Service;
-import org.springframework.web.multipart.MultipartFile;
-
-import java.io.*;
-import java.math.BigDecimal;
-import java.util.*;
-import java.util.zip.ZipEntry;
-import java.util.zip.ZipInputStream;
-
-/**
- * @program: yh_yzt
- * @description: 农转用数据上传Service实现类
- * @author: chen
- * @create: 2024-10-11 14:45
- **/
-@Service
-@Slf4j
-public class StNzydkUploadServiceImpl extends ServiceImpl<StNzydkMapper, StNzydk> implements StNzydkUploadService {
-
-    @Autowired
-    private StNzydkService stNzydkService;
-
-    @Autowired
-    private StNzydkMapper stNzydkMapper;
-
-    @Override
-    public boolean uploadNzydkData(MultipartFile zipFile) {
-        if (zipFile.isEmpty()) {
-            log.error("上传文件为空");
-            return false;
-        }
-
-        File tempDir = null;
-        try {
-            // 创建临时目录
-            tempDir = new File(System.getProperty("java.io.tmpdir"), UUID.randomUUID().toString());
-            System.out.println("tempDir:"+tempDir.getAbsolutePath());
-            if (!tempDir.exists()) {
-                tempDir.mkdirs();
-            }
-
-            // 解压ZIP文件
-            Map<String, File> txtFiles = new HashMap<>();
-            File xlsxFile = null;
-
-            try (ZipInputStream zis = new ZipInputStream(zipFile.getInputStream())) {
-                ZipEntry entry;
-                while ((entry = zis.getNextEntry()) != null) {
-                    File newFile = new File(tempDir, entry.getName());
-                    if (entry.isDirectory()) {
-                        newFile.mkdirs();
-                    } else {
-                        // 确保父目录存在
-                        new File(newFile.getParent()).mkdirs();
-                        try (FileOutputStream fos = new FileOutputStream(newFile)) {
-                            byte[] buffer = new byte[1024];
-                            int len;
-                            while ((len = zis.read(buffer)) > 0) {
-                                fos.write(buffer, 0, len);
-                            }
-                        }
-
-                        if (newFile.getName().toLowerCase().endsWith(".xlsx")) {
-                            xlsxFile = newFile;
-                        } else if (newFile.getName().toLowerCase().endsWith(".txt")) {
-                            String fileNameWithoutExt = newFile.getName().substring(0, newFile.getName().lastIndexOf('.'));
-                            txtFiles.put(fileNameWithoutExt, newFile);
-                        }
-                    }
-                }
-            }
-
-            if (xlsxFile == null) {
-                log.error("ZIP包中未找到XLSX文件");
-                return false;
-            }
-
-            // --- Start: 使用 ExcelUtils.importExcel 解析 XLSX 文件 ---
-            Map<String, Map<String, Object>> batchInfoMap = new HashMap<>();
-            List<Map<String, Object>> projectInfoList = new ArrayList<>();
-
-
-            int batchSheetIndex = 0;
-            int projectSheetIndex = 1;
-
-            List<Map<String, Object>> batchDataList = ExcelUtils.importExcel(
-                    new FileInputStream(xlsxFile), xlsxFile.getName(), 0, 1, batchSheetIndex
-            );
-            if (!batchDataList.isEmpty()) {
-                Map<String, Object> batchData = batchDataList.get(0);
-                String pcbh = batchData.containsKey("批次索引") ? String.valueOf(batchData.get("批次索引")) : null;
-                if (pcbh != null && !pcbh.isEmpty()) {
-                    batchInfoMap.put(pcbh, batchData);
-                }
-            } else {
-                log.warn("未能从 '农转用批次信息' sheet 读取到数据。");
-            }
-
-
-            projectInfoList = ExcelUtils.importExcel(
-                    new FileInputStream(xlsxFile), xlsxFile.getName(), 0, 1, projectSheetIndex
-            );
-            if (projectInfoList.isEmpty()) {
-                log.warn("未能从 '农转用项目信息' sheet 读取到数据。");
-            }
-
-            // --- End: 使用 ExcelUtils.importExcel 解析 XLSX 文件 ---
-
-            for (Map<String, Object> projectInfo : projectInfoList) {
-                StNzydk stNzydk = new StNzydk();
-
-                // 批次信息 (通过批次索引关联)
-                String pcbh = projectInfo.containsKey("批次索引") ? String.valueOf(projectInfo.get("批次索引")) : null;
-                Map<String, Object> batchInfo = batchInfoMap.get(pcbh);
-                if (batchInfo != null) {
-                    stNzydk.setPcbh(batchInfo.containsKey("批次索引") ? String.valueOf(batchInfo.get("批次索引")) : null);
-                    stNzydk.setPch(batchInfo.containsKey("批次名称") ? String.valueOf(batchInfo.get("批次名称")) : null);
-                    stNzydk.setLx(batchInfo.containsKey("类型") ? String.valueOf(batchInfo.get("类型")) : null);
-                    stNzydk.setPwh(batchInfo.containsKey("批准文号") ? String.valueOf(batchInfo.get("批准文号")) : null);
-                    stNzydk.setPzrq(batchInfo.containsKey("批准日期") ? String.valueOf(batchInfo.get("批准日期")) : null);
-                    stNzydk.setPzMj(parseBigDecimal(batchInfo.containsKey("批次总面积（公顷）") ? String.valueOf(batchInfo.get("批次总面积（公顷）")) : null));
-                    stNzydk.setNmjf(batchInfo.containsKey("是否农民建房") ? String.valueOf(batchInfo.get("是否农民建房")) : null);
-                } else {
-                    log.warn("未找到批次索引为 {} 的批次信息，项目 {} 将跳过批次信息填充。", pcbh, projectInfo.containsKey("项目编号") ? String.valueOf(projectInfo.get("项目编号")) : "未知项目");
-                }
-
-                // 项目信息
-                stNzydk.setDkBh(projectInfo.containsKey("项目编号") ? String.valueOf(projectInfo.get("项目编号")) : null);
-                stNzydk.setDkMc(projectInfo.containsKey("项目名称") ? String.valueOf(projectInfo.get("项目名称")) : null);
-                stNzydk.setXzz(projectInfo.containsKey("乡镇街道") ? String.valueOf(projectInfo.get("乡镇街道")) : null);
-                stNzydk.setDkYt(projectInfo.containsKey("土地用途") ? String.valueOf(projectInfo.get("土地用途")) : null);
-                stNzydk.setTdZl(projectInfo.containsKey("土地坐落") ? String.valueOf(projectInfo.get("土地坐落")) : null);
-                stNzydk.setDkMj(parseBigDecimal(projectInfo.containsKey("项目面积（公顷）") ? String.valueOf(projectInfo.get("项目面积（公顷）")) : null));
-                stNzydk.setDkQs(projectInfo.containsKey("用地主体") ? String.valueOf(projectInfo.get("用地主体")) : null);
-                stNzydk.setBz(projectInfo.containsKey("备注") ? String.valueOf(projectInfo.get("备注")) : null);
-                // 保存到数据库
-                if (!stNzydkService.save(stNzydk)) {
-                    log.error("保存农转用数据失败：{}", stNzydk.getDkBh());
-                    return false;
-                }
-                // 获取保存后的objectid
-                Integer objectId = stNzydk.getObjectid();
-                if (objectId == null) {
-                    log.error("获取保存后的objectid失败：{}", stNzydk.getDkBh());
-                    return false;
-                }
-                // 解析图形信息
-                String projectCode = stNzydk.getDkBh();
-                File txtFile = txtFiles.get(projectCode);
-                if (txtFile != null) {
-                    try (FileInputStream txtInputStream = new FileInputStream(txtFile)) {
-                        TxtReader txtReader = new TxtReader(txtInputStream);
-                        ParseResult parseResult = txtReader.read();
-                        if (parseResult != null && parseResult.getGeometry() != null) {
-                            String wkt = parseResult.getGeometry().toText();
-                            stNzydkMapper.updateShapeById(objectId, wkt, 4528);
-                        } else {
-                            log.warn("项目 {} 的TXT文件解析失败或未获取到Geometry数据。", projectCode);
-                        }
-                    } catch (Exception e) {
-                        log.error("解析项目 {} 的TXT文件失败: {}", projectCode, e.getMessage());
-                    }
-                } else {
-                    log.warn("未找到项目 {} 对应的TXT界址点文件。", projectCode);
-                }
-
-
-            }
-
-            return true;
-
-        } catch (IOException e) {
-            log.error("文件上传或处理失败", e);
-            return false;
-        } finally {
-            // 清理临时文件
-            if (tempDir != null && tempDir.exists()) {
-                deleteDirectory(tempDir);
-            }
-        }
-    }
-
-    private BigDecimal parseBigDecimal(String value) {
-        if (value == null || value.isEmpty()) {
-            return null;
-        }
-        try {
-            return new BigDecimal(value);
-        } catch (NumberFormatException e) {
-            log.warn("无法将字符串 '{}' 转换为 BigDecimal。", value);
-            return null;
-        }
-    }
-
-    private void deleteDirectory(File directory) {
-        File[] allContents = directory.listFiles();
-        if (allContents != null) {
-            for (File file : allContents) {
-                deleteDirectory(file);
-            }
-        }
-        directory.delete();
-    }
-}