DataManager/src/main/java/com/fd/controller/VtsVectorController.java

package com.fd.controller;

import com.fd.Dto.PageDto;
import com.fd.Dto.VectorResourcesJsonDto;
import com.fd.constant.CommandMsg;
import com.fd.constant.MsgCode;
import com.fd.constant.TypeCode;
import com.fd.entity.FileEntity;
import com.fd.repository.FileRepository;
import com.fd.repository.GenerateFileRepository;
import com.fd.repository.ScheduleRepository;
import com.fd.util.ResponseResult;
import com.fd.server.CmdServer;
import com.fd.server.FileServer;
import com.fd.server.JsonServer;
import io.swagger.annotations.ApiOperation;
import lombok.extern.log4j.Log4j2;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Sort;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.util.Date;
import java.util.Optional;

/**
 * Created by Owen on 2019/11/7 0007 10:19
 */
@Log4j2
@RestController
@RequestMapping("api/vts/vector")
public class VtsVectorController {

    @Value("${uploadFile.fdmodelPath}")
    private String fdmodelPath;

    @Value("${uploadFile.vectorPath}")
    private String vectorPath;

    @Autowired
    private FileServer fileServer;

    @Autowired
    private FileRepository fileRepository;

    @Autowired
    private JsonServer jsonServer;

    @Autowired
    private CmdServer cmdServer;

    @Autowired
    private ScheduleRepository scheduleRepository;

    @Autowired
    private GenerateFileRepository generateFileRepository;


    @ApiOperation("upload Vector file list")
    @PostMapping(value = "list")
    private ResponseResult list(@RequestBody PageDto param){
        log.info("run list");
        Page<FileEntity> page = fileRepository.findByType(
                TypeCode.FILE_TYPE_VECTOR, TypeCode.FILE_TYPE_VECTOR_GEOJSON,
                PageRequest.of(param.getPageNum(), param.getPageSize(), Sort.by("createTime").descending()));
        return new ResponseResult(200,page);
    }

    /**
     * 文件上传
     * 需要创建目录
     * 不支持大文件上传
     *
     * @return
     */
    @ApiOperation("upload Vector file")
    @PostMapping(value = "upload/{directoryName}/", consumes = { "multipart/form-data" })
    private ResponseResult uploadVector(@RequestParam("file") MultipartFile file, @PathVariable("directoryName") String directoryName){
        log.info("run uploadVector");
        return fileServer.upload2(file, vectorPath+directoryName+"/");
    }


    @ApiOperation("commandOgr2ogr")
    @GetMapping("/command/ogr2ogr/{fileId}/")
    private ResponseResult commandOgr2ogr(@PathVariable("fileId") Long fileId) {
        log.info("run commandOgr2ogr");

        Optional<FileEntity> entity = fileRepository.findById(fileId);
        if (!entity.isPresent()) {
            return new ResponseResult(50002, MsgCode.E50002);
        }

        FileEntity fileEntity = entity.get();
        String fileUrl = fileEntity.getFileUrl();
        fileUrl = fileUrl.substring(0, fileUrl.lastIndexOf("."));

        String cmd = CommandMsg.VECTOR_OGR2OGR;
        cmd = cmd.replaceAll("@fileName", fileUrl);
        log.info("cmd: {}", cmd);

        Integer s = cmdServer.exeCmd(cmd, fileId);
        if (s == 1) {
            log.info("cmd  run: {}", MsgCode.E50005);
            return new ResponseResult(50005, MsgCode.E50005);
        }
        log.info("cmd  run : success");

        // 文件名
        String fileName = fileEntity.getFileName();

        // 把生成的文件信息写入数据库
        FileEntity geojson = new FileEntity();
        geojson.setFileName(fileName.replace(".shp",".geojson"));
        geojson.setFileUrl(fileUrl+".geojson");
        geojson.setCreateTime(new Date());
        geojson.setUpdateTime(new Date());
        geojson.setType(TypeCode.FILE_TYPE_VECTOR_GEOJSON);
        fileRepository.save(geojson);

        return new ResponseResult(200, geojson);
    }

    @ApiOperation("commandTippecanoe ")
    @GetMapping("/command/tippecanoe/{fileId}/")
    private ResponseResult commandTippecanoe(@PathVariable("fileId") Long fileId) {
        log.info("run commandTippecanoe");

        Optional<FileEntity> entity = fileRepository.findById(fileId);
        if (!entity.isPresent()) {
            return new ResponseResult(50002, MsgCode.E50002);
        }

        FileEntity fileEntity = entity.get();
        String fileUrl = fileEntity.getFileUrl();
        fileUrl = fileUrl.substring(0, fileUrl.lastIndexOf("."));


        String cmd = CommandMsg.VECTOR_TIPPECANOE;
        cmd = cmd.replaceAll("@fileName", fileUrl);
        log.info("cmd: {}", cmd);
        Integer s = cmdServer.exeCmd(cmd, fileId);
        if (s ==1) {
            log.info("cmd  run: {}", MsgCode.E50005);
            return new ResponseResult(50005, MsgCode.E50005);
        }
        log.info("cmd  run : success");

        return new ResponseResult(200, "success");
    }

    /**
     * 删除文件
     */
    @GetMapping("delete/{fileId}/")
    private ResponseResult deleteFile(@PathVariable("fileId") Long fileId) {
        log.info("run deleteFile: {}", fileId);
        return fileServer.deleteFileById(fileId);
    }


//    @ApiOperation("update 4dage")
//    @GetMapping("update/fdage/{param}/")
//    private ResponseResult updateFdage(@PathVariable("param") String param){
//        log.info("run updateFdage");
//        jsonServer.updateVectroFdage(param);
//        return new ResponseResult(200,"success");
//    }


    /**
     * restful 风格会，参数多，会传不进来
     */
    @ApiOperation("update 4dage")
    @GetMapping("update/fdage")
    private ResponseResult updateFdage(String param){
        log.info("run updateFdage");
        jsonServer.updateVectroFdage(param);
        return new ResponseResult(200,"success");
    }


    /**
     * 会增加一块数据
     * @param param
     * @return
     */
    @ApiOperation("update vector resources.json")
    @PostMapping("update/resources")
    private ResponseResult updateVectorResources(@RequestBody VectorResourcesJsonDto param){
        log.info("run updateVectorResources");
        System.out.println(param.toString());
        jsonServer.updateVectorResourcesJson(param);
        return new ResponseResult(200,"success");
    }

}