11

13693261870

2024-11-11 138b959cc11dc9a73b0c766030b99ba1180d8650

 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);
                }
@@ -133,12 +123,32 @@
        return simuMapper.selectOne(wrapper);
    }

    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 +183,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);
@@ -195,27 +199,27 @@
            gedbService.copeDem(token, data);

            update(simu, 4, null);
            uwService.createRainFile();
            uwService.createRainFile(data);

            update(simu, 5, null);
            uwService.createConfig(data);

            update(simu, 6, null);
            uwService.callExe();
            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();
            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());
        }
    }