1

13693261870

2023-09-19 03d4fc607b5df795a4319120036ef138cde70896

 src/main/java/com/lf/server/controller/data/PublishController.java

@@ -3,20 +3,31 @@
import com.lf.server.annotation.SysLog;
import com.lf.server.controller.all.BaseController;
import com.lf.server.entity.all.ResponseMsg;
import com.lf.server.entity.all.StaticData;
import com.lf.server.entity.ctrl.PubEntity;
import com.lf.server.entity.data.MetaEntity;
import com.lf.server.entity.data.MetaFileEntity;
import com.lf.server.entity.data.PublishEntity;
import com.lf.server.entity.sys.UserEntity;
import com.lf.server.helper.PathHelper;
import com.lf.server.helper.StringHelper;
import com.lf.server.helper.WebHelper;
import com.lf.server.service.data.LayerService;
import com.lf.server.service.data.MetaService;
import com.lf.server.service.data.PublishService;
import com.lf.server.service.data.RasterService;
import com.lf.server.service.sys.TokenService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import java.io.File;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
@@ -27,11 +38,23 @@
@RestController
@RequestMapping("/publish")
public class PublishController extends BaseController {
    @Autowired
    @Resource
    PublishService publishService;

    @Autowired
    @Resource
    TokenService tokenService;

    @Resource
    LayerService layerService;

    @Resource
    MetaService metaService;

    @Resource
    RasterService rasterService;

    @Resource
    PathHelper pathHelper;

    @SysLog()
    @ApiOperation(value = "分页查询元数据")
@@ -51,10 +74,10 @@
                return fail("每页页数或分页数小于1", null);
            }
            if (StringHelper.isEmpty(type)) {
                return fail("请输入数据类别");
                return fail("数据类别为空", null);
            }

            String types = getType(type);
            String types = getType(dircode, type);
            int count = publishService.selectMetasByCount(depcode, dircode, verid, types, name);
            if (count == 0) {
                return success(0, null);
@@ -71,45 +94,114 @@
    /**
     * 获取类型
     */
    private String getType(String type) throws Exception {
    private String getType(String dircode, String type) throws Exception {
        switch (type) {
            case "DOM":
                return "type in ('tif', 'tiff', 'img')";
                return "type in ('tif', 'tiff', 'img')" + getFilter(dircode, type);
            case "DEM":
                return "type in ('tif', 'tiff')" + getFilter(dircode, type);
            case "MPT":
                return "type = 'mpt'";
            case "3DML":
                return "type = '3dml'";
            case "CPT":
                return "type = 'cpt'";
            case "BIM":
                return "type in ('ifc', 'fbx', 'rvt')";
            case "LAS":
                return "type in ('las', 'laz')";
            case "OSGB":
                return "type = 'osgb'";
            default:
                throw new Exception("数据类型不匹配");
        }
    }

    /**
     * 获取过滤条件
     */
    private String getFilter(String dircode, String type) {
        dircode = StringHelper.isEmpty(dircode) ? "" : StringHelper.getRightLike(dircode);

        List<String> list = null;
        switch (type) {
            case "DOM":
                list = publishService.selectCodesForDir(dircode, 0);
                break;
            case "DEM":
                list = publishService.selectCodesForDir(dircode, 1);
                break;
            default:
                break;
        }
        if (null == list || list.isEmpty()) {
            return "";
        }

        for (int i = 0, c = list.size(); i < c; i++) {
            list.set(i, "'" + list.get(i) + "'");
        }

        return " and dircode not in (" + StringHelper.join(list, ",") + ")";
    }

    @SysLog()
    @ApiOperation(value = "分页查询并返回记录数")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "name", value = "名称", dataType = "String", paramType = "query", example = ""),
            @ApiImplicitParam(name = "dircode", value = "目录", dataType = "String", paramType = "query", example = ""),
            @ApiImplicitParam(name = "type", value = "类别", dataType = "String", paramType = "query", example = ""),
            @ApiImplicitParam(name = "pageSize", value = "每页条数", dataType = "Integer", paramType = "query", example = "10"),
            @ApiImplicitParam(name = "pageIndex", value = "分页数（从1开始）", dataType = "Integer", paramType = "query", example = "1")
    })
    @GetMapping(value = "/selectByPageAndCount")
    public ResponseMsg<List<PublishEntity>> selectByPageAndCount(String name, Integer pageSize, Integer pageIndex) {
    public ResponseMsg<List<PublishEntity>> selectByPageAndCount(String name, String dircode, String type, Integer pageSize, Integer pageIndex) {
        try {
            if (pageSize < 1 || pageIndex < 1) {
                return fail("每页页数或分页数小于1", null);
            }
            type = getPubType(type);

            int count = publishService.selectCount(name);
            int count = publishService.selectCount(name, dircode, type);
            if (count == 0) {
                return success(0, null);
            }

            List<PublishEntity> rs = publishService.selectByPage(name, pageSize, pageSize * (pageIndex - 1));
            List<PublishEntity> rs = publishService.selectByPage(name, dircode, type, pageSize, pageSize * (pageIndex - 1));

            return success(count, rs);
        } catch (Exception ex) {
            return fail(ex.getMessage(), null);
        }
    }

    /**
     * 获取发布类型
     */
    private String getPubType(String type) {
        if (StringHelper.isEmpty(type)) {
            return null;
        }

        switch (type) {
            case "DOM":
                return "type = 'DOM'";
            case "DEM":
                return "type = 'DEM'";
            case "MPT":
                return "type = 'mpt'";
            case "3DML":
                return "type = '3dml'";
            case "CPT":
                return "type = 'cpt'";
            case "BIM":
                return "type in ('ifc', 'fbx', 'rvt')";
            case "LAS":
                return "type in ('las', 'laz')";
            case "OSGB":
                return "type = 'osgb'";
            default:
                return null;
        }
    }

@@ -130,23 +222,145 @@
    }

    @SysLog()
    @ApiOperation(value = "插入一条")
    @ApiOperation(value = "查询Las文件坐标系ID")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "entity", value = "实体类", dataType = "PublishEntity", paramType = "body")
            @ApiImplicitParam(name = "ids", value = "元数据ID集合", dataType = "Integer[]", paramType = "query", example = "10483,10481,10456,10285")
    })
    @PostMapping(value = "/insert", produces = "application/json; charset=UTF-8")
    public ResponseMsg<Integer> insert(@RequestBody PublishEntity entity, HttpServletRequest req) {
    @GetMapping(value = "/selectLasCs")
    public ResponseMsg<Object> selectLasCs(Integer[] ids, HttpServletRequest req) {
        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue != null) {
                entity.setCreateUser(ue.getId());
            if (null == ids || ids.length == 0) {
                return fail("元数据ID集合不能为空");
            }

            int count = publishService.insert(entity);
            PubEntity entity = new PubEntity();
            entity.setIds(Arrays.asList(ids));
            entity.setDircode("00");
            entity.setToken(WebHelper.getToken(req));

            List<Integer> list = publishService.selectLasCs(entity, "/Convert/ReadLasCs", req);

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

    @SysLog()
    @ApiOperation(value = "查询栅格数据的坐标系ID")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "ids", value = "元数据ID集合", dataType = "Integer[]", paramType = "query", example = "10531,10527,10526")
    })
    @GetMapping(value = "/selectRasterCs")
    public ResponseMsg<Object> selectRasterCs(Integer[] ids, HttpServletRequest req) {
        try {
            if (null == ids || ids.length == 0) {
                return fail("元数据ID集合不能为空");
            }

            List<MetaEntity> list = metaService.selectByIds(StringHelper.join(Arrays.asList(ids), ","));
            if (null == list || list.isEmpty()) {
                return fail("查无数据");
            }

            String basePath = pathHelper.getConfig().getUploadPath() + File.separator;
            List<MetaEntity> errList = new ArrayList<>();
            for (MetaEntity me : list) {
                if (!StaticData.RASTER_EXT.contains("." + me.getType())) {
                    me.setBak("不是栅格数据");
                    errList.add(me);
                    continue;
                }

                String filePath = basePath + me.getPath();
                File f = new File(filePath);
                if (!f.exists() || f.isDirectory()) {
                    me.setBak("文件不存在");
                    errList.add(me);
                    continue;
                }

                Integer epsg = rasterService.getRaterEpsg(filePath);
                if (null == epsg || epsg < 1) {
                    me.setBak("坐标系ID无效");
                    errList.add(me);
                }
            }

            return success(errList.size(), errList);
        } catch (Exception ex) {
            return fail(ex.getMessage(), null);
        }
    }

    @SysLog()
    @ApiOperation(value = "查询坐标转换")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "x", value = "X坐标", dataType = "double", paramType = "query", example = "40476573.02"),
            @ApiImplicitParam(name = "y", value = "Y坐标", dataType = "double", paramType = "query", example = "3391108.90"),
            @ApiImplicitParam(name = "epsg", value = "EPSG编码", dataType = "int", paramType = "query", example = "4528")
    })
    @GetMapping(value = "/selectCsTransform")
    public Object selectCsTransform(double x, double y, int epsg) {
        try {
            return rasterService.csTransform(x, y, epsg);
        } catch (Exception ex) {
            return null;
        }
    }

    @SysLog()
    @ApiOperation(value = "插入发布数据")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "entity", value = "实体类", dataType = "PubEntity", paramType = "body")
    })
    @PostMapping(value = "/insertForPub", produces = "application/json; charset=UTF-8")
    public ResponseMsg<Object> insertForPub(@RequestBody PubEntity entity, HttpServletRequest req) {
        try {
            if (null == entity || null == entity.getIds() || entity.getIds().isEmpty()) {
                return fail("实体类为空或找不到元数据ID", 0);
            }
            if (StringHelper.isEmpty(entity.getType())) {
                return fail("数据类别为空", null);
            }

            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue != null) {
                entity.setUserId(ue.getId());
                entity.setToken(WebHelper.getToken(req));
            }

            layerService.clearCache();
            String method = getConvertMethod(entity.getType());
            long count = publishService.postForPub(entity, method, req);

            return success(count);
        } catch (Exception ex) {
            return fail(ex.getMessage(), -1);
        }
    }

    /**
     * 获取转换方法
     */
    private String getConvertMethod(String type) throws Exception {
        switch (type) {
            case "DOM":
                return "/Convert/ToTiles";
            case "DEM":
                return "/Convert/ToTerra";
            case "MPT":
            case "3DML":
            case "CPT":
                return "/Convert/ToSG";
            case "BIM":
                return "/Convert/ToTileset";
            case "LAS":
                return "/Convert/ToLasByPy";
            case "OSGB":
                return "/Convert/ToOsgb";
            default:
                throw new Exception("数据类型不匹配");
        }
    }

@@ -156,13 +370,23 @@
            @ApiImplicitParam(name = "ids", value = "ID数组", dataType = "Integer", paramType = "query", allowMultiple = true, example = "1")
    })
    @GetMapping(value = "/deletes")
    public ResponseMsg<Integer> deletes(@RequestParam List<Integer> ids) {
    public ResponseMsg<Integer> deletes(@RequestParam List<Integer> ids, HttpServletRequest req) {
        try {
            if (ids == null || ids.isEmpty()) {
                return fail("id数组不能为空", -1);
            }

            int count = publishService.deletes(StringHelper.join(ids, ","));
            String strs = StringHelper.join(ids, ",");
            List<PublishEntity> list = publishService.selectByIds(strs);
            if (null == list || list.isEmpty()) {
                return fail("没有找到要删除的数据", -1);
            }

            // publishService.deleteFiles(list)
            layerService.clearCache();
            publishService.deleteFiles(ids, req);

            int count = publishService.deletes(ids, req);

            return success(count);
        } catch (Exception ex) {
@@ -184,6 +408,7 @@
                entity.setUpdateUser(ue.getId());
            }

            layerService.clearCache();
            int count = publishService.update(entity);

            return success(count);