disaster_sever/src/main/java/com/miniframe/bisiness/system/D00008Service.java

package com.miniframe.bisiness.system;

import com.miniframe.core.ExecProcessFlow;
import com.miniframe.core.ext.UtilTools;
import com.miniframe.disaster.service.LogService;
import com.miniframe.generate.business.system.model.D00008BaseModel;
import com.miniframe.model.system.*;
import com.miniframe.model.system.dao.DAccidentMapper;
import com.miniframe.model.system.dao.DJobMapper;
import com.miniframe.model.system.dao.DNodeMapper;
import com.miniframe.model.system.dao.DNodeValMapper;
import com.miniframe.service.impl.FileServiceImpl;
import com.miniframe.template.TemplateGenerator;
import com.miniframe.tools.XIDateTimeUtils;
import com.miniframe.websocket.WebsocketEndPoint;
import lombok.SneakyThrows;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.List;
import java.util.Map;
import java.util.concurrent.CompletableFuture;

/**
 * 基础系统,“灾情演练”逻辑处理（重新生成不覆盖）。
 */
public class D00008Service extends D00008BaseModel implements ExecProcessFlow {
	
	private static final long serialVersionUID = -7051358269847459502L;
		
	/**
	 * 基础系统,“灾情演练”业务核心处理
 	 */
	public void transExecute() throws Exception {
		Integer aid =this.getA_d00008().getAid();
		/**
		 * totaltime: 10800  - 模拟时长
		 * dt: 0.1   - 时间步长
		 * dx: 10.0 -空间步长
		 * report step: 60 -输出步长
		 * interactionstep : 1200 -交互步长
		 */
		String totaltime =this.getA_d00008().getTotaltime();
		String dt =this.getA_d00008().getDt();
		String dx =this.getA_d00008().getDx();
		String reportstep =this.getA_d00008().getReportstep();
		String interactionstep =this.getA_d00008().getInteractionstep();
		String acctime =this.getA_d00008().getAcctime();
		DAccidentMapper accidentDao  = UtilTools.getBean(DAccidentMapper.class);
		DAccident accident= accidentDao.selectByPrimaryKey(aid);
		DJobMapper dJobDao = UtilTools.getBean(DJobMapper.class);


		DJob job = CreateJob(aid, totaltime, dt, dx, reportstep, interactionstep, acctime);
		dJobDao.insertSelective(job);
		//获取最新JOB
		DJob newJob = getNewJob(aid, dJobDao);
		Integer jid =newJob.getId();
		//存储最新求解ID
		changeAccident(accidentDao, accident, jid);
		//创建日志
		LogService.createLog(accident);

		//获取传感器数据
		//TODO:  需要从第三方获取
		initDNodeVal(aid);

		if(accident.getStype().equals("Fire")){
			TemplateGenerator.createFireControl(aid,jid,totaltime,dt,dx,reportstep,interactionstep);
			LogService.addLog(accident,"fireControl.ftl配置生成——————————————————成功");
			TemplateGenerator.createFireRunsh(aid,jid);
			LogService.addLog(accident,"runFile.sh配置生成——————————————————成功");
			TemplateGenerator.createFireInit(aid,jid);
			LogService.addLog(accident,"Fire.init配置生成——————————————————成功");
			exeFire(aid,jid);
		}
		if(accident.getStype().equals("Water")){
			TemplateGenerator.createWaterControl(aid,jid,totaltime,dt,dx,reportstep,interactionstep);
			LogService.addLog(accident,"water.control配置生成——————————————————成功");
			TemplateGenerator.createWaterRunsh(aid,jid);
			LogService.addLog(accident,"runWater.sh配置生成——————————————————成功");
			TemplateGenerator.createWaterInit(aid,jid);
			LogService.addLog(accident,"water.init配置生成——————————————————成功");
			exeWater(aid,jid);
		}

	}

	private DJob CreateJob(Integer aid, String totaltime, String dt, String dx, String reportstep, String interactionstep, String acctime) {
		DJob job =new DJob();
		job.setAid(aid);
		job.setAcctime(acctime);
		job.setTotaltime(totaltime);
		job.setDt(dt);
		job.setDx(dx);
		job.setReportstep(reportstep);
		job.setInteractionstep(interactionstep);
		job.setStarttime(XIDateTimeUtils.getNowStr());
		job.setState("0");//未执行
		return job;
	}

	private DJob getNewJob(Integer aid, DJobMapper dJobDao) {
		DJobSQLBuilder jsb = new DJobSQLBuilder();
		DJobSQLBuilder.Criteria jsc =jsb.createCriteria();
		jsc.andAidEqualTo(aid);
		jsb.setOrderByClause("starttime desc");
		DJob newJob= dJobDao.selectByExample(jsb).get(0);
		return newJob;
	}

	private void changeAccident(DAccidentMapper accidentDao, DAccident accident, Integer jid) {
		accident.setJid(jid);
		accidentDao.updateByPrimaryKey(accident);
	}

	private void initDNodeVal(Integer aid) {
		DNodeValMapper dnvm= UtilTools.getBean(DNodeValMapper.class);
		DNodeValSQLBuilder sb =new DNodeValSQLBuilder();
		DNodeValSQLBuilder.Criteria sc =sb.createCriteria();
		sc.andAidEqualTo(aid);
		dnvm.deleteByExample(sb);
		DNodeMapper dnm = UtilTools.getBean(DNodeMapper.class);
		List<DNode> nlist = dnm.selectAll();
		for (DNode dn:nlist) {
			DNodeVal nv = new DNodeVal();
			nv.setAid(aid);
			nv.setNid(dn.getId());
			nv.setNname(dn.getName());
			nv.setVal1(0.0f);
			nv.setVal2(300.0f);
			nv.setVal3(0.0f);
			nv.setVal4(0.0f);
			nv.setNtype(dn.getNtype());
			dnvm.insertSelective(nv);
		}
	}

	//异步执行
	@SneakyThrows
	public void exeWater(Integer aid,Integer jid) throws Exception{
		CompletableFuture<Void> cf = CompletableFuture.supplyAsync(()->{
			try {

				LogService.addLog(aid,jid,"Water","求解——————————————————开始");
				Runtime runtime = Runtime.getRuntime();
				Process p =null;
				p= runtime.exec("sh "+TemplateGenerator.BPATH+"/"+aid+"/"+jid+"/water"+"/"+"runWater.sh");
				InputStream fis = p.getInputStream();
				InputStreamReader isr = new InputStreamReader(fis);
				BufferedReader br = new BufferedReader(isr);
				String line = null;
				while ((line = br.readLine()) != null) {
					LogService.addLog(aid,jid,"Water",line);
					System.out.println(line);
				}
				LogService.addLog(aid,jid,"Water","求解——————————————————成功");
			} catch (IOException e) {
				LogService.addLog(aid,jid,"Water","求解——————————————————失败");
				e.printStackTrace();
			}
			return null;
		});

	}

	//异步执行
	@SneakyThrows
	public void exeFire(Integer aid,Integer jid) throws Exception{

		CompletableFuture<Void> cf = CompletableFuture.supplyAsync(()->{
			try {
			Runtime runtime = Runtime.getRuntime();
			Process p =null;
			LogService.addLog(aid,jid,"Fire","求解——————————————————开始");
			p= runtime.exec("sh "+TemplateGenerator.BPATH+"/"+
					aid+"/"+jid+"/fire"+"/"+"runFile.sh");
			InputStream fis = p.getInputStream();
			InputStreamReader isr = new InputStreamReader(fis);
			BufferedReader br = new BufferedReader(isr);
			String line = null;
			while ((line = br.readLine()) != null) {
				LogService.addLog(aid,jid,"Fire",line);
				System.out.println(line);
			}
			LogService.addLog(aid,jid,"Fire","求解——————————————————成功");
			} catch (IOException e) {
				LogService.addLog(aid,jid,"Fire","求解——————————————————失败");
				e.printStackTrace();
			}
			return null;
		});

	}
	
	/**
	 *
	 *
	 *
	 *
	 * 基础系统,“灾情演练”业务前处理
 	 */
	public void preTransFlow() throws Exception {

		this.validater();
	}
	
	/**
	 * 基础系统,“灾情演练”业务后处理
 	 */
	public void afterTransFlow() throws Exception {
	
	}
	
	/**
	 * 基础系统,“灾情演练”逻辑入口处理方法
 	 */
 	@SuppressWarnings("rawtypes")
	@Override
	public Map execute(Map vars) throws Exception {
		this.setTransMap(vars);
		preTransFlow();// 执行业务开始的规则检查和校验
		transExecute();// 执行核心业务段
		afterTransFlow();// 执行核心逻辑完成后的收尾逻辑
		return this.getTransMap();
	}

}