张洋洋
2025-02-18 5c87778c13e3e33bb30edfcc344e5e69754bc9bf
src/main/java/com/se/simu/service/SimuService.java
@@ -20,20 +20,15 @@
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;
import javax.annotation.Resource;
import java.io.File;
import java.sql.Timestamp;
import java.util.Date;
import java.util.List;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
/**
 * 仿真服务类
 *
 * @author WWW
 * @date   2024-09-18
 */
@Slf4j
@Service
@SuppressWarnings("ALL")
@@ -53,9 +48,6 @@
    @Resource
    ResultService resultService;
    /**
     * 获取
     */
    public IPage<SimuPo> get(SimuVo vo) {
        QueryWrapper<SimuPo> wrapper = getPageWrapper(vo);
@@ -78,7 +70,7 @@
        if (null != vo.getNum()) {
            wrapper.eq("num", vo.getNum());
        }
        if (!StringUtils.isEmpty(vo.getName())) {
        if (!StringHelper.isEmpty(vo.getName())) {
            wrapper.like("lower(name)", vo.getName().trim().toLowerCase());
        }
        if (!CollUtil.isEmpty(vo.getStatus())) {
@@ -88,9 +80,6 @@
        return wrapper;
    }
    /**
     * 删除
     */
    public int del(List<Integer> ids) {
        List<SimuPo> list = simuMapper.selectBatchIds(ids);
        if (null != list && list.size() > 0) {
@@ -103,6 +92,7 @@
                    delDir(config.getInPath() + File.separator + dp.getInPath());
                    delDir(config.getOutPath() + File.separator + dp.getOutPath());
                    FileUtil.del(config.getInPath() + File.separator + dp.getInPath() + ".json");
                } catch (Exception ex) {
                    log.error(ex.getMessage(), ex);
                }
@@ -132,13 +122,36 @@
        return simuMapper.selectOne(wrapper);
    }
    public SimuPo getSimuById(Integer id) {
        return simuMapper.selectById(id);
    }
    public SimuPo getSimuByServiceName(String serviceName) {
        if (StringHelper.isEmpty(serviceName)) {
            return null;
        }
        QueryWrapper<SimuPo> wrapper = new QueryWrapper<>();
        wrapper.eq("service_name", serviceName);
        wrapper.last("limit 1");
        return simuMapper.selectOne(wrapper);
    }
    public boolean create(CreateSimuVo vo) {
        Date now = new Date();
        String date = StringHelper.YMDHMS2_FORMAT.format(now);
        if (StringHelper.isEmpty(vo.getName())) {
            vo.setName(date);
        }
        DataPo data = BeanUtil.copyProperties(vo, DataPo.class);
        data.setPath(vo.getName(), vo.getName());
        data.setPath(date, date);
        initPath(data);
        SimuPo simu = new SimuPo(vo.getNum(), vo.getPid(), vo.getName(), JSONUtil.toJsonStr(data), 0, vo.getBak());
        simu.setServiceName(date);
        simu.setCreateTime(new Timestamp(now.getTime()));
        int rows = simuMapper.insert(simu);
        if (rows > 0) {
@@ -173,12 +186,6 @@
        executor.shutdown();
    }
    /**
     * 处理数据,状态标识:
     * 0-创建仿真任务,1-连接GEDB库,2-下载空间数据,3-下载高程数据,
     * 4-生成降雨文件,5-生成配置文件,6-模拟内涝仿真,7-处理水位文件,
     * 8-处理排水文件,9-处理仿真结果,10-完成,-10-出错
     */
    private void cope(SimuPo simu) {
        try {
            DataPo data = JSONUtil.toBean(simu.getData(), DataPo.class);
@@ -204,18 +211,18 @@
            uwService.callExe(data);
            update(simu, 7, null);
            uwService.copeWaterFiles();
            //uwService.copeWaterFiles();
            update(simu, 8, null);
            uwService.copeDrainFiles();
            uwService.copeDrainFiles(data);
            update(simu, 9, null);
            //resultService.process(data);
            resultService.process(data);
            //update(simu, 10, "完成");
            update(simu, 10, "完成");
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            update(simu, -10, ex.getMessage());
            update(simu, -simu.getStatus(), ex.getMessage());
        }
    }