12.29.6

13693261870

2022-12-29 71e6f7b5e94f0cfb322772201069053c762e3e41

12.29.6

 src/main/java/com/lf/server/controller/show/PipelineController.java

@@ -100,7 +100,7 @@
    })
    @ResponseBody
    @PostMapping(value = "/downloadReq")
    public ResponseMsg<Object> downloadReq(@RequestBody PipelineEntity pe) {
    public ResponseMsg<Object> downloadReq(@RequestBody PipelineEntity pe, HttpServletRequest req) {
        try {
            if (null == pe || StringHelper.isEmpty(pe.getPwd())) {
                return fail("密码不能为空");
@@ -118,17 +118,18 @@
                return fail("密码不符合要求");
            }

            Map<String, Object> map = new HashMap<>(4);
            Map<String, List<?>> map = new HashMap<>(4);
            for (String tab : pe.getTabs()) {
                List<PipelineEntity> rs = pipelineService.selectPipeAnalysis(tab, pe.getGid());
                List<?> rs = pipelineService.selectPipeAnalysis(tab, pe.getGid());
                if (null != rs && rs.size() > 0) {
                    map.put(tab, rs);
                }
            }

            //
            UserEntity ue = tokenService.getCurrentUser(req);
            String guid = pipelineService.createZipFile(ue, map, pe.getPwd());

            return success(map);
            return success(guid);
        } catch (Exception ex) {
            return fail(ex.getMessage(), null);
        }

 src/main/java/com/lf/server/helper/GdbHelper.java

 src/main/java/com/lf/server/service/show/PipelineService.java

@@ -1,11 +1,31 @@
package com.lf.server.service.show;

import com.lf.server.entity.data.DownloadEntity;
import com.lf.server.entity.data.MetaFileEntity;
import com.lf.server.entity.show.PipelineEntity;
import com.lf.server.entity.sys.UserEntity;
import com.lf.server.helper.*;
import com.lf.server.mapper.all.GeomBaseMapper;
import com.lf.server.mapper.data.DownloadMapper;
import com.lf.server.mapper.show.PipelineMapper;
import net.lingala.zip4j.ZipFile;
import net.lingala.zip4j.model.FileHeader;
import net.lingala.zip4j.model.ZipParameters;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.gdal.ogr.DataSource;
import org.gdal.ogr.Driver;
import org.gdal.ogr.Layer;
import org.gdal.ogr.ogr;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.io.File;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;

/**
 * 管道分析服务类
@@ -14,7 +34,15 @@
@Service
public class PipelineService implements PipelineMapper {
    @Autowired
    PathHelper pathHelper;

    @Autowired
    DownloadMapper downloadMapper;

    @Autowired
    PipelineMapper pipelineMapper;

    private final static Log log = LogFactory.getLog(PipelineService.class);

    @Override
    public List<PipelineEntity> selectSegNames() {
@@ -30,4 +58,135 @@
    public List<PipelineEntity> selectAnalysisResult(String tab, Integer gid) {
        return pipelineMapper.selectAnalysisResult(tab, gid);
    }

    /**
     * 创建Zip包
     *
     * @param ue  用户实体
     * @param map 管道分析数据集合
     * @param pwd 密码
     * @return 下载文件GUID
     */
    public String createZipFile(UserEntity ue, Map<String, List<?>> map, String pwd) throws Exception {
        if (map.size() == 0) {
            return null;
        }

        String tempName = StringHelper.YMDHMS2_FORMAT.format(new Date());
        String tempPath = pathHelper.getTempPath(tempName);
        String filePath = tempPath + File.separator + tempName + ".gdb";

        File file = new File(filePath);
        if (file.exists() && file.isDirectory()) {
            FileHelper.deleteDir(filePath);
        }
        //createGdb(filePath, map);

        String zipName = tempName + ".gdb.zip";
        String zipFile = pathHelper.getDownloadFullPath() + File.separator + zipName;

        ZipFile zip = Zip4jHelper.createZipFile(zipFile, pwd);
        ZipParameters params = Zip4jHelper.getZipParams();
        addZipFiles(zip, params, file.listFiles());

        String dbPwd = Md5Helper.reverse(Md5Helper.generate(pwd));
        DownloadEntity downloadEntity = getDownloadEntity(ue, zipFile, dbPwd);
        int rows = downloadMapper.insert(downloadEntity);

        return rows > 0 ? downloadEntity.getGuid() : null;
    }

    /**
     * 创建GDB
     */
    public static void createGdb(String filePath, Map<String, List<?>> map) throws Exception {
        Driver driver = null;
        DataSource dataSource = null;
        try {
            driver = ogr.GetDriverByName("FileGDB");
            if (null == driver) {
                log.error("GdbHelper.createGdb.driver(FileGDB) is null.");
                return;
            }
            dataSource = driver.CreateDataSource(filePath, null);
            if (null == dataSource) {
                log.error("GdbHelper.createGdb.dataSource is null. " + filePath);
                return;
            }

            for (String key : map.keySet()) {
                Layer layer = null;
                try {

//                    layer = createLayer(dataSource, baseMapper);
//                    if (null == layer) {
//                        continue;
//                    }
//
//                    String className = ClassHelper.getClassName(baseMapper);
//                    Class clazz = ClassHelper.getEntityClass(className);
//                    if (null == clazz) {
//                        continue;
//                    }
//
//                    List<Field> fields = new ArrayList<>();
//                    // fields.add(getGeomField(clazz))
//                    getFields(clazz, fields);
//                    addLayerField(layer, fields);
//
//                    setLayerData(layer, fields, map.get(key));
                } finally {
                    if (null != layer) {
                        layer.delete();
                    }
                }
            }

            dataSource.SyncToDisk();
            dataSource.FlushCache();
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            throw ex;
        } finally {
            GdbHelper.delete(dataSource, driver);
        }
    }

    /**
     * 添加Zip文件
     */
    private void addZipFiles(ZipFile zip, ZipParameters params, File[] files) {
        if (null == files || files.length == 0) {
            return;
        }

        for (File f : files) {
            try {
                zip.addFile(f, params);
            } catch (Exception ex) {
                log.error(ex.getMessage(), ex);
            }
        }
    }

    /**
     * 获取下载实体类
     */
    private DownloadEntity getDownloadEntity(UserEntity ue, String file, String pwd) throws Exception {
        DownloadEntity de = new DownloadEntity();
        de.setName(FileHelper.getFileName(file));
        // 1-Shp文件，2-专题图，3-元数据，4-业务数据，5-管道分析
        de.setType(5);
        de.setSizes(FileHelper.sizeToMb(new File(file).length()));
        de.setDepid(ue.getDepid());
        de.setDcount(0);
        de.setPwd(pwd);
        de.setUrl(FileHelper.getRelativePath(file));
        de.setDescr("管道分析文件");
        de.setGuid(FileHelper.getFileMd5(file));
        de.setCreateUser(ue.getId());
        // de.setGeom(null)

        return de;
    }
}