package com.se.simu.controller;
|
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.se.simu.config.PropertiesConfig;
|
import com.se.simu.domain.dto.GeDb;
|
import com.se.simu.domain.dto.GeLayer;
|
import com.se.simu.domain.po.DataPo;
|
import com.se.simu.domain.po.SimuPo;
|
import com.se.simu.domain.vo.CreateSimuVo;
|
import com.se.simu.domain.vo.R;
|
import com.se.simu.domain.vo.SimuVo;
|
import com.se.simu.service.*;
|
import io.swagger.annotations.*;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.http.HttpStatus;
|
import org.springframework.http.ResponseEntity;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.annotation.Resource;
|
import java.io.IOException;
|
import java.nio.file.Files;
|
import java.nio.file.Path;
|
import java.nio.file.Paths;
|
import java.util.ArrayList;
|
import java.util.Arrays;
|
import java.util.Date;
|
import java.util.List;
|
import java.util.concurrent.CompletableFuture;
|
import java.util.concurrent.ExecutionException;
|
import java.util.stream.Collectors;
|
|
@Api(tags = "仿真管理")
|
@Slf4j
|
@RestController
|
@RequestMapping("/simu")
|
@SuppressWarnings("ALL")
|
public class SimuController extends BaseController {
|
@Resource
|
UwService uwService;
|
|
@Resource
|
GedbService gedbService;
|
|
@Resource
|
SimuService simuService;
|
|
@Resource
|
SimuFilesService simuFilesService;
|
|
@Resource
|
PropertiesConfig config;
|
|
@Resource
|
ResultService resultService;
|
|
private final List<String> FLOOD_TYPE = new ArrayList<>(Arrays.asList("沙袋", "防水板"));
|
|
@ApiOperation(value = "创建")
|
@PostMapping(value = "/create", produces = "application/json; charset=UTF-8")
|
public R<Object> create(@RequestBody @ApiParam("创建仿真视图类") CreateSimuVo vo) {
|
try {
|
if (null == vo.getTotal() || vo.getTotal() < 1 || vo.getTotal() > 1000) {
|
return fail("降雨量不能为空，且取值在1~1000之间");
|
}
|
if (null == vo.getDuration() || vo.getDuration() < 1 || vo.getDuration() > 10080) {
|
return fail("仿真时长不能为空，且取值在1~10080之间");
|
}
|
if (null == vo.getMinx() || null == vo.getMiny() || null == vo.getMaxx() || null == vo.getMaxy()) {
|
return fail("选择范围不能为空", false);
|
}
|
if (null == vo.getPid() || vo.getPid() < 0) {
|
vo.setPid(0);
|
}
|
if (null == vo.getNum() || vo.getNum() < 1) {
|
vo.setNum(simuService.getMaxId() + 1);
|
}
|
if (null == vo.getPid() || vo.getPid() < 0) {
|
vo.setPid(0);
|
}
|
if (null == vo.getStartTime()) {
|
vo.setStartTime(new Date());
|
}
|
if (vo.getPid() > 0) {
|
SimuPo pp = simuService.getSimuByPid(vo.getPid());
|
if (null == pp) {
|
return fail("pid不存在");
|
}
|
if (null == vo.getFloodStart() || vo.getFloodStart() < 1 || vo.getFloodStart() > vo.getDuration() * 60) {
|
return fail("防汛开始时间不能为空，且取值在1~" + (vo.getDuration() * 60) + "之间");
|
}
|
if (null == vo.getFloodEnd() || vo.getFloodEnd() < vo.getFloodStart() || vo.getFloodEnd() > vo.getDuration() * 60) {
|
return fail("防汛结束时间不能为空，且取值在" + vo.getFloodStart() + "~" + (vo.getDuration() * 60) + "之间");
|
}
|
if (null == vo.getFloodHeight() || vo.getFloodHeight() < 1 || vo.getFloodHeight() > 2000) {
|
return fail("防汛高度不能为空，且取值在1~2000之间");
|
}
|
if (!FLOOD_TYPE.contains(vo.getFloodType())) {
|
return fail("防汛类型不能为空，且只能是沙袋和防水板");
|
}
|
if (null == vo.getFloodMinx() || null == vo.getFloodMiny() || null == vo.getFloodMaxx() || null == vo.getFloodMaxy()) {
|
return fail("防汛范围不能为空", false);
|
}
|
}
|
|
boolean flag = simuService.create(vo);
|
|
return success(flag, flag ? "成功" : "失败");
|
} catch (Exception ex) {
|
return fail(ex, null);
|
}
|
}
|
|
@ApiOperation(value = "删除")
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "ids", value = "ID", dataType = "Integer", paramType = "query", allowMultiple = true, example = "0")
|
})
|
@GetMapping(value = "/del")
|
public R<Object> del(@RequestParam List<Integer> ids) {
|
try {
|
if (ids == null || ids.isEmpty()) {
|
return fail("日志ID不能为空", null);
|
}
|
|
int count = simuService.del(ids);
|
|
return success(count);
|
} catch (Exception ex) {
|
return fail(ex, null);
|
}
|
}
|
|
@ApiOperation(value = "获取")
|
@GetMapping("/get")
|
public R<Object> get(@ApiParam("仿真视图类") SimuVo vo) {
|
try {
|
if (null == vo) {
|
vo = new SimuVo();
|
}
|
if (null == vo.getPageSize() || vo.getPageSize() < 1) {
|
vo.setPageSize(10);
|
}
|
if (null == vo.getPageIndex() || vo.getPageIndex() < 1) {
|
vo.setPageIndex(1);
|
}
|
|
IPage<SimuPo> paged = simuService.get(vo);
|
if (null == paged) {
|
return success(null, 0);
|
}
|
|
return success(paged.getRecords(), paged.getTotal());
|
} catch (Exception ex) {
|
return fail(ex, null);
|
}
|
}
|
|
@ApiOperation(value = "降水范围校验")
|
@GetMapping("/rangeVerify")
|
public R<Object> rangeVerify(@RequestParam @ApiParam("最小X") double minx,
|
@RequestParam @ApiParam("最小y") double miny,
|
@RequestParam @ApiParam("最大X") double maxx,
|
@RequestParam @ApiParam("最大y") double maxy) {
|
try {
|
DataPo data = new DataPo();
|
data.setMinx(minx);
|
data.setMiny(miny);
|
data.setMaxx(maxx);
|
data.setMaxy(maxy);
|
data.setEpsg(4326);
|
|
String token = gedbService.getToken();
|
GeDb db = gedbService.connectGedb(token, data);
|
List<GeLayer> layers = gedbService.getLayers(token, db);
|
|
return success(gedbService.queryBboxCount(token, db, layers));
|
} catch (Exception ex) {
|
return fail(ex, false);
|
}
|
}
|
|
@ApiOperation("test-上传多个文件")
|
@ApiImplicitParam(name = "schemeName", value = "方案名称", required = true, dataType = "String", paramType = "query", example = "edf243d3-3bec-4c51-9930-1ba862d9b5a3", dataTypeClass = String.class)
|
@PostMapping("/uploadMultipleFiles")
|
public ResponseEntity<String> uploadMultipleFiles(List<MultipartFile> files, @RequestParam("schemeName") String schemeName) throws IOException {
|
// 上传文件路径
|
String targetDir = Paths.get(config.getOutPath(), schemeName).toString();
|
log.info("目标目录: {}", targetDir);
|
createDirectoriesIfNotExists(targetDir);
|
List<CompletableFuture<Void>> uploadTasks = files.stream()
|
.map(file -> CompletableFuture.runAsync(() -> {
|
try {
|
file.transferTo(Paths.get(targetDir, file.getOriginalFilename()));
|
} catch (IOException e) {
|
log.error("文件上传失败", e);
|
throw new RuntimeException("文件上传失败");
|
}
|
}))
|
.collect(Collectors.toList());
|
|
CompletableFuture<Void> allUploadTasks = CompletableFuture.allOf(uploadTasks.toArray(new CompletableFuture[0]));
|
|
try {
|
allUploadTasks.get();
|
return ResponseEntity.ok("所有文件上传成功");
|
} catch (InterruptedException | ExecutionException e) {
|
log.error("文件上传失败", e);
|
return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("文件上传失败");
|
}
|
}
|
|
|
/**
|
* @param targetDir 目标目录
|
* @throws IOException
|
*/
|
private void createDirectoriesIfNotExists(String targetDir) throws IOException {
|
Path path = Paths.get(targetDir);
|
if (!Files.exists(path)) {
|
Files.createDirectories(path);
|
}
|
}
|
|
/**
|
* 文件创建模拟
|
*
|
* @param vo VO
|
* @return {@link R}<{@link Object}>
|
*/
|
@ApiOperation(value = "文件创建")
|
@PostMapping(value = "/file_create", produces = "application/json; charset=UTF-8")
|
public R<Object> fileCreate(List<MultipartFile> files,@RequestBody @ApiParam("创建仿真视图类") CreateSimuVo vo) throws IOException {
|
// 上传文件路径
|
String targetDir = Paths.get(config.getOutPath(), "files").toString();
|
log.info("目标目录: {}", targetDir);
|
createDirectoriesIfNotExists(targetDir);
|
List<CompletableFuture<Void>> uploadTasks = files.stream()
|
.map(file -> CompletableFuture.runAsync(() -> {
|
try {
|
file.transferTo(Paths.get(targetDir, file.getOriginalFilename()));
|
} catch (IOException e) {
|
log.error("文件上传失败", e);
|
throw new RuntimeException("文件上传失败");
|
}
|
}))
|
.collect(Collectors.toList());
|
CompletableFuture<Void> allUploadTasks = CompletableFuture.allOf(uploadTasks.toArray(new CompletableFuture[0]));
|
try {
|
allUploadTasks.get();
|
log.info("文件上传成功");
|
} catch (InterruptedException | ExecutionException e) {
|
log.error("文件上传失败", e);
|
}
|
|
try {
|
if (null == vo.getTotal() || vo.getTotal() < 1 || vo.getTotal() > 1000) {
|
return fail("降雨量不能为空，且取值在1~1000之间");
|
}
|
if (null == vo.getDuration() || vo.getDuration() < 1 || vo.getDuration() > 10080) {
|
return fail("仿真时长不能为空，且取值在1~10080之间");
|
}
|
if (null == vo.getMinx() || null == vo.getMiny() || null == vo.getMaxx() || null == vo.getMaxy()) {
|
return fail("选择范围不能为空", false);
|
}
|
if (null == vo.getPid() || vo.getPid() < 0) {
|
vo.setPid(0);
|
}
|
if (null == vo.getNum() || vo.getNum() < 1) {
|
vo.setNum(simuService.getMaxId() + 1);
|
}
|
if (null == vo.getPid() || vo.getPid() < 0) {
|
vo.setPid(0);
|
}
|
if (null == vo.getStartTime()) {
|
vo.setStartTime(new Date());
|
}
|
if (vo.getPid() > 0) {
|
SimuPo pp = simuService.getSimuByPid(vo.getPid());
|
if (null == pp) {
|
return fail("pid不存在");
|
}
|
if (null == vo.getFloodStart() || vo.getFloodStart() < 1 || vo.getFloodStart() > vo.getDuration() * 60) {
|
return fail("防汛开始时间不能为空，且取值在1~" + (vo.getDuration() * 60) + "之间");
|
}
|
if (null == vo.getFloodEnd() || vo.getFloodEnd() < vo.getFloodStart() || vo.getFloodEnd() > vo.getDuration() * 60) {
|
return fail("防汛结束时间不能为空，且取值在" + vo.getFloodStart() + "~" + (vo.getDuration() * 60) + "之间");
|
}
|
if (null == vo.getFloodHeight() || vo.getFloodHeight() < 1 || vo.getFloodHeight() > 2000) {
|
return fail("防汛高度不能为空，且取值在1~2000之间");
|
}
|
if (!FLOOD_TYPE.contains(vo.getFloodType())) {
|
return fail("防汛类型不能为空，且只能是沙袋和防水板");
|
}
|
if (null == vo.getFloodMinx() || null == vo.getFloodMiny() || null == vo.getFloodMaxx() || null == vo.getFloodMaxy()) {
|
return fail("防汛范围不能为空", false);
|
}
|
}
|
boolean flag = simuFilesService.createByfiles(vo);
|
return success(flag, flag ? "成功" : "失败");
|
} catch (Exception ex) {
|
return fail(ex, null);
|
}
|
}
|
|
|
|
|
|
|
|
|
}
|
