添加ewkt生成shp文件，修改下载文件功能添加zip下载

onemap-modules/onemap-analyse/src/main/java/com/onemap/analyse/controller/FzssController.java

@@ -21,8 +21,6 @@ import javax.servlet.http.HttpServletResponse;
 import java.io.File;
 import java.io.IOException;
 import java.net.URLEncoder;
-import java.text.SimpleDateFormat;
-import java.util.Date;
 import java.util.HashMap;
 import java.util.Map;
 
@@ -118,11 +116,17 @@ public class FzssController extends BaseController {
         File file = new File(realFileName);
         // 取得文件名。
         String downloadName = file.getName();
-        response.setCharacterEncoding("utf-8");
-        response.setContentType("multipart/form-data");
-        response.setContentType("application/vnd.openxmlformats-officedocument.wordprocessingml.document");
-//        response.setHeader("Content-Disposition", "attachment;fileName=\"" + downloadName + "\"");
-        response.setHeader("Content-Disposition", "attachment;fileName=" + URLEncoder.encode(downloadName, "UTF-8"));
-        FileUtils.writeBytes(realFileName, response.getOutputStream());
+
+        int lastIndex = downloadName.lastIndexOf(".");
+        String extension = downloadName.substring(lastIndex + 1);
+        if ("zip".equals(extension)) {
+            DownloadUtils.download(response, realFileName, null);
+        } else {
+            response.setCharacterEncoding("utf-8");
+            response.setContentType("multipart/form-data");
+            response.setContentType("application/vnd.openxmlformats-officedocument.wordprocessingml.document");
+            response.setHeader("Content-Disposition", "attachment;fileName=" + URLEncoder.encode(downloadName, "UTF-8"));
+            FileUtils.writeBytes(realFileName, response.getOutputStream());
+        }
     }
 }

onemap-modules/onemap-analyse/src/main/java/com/onemap/analyse/utils/DownloadUtils.java

@@ -7,8 +7,8 @@ import java.io.*;
 import java.nio.file.Files;
 
 public class DownloadUtils {
-    public static void download(HttpServletResponse response, String filepath, String name){
-        try{
+    public static void download(HttpServletResponse response, String filepath, String name) {
+        try {
             filepath = StringUtils.getFileAbsolutePath(filepath);
             File file = new File(filepath);
             // 取得文件名。
@@ -16,7 +16,7 @@ public class DownloadUtils {
             // 取得文件的后缀名。
             String ext = filename.substring(filename.lastIndexOf(".") + 1).toUpperCase();
             // 以流的形式下载文件。
-            InputStream fis = new BufferedInputStream(new FileInputStream(filepath));
+            InputStream fis = new BufferedInputStream(new FileInputStream(file));
             byte[] buffer = new byte[fis.available()];
             fis.read(buffer);
             fis.close();
@@ -30,7 +30,7 @@ public class DownloadUtils {
             toClient.write(buffer);
             toClient.flush();
             toClient.close();
-        }catch (Exception e){
+        } catch (Exception e) {
             e.printStackTrace();
         }
     }

onemap-modules/onemap-spatial/src/main/java/com/onemap/spatial/controller/ShpController.java

@@ -0,0 +1,39 @@
+package com.onemap.spatial.controller;
+
+import com.onemap.common.core.web.domain.RequestResult;
+import com.onemap.spatial.domain.ShpVo;
+import com.onemap.spatial.domain.WktsVo;
+import com.onemap.spatial.service.IImageService;
+import com.onemap.spatial.service.IWriteShpServer;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+import javax.annotation.Resource;
+import java.util.HashMap;
+import java.util.Map;
+
+@RestController
+@RequestMapping("/shp")
+public class ShpController {
+    @Resource
+    private IWriteShpServer writeShpServer;
+
+    @PostMapping("/write/ewkt")
+    public RequestResult writeShpByEWkt(@RequestBody ShpVo shpVo) throws Exception {
+        String path = writeShpServer.writeShpByEWkt(shpVo.getWkt(), shpVo.getFileName());
+        Map<String, String> map = new HashMap<>();
+        map.put("path", path);
+        return RequestResult.success(map);
+    }
+
+
+    @PostMapping("/write/wkt")
+    public RequestResult writeShpByWkt(@RequestBody ShpVo shpVo) throws Exception {
+        String path = writeShpServer.writeShpByWkt(shpVo.getWkt(), shpVo.getFileName());
+        Map<String, String> map = new HashMap<>();
+        map.put("path", path);
+        return RequestResult.success(map);
+    }
+}

onemap-modules/onemap-spatial/src/main/java/com/onemap/spatial/domain/ShpVo.java

@@ -0,0 +1,22 @@
+package com.onemap.spatial.domain;
+
+public class ShpVo {
+    private String wkt;
+    private String fileName;
+
+    public String getWkt() {
+        return wkt;
+    }
+
+    public void setWkt(String wkt) {
+        this.wkt = wkt;
+    }
+
+    public String getFileName() {
+        return fileName;
+    }
+
+    public void setFileName(String fileName) {
+        this.fileName = fileName;
+    }
+}

onemap-modules/onemap-spatial/src/main/java/com/onemap/spatial/domain/WriteShpVo.java

@@ -0,0 +1,13 @@
+package com.onemap.spatial.domain;
+
+public class WriteShpVo {
+    private String wkt;
+    
+    public String getWkt() {
+        return wkt;
+    }
+
+    public void setWkt(String wkt) {
+        this.wkt = wkt;
+    }
+}

onemap-modules/onemap-spatial/src/main/java/com/onemap/spatial/service/IWriteShpServer.java

@@ -0,0 +1,8 @@
+package com.onemap.spatial.service;
+
+public interface IWriteShpServer {
+
+    String writeShpByEWkt(String ewkt, String fileName);
+
+    String writeShpByWkt(String wkt, String fileName);
+}

onemap-modules/onemap-spatial/src/main/java/com/onemap/spatial/service/impl/WriteShpServerImpl.java

@@ -0,0 +1,68 @@
+package com.onemap.spatial.service.impl;
+
+import com.onemap.common.core.utils.StringUtils;
+import com.onemap.spatial.domain.WriteShpVo;
+import com.onemap.spatial.service.IWriteShpServer;
+import com.onemap.spatial.util.WriteShapeUtil;
+import lombok.extern.slf4j.Slf4j;
+import org.jetbrains.annotations.NotNull;
+import org.locationtech.jts.io.WKTReader;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Service;
+
+import java.io.File;
+import java.util.*;
+
+/**
+ * 生成SHP文件
+ */
+@Slf4j
+@Service
+public class WriteShpServerImpl implements IWriteShpServer {
+
+    @Value("${Path.rootPath}")
+    public String rootPath;
+
+    @Override
+    public String writeShpByEWkt(String ewkt, String fileName) {
+        if (StringUtils.isNull(ewkt)) {
+            return "";
+        }
+        String wkt = ewkt.replaceFirst("^SRID=\\d+;", "");
+        return writeShpByWkt(wkt, fileName);
+    }
+
+    @Override
+    public String writeShpByWkt(String wkt, String fileName) {
+        if (StringUtils.isNull(wkt)) {
+            return "";
+        }
+        // 图片保存路径
+        String uuid = StringUtils.getUUID();
+        String root = "/shp/" + uuid;
+        String fileDir = rootPath + root;
+        String shpName = System.currentTimeMillis() + ".shp";
+        String zipName = System.currentTimeMillis() + ".zip";
+        if (StringUtils.isNotEmpty(fileName)) {
+            int len = fileName.indexOf(".shp");
+            if (len > 0) {
+                shpName = fileName;
+            } else {
+                shpName = fileName + ".shp";
+            }
+        }
+        WriteShpVo wktVo = new WriteShpVo();
+        wktVo.setWkt(wkt);
+
+        List<WriteShpVo> voList = new ArrayList<>();
+        voList.add(wktVo);
+        try {
+            WriteShapeUtil.write2Shape(fileDir, shpName, "utf-8", voList, new WKTReader());
+            WriteShapeUtil.zipShapeFile(fileDir, zipName);
+        } catch (Exception e) {
+            log.error("shp生成异常：" + e.getMessage(), e);
+            return null;
+        }
+        return root + "/" + zipName;
+    }
+}

onemap-modules/onemap-spatial/src/main/java/com/onemap/spatial/util/WriteShapeUtil.java

@@ -0,0 +1,234 @@
+package com.onemap.spatial.util;
+
+import com.onemap.common.core.utils.StringUtils;
+import com.onemap.spatial.domain.WriteShpVo;
+import lombok.extern.slf4j.Slf4j;
+import org.geotools.data.FeatureWriter;
+import org.geotools.data.Transaction;
+import org.geotools.data.shapefile.ShapefileDataStore;
+import org.geotools.data.shapefile.ShapefileDataStoreFactory;
+import org.geotools.feature.simple.SimpleFeatureTypeBuilder;
+import org.geotools.referencing.crs.DefaultGeographicCRS;
+import org.locationtech.jts.geom.Geometry;
+import org.locationtech.jts.geom.MultiPolygon;
+import org.locationtech.jts.geom.Polygon;
+import org.locationtech.jts.io.ParseException;
+import org.locationtech.jts.io.WKTReader;
+import org.opengis.feature.simple.SimpleFeature;
+import org.opengis.feature.simple.SimpleFeatureType;
+import org.springframework.stereotype.Component;
+
+import javax.servlet.http.HttpServletResponse;
+import java.io.*;
+import java.nio.charset.Charset;
+import java.text.DateFormat;
+import java.text.SimpleDateFormat;
+import java.util.Date;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipOutputStream;
+
+/**
+ * shp导出工具类
+ */
+@Slf4j
+@Component
+public class WriteShapeUtil {
+
+    /**
+     * 生成shape文件 ，只支持4326坐标系的
+     *
+     * @param shpPath 生成shape文件路径（包含文件名称）
+     * @param encode  编码
+     *                //     * @param geoms   图幅集合
+     */
+    public static void write2Shape(String shpPath, String fileName, String encode, List<WriteShpVo> exp_shp, WKTReader reader) throws IOException, ParseException {
+        if (exp_shp == null || exp_shp.size() == 0) {
+            return;
+        }
+        if (StringUtils.isNull(shpPath)) {
+            return;
+        }
+        if (StringUtils.isNull(fileName)) {
+            return;
+        }
+        //创建shape文件对象
+        File file = new File(shpPath);
+        File fileNameUrl = new File(shpPath + File.separator + fileName);
+        if (!file.exists()) {// 如果目录不存在，创建目录
+            file.mkdirs();
+            fileNameUrl.createNewFile();
+        } else {
+            fileNameUrl.delete();
+        }
+
+        Map<String, Serializable> params = new HashMap<String, Serializable>();
+        params.put(ShapefileDataStoreFactory.URLP.key, fileNameUrl.toURI().toURL());
+        ShapefileDataStore ds = (ShapefileDataStore) new ShapefileDataStoreFactory().createNewDataStore(params);
+        //定义图形信息和属性信息
+        SimpleFeatureTypeBuilder tb = new SimpleFeatureTypeBuilder();
+        tb.setCRS(DefaultGeographicCRS.WGS84);
+        tb.setName("shapefile");
+        //判断空间类型，先只支持面，后期再加
+        Geometry geomTypeData = reader.read(exp_shp.get(0).getWkt());
+        if ("Polygon".equals(geomTypeData.getGeometryType())) {
+            tb.add("the_geom", Polygon.class);
+        } else if ("MultiPolygon".equals(geomTypeData.getGeometryType())) {
+            tb.add("the_geom", MultiPolygon.class);
+        }
+        //下面添加属性，暂时不做
+//        tb.add("pro_Name", String.class);
+//        tb.add("las_Name", String.class);
+        ds.createSchema(tb.buildFeatureType());
+        //设置编码
+        Charset charset = Charset.forName(encode);
+        ds.setCharset(charset);
+        //设置Writer
+        FeatureWriter<SimpleFeatureType, SimpleFeature> writer = ds.getFeatureWriter(ds.getTypeNames()[0], Transaction.AUTO_COMMIT);
+
+        for (WriteShpVo shp : exp_shp) {
+            SimpleFeature feature = writer.next();
+            Geometry geom = reader.read(shp.getWkt());
+            feature.setAttribute("the_geom", geom);
+//            feature.setAttribute("pro_Name", shp.getProjectName());
+//            feature.setAttribute("las_Name", shp.getLasname());
+        }
+        System.out.println("保存完成： " + fileNameUrl.getPath());
+        writer.write();
+        writer.close();
+        ds.dispose();
+    }
+
+    /**
+     * 压缩shape文件
+     *
+     * @param shpPath shape文件路径（包含shape文件名称）
+     */
+    public static void zipShapeFile(String shpPath, String zipName) {
+        try {
+            String zipPath = shpPath + File.separator + zipName;
+            File zipFile = new File(zipPath);
+            InputStream input = null;
+            ZipOutputStream zipOut = new ZipOutputStream(new FileOutputStream(zipFile));
+
+            File[] shpFiles = new File(shpPath).listFiles();
+
+            for (File file : shpFiles) {
+                input = new FileInputStream(file);
+                if (zipName.equals(file.getName())) {
+                    continue;
+                }
+                zipOut.putNextEntry(new ZipEntry(file.getName()));
+                int temp = 0;
+                while ((temp = input.read()) != -1) {
+                    zipOut.write(temp);
+                }
+                input.close();
+            }
+            zipOut.close();
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+    }
+
+    public static void zip(String sourceFileName, HttpServletResponse response) throws IOException {
+        ZipOutputStream out = null;
+        BufferedOutputStream bos = null;
+        DateFormat sf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
+        try {
+            //将zip以流的形式输出到前台
+            response.setHeader("content-type", "application/octet-stream");
+            response.setCharacterEncoding("utf-8");
+            // 设置浏览器响应头对应的Content-disposition
+            //参数中 testZip 为压缩包文件名，尾部的.zip 为文件后缀
+            response.setHeader("Content-disposition",
+                    "attachment;filename=" + java.net.URLEncoder.encode("expProject", "utf-8") + sf.format(new Date()) + ".zip");
+            //创建zip输出流
+            out = new ZipOutputStream(response.getOutputStream());
+            //创建缓冲输出流
+            bos = new BufferedOutputStream(out);
+            File sourceFile = new File(sourceFileName);
+            //调用压缩函数
+            compress(out, bos, sourceFile, sourceFile.getName());
+            out.flush();
+            log.info("压缩完成");
+        } catch (Exception e) {
+            log.error("ZIP压缩异常：" + e.getMessage(), e);
+        } finally {
+            //关闭流
+            if (null != bos) {
+                bos.close();
+            }
+            if (null != out) {
+                out.close();
+            }
+            deleteFile(new File(sourceFileName));
+            log.info("删除生成的文件夹：" + sourceFileName);
+        }
+    }
+
+    /**
+     * 将服务器上生成的目录及文件删除
+     *
+     * @param file
+     */
+    public static void deleteFile(File file) {
+        //判断文件不为null或文件目录存在
+        if (file == null || !file.exists()) {
+            return;
+        }
+        //取得这个目录下的所有子文件对象
+        File[] files = file.listFiles();
+        //遍历该目录下的文件对象
+        for (File f : files) {
+            //判断子目录是否存在子目录,如果是文件则删除
+            if (f.isDirectory()) {
+                deleteFile(f);
+            } else {
+                f.delete();
+            }
+        }
+        //删除空文件夹 for循环已经把上一层节点的目录清空。
+        file.delete();
+
+    }
+
+    public static void compress(ZipOutputStream out, BufferedOutputStream bos, File sourceFile, String base) {
+        FileInputStream fos = null;
+        BufferedInputStream bis = null;
+        try {
+            //如果路径为目录（文件夹）
+            if (sourceFile.isDirectory()) {
+                //取出文件夹中的文件（或子文件夹）
+                File[] flist = sourceFile.listFiles();
+                if (flist.length == 0) {//如果文件夹为空，则只需在目的地zip文件中写入一个目录进入点
+                    out.putNextEntry(new ZipEntry(base + "/"));
+                } else {//如果文件夹不为空，则递归调用compress，文件夹中的每一个文件（或文件夹）进行压缩
+                    for (int i = 0; i < flist.length; i++) {
+                        compress(out, bos, flist[i], base + "/" + flist[i].getName());
+                    }
+                }
+            } else {//如果不是目录（文件夹），即为文件，则先写入目录进入点，之后将文件写入zip文件中
+                out.putNextEntry(new ZipEntry(base));
+                fos = new FileInputStream(sourceFile);
+                bis = new BufferedInputStream(fos);
+
+                int tag;
+                //将源文件写入到zip文件中
+                while ((tag = bis.read()) != -1) {
+                    out.write(tag);
+                }
+
+                bis.close();
+                fos.close();
+            }
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+    }
+}
+
+
+