liyudong/simulation-backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

初次提交

Browse Source
This commit is contained in:
李玉东 2025-09-14 21:56:23 +08:00
parent 1525954f1a
commit 2f40a8556c
10 changed files with 129 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/main/java/com/hivekion/baseData/entity/OrgSupplier.java
View File

@ -33,5 +33,6 @@ public class OrgSupplier implements Serializable {
@TableField(exist = false)
private String name;
}

4
src/main/java/com/hivekion/baseData/mapper/OrgSupplierMapper.java
View File

@ -2,7 +2,7 @@ package com.hivekion.baseData.mapper;
import com.hivekion.baseData.entity.OrgSupplier;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import java.util.List;
/**
* <p>
* Mapper 接口
@ -12,5 +12,5 @@ import com.baomidou.mybatisplus.core.mapper.BaseMapper;
* @since 2025-09-14
*/
public interface OrgSupplierMapper extends BaseMapper<OrgSupplier> {
List<OrgSupplier> selectByOrgIds(List<Integer> ids);
}

3
src/main/java/com/hivekion/baseData/service/OrgSupplierService.java
View File

@ -13,5 +13,6 @@ import java.util.List;
* @since 2025-09-14
*/
public interface OrgSupplierService extends IService<OrgSupplier> {
public List<OrgSupplier> getByOrgId(Integer orgId);
List<OrgSupplier> getByOrgId(Integer orgId);
List<OrgSupplier> selectByOrgIds(List<Integer> ids);
}

5
src/main/java/com/hivekion/baseData/service/impl/OrgSupplierServiceImpl.java
View File

@ -27,4 +27,9 @@ public class OrgSupplierServiceImpl extends ServiceImpl<OrgSupplierMapper, OrgSu
queryWrapper.eq("org_id", orgId);
return this.list(queryWrapper);
}
@Override
public List<OrgSupplier> selectByOrgIds(List<Integer> ids) {
return this.baseMapper.selectByOrgIds(ids);
}
}

1
src/main/java/com/hivekion/scenario/service/IScenarioOrgPostService.java
View File

@ -17,4 +17,5 @@ public interface IScenarioOrgPostService extends IService<ScenarioOrgPost> {
void removeByCondition(ScenarioOrgPost scenarioOrgPost);
List<ScenarioOrgPost> selectByCondition(ScenarioOrgPost scenarioOrgPost);
List<Integer> getSelectOrgId(ScenarioOrgPost scenarioOrgPost);
}

63
src/main/java/com/hivekion/scenario/service/impl/ScenarioTaskServiceImpl.java
View File

@ -136,33 +136,50 @@ public class ScenarioTaskServiceImpl extends
* @param roomId 房间ID
*/
private void weatherTrigger(Scenario currentScenario, String roomId) {
try {
QueryWrapper<WeatherResource> weatherResourceQueryWrapper = new QueryWrapper<>();
weatherResourceQueryWrapper.eq("scenario_id", currentScenario.getId());
List<WeatherResource> weatherResourceList = this.weatherResourceService.list(weatherResourceQueryWrapper);
ResponseCmdInfo<JSONArray> responseCmdInfo = new ResponseCmdInfo();
responseCmdInfo.setScenarioId(currentScenario.getId());
responseCmdInfo.setRoom(roomId);
JSONArray weatherMsgArray = new JSONArray();
for (WeatherResource weatherResource : weatherResourceList) {
JSONObject jsonObject = new JSONObject();
DateTimeFormatter dtf = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
jsonObject.putOnce("begTime", dtf.format(weatherResource.getLastBegTime()));
jsonObject.putOnce("endTime", dtf.format(weatherResource.getLastEndTime()));
weatherMsgArray.add(jsonObject);
responseCmdInfo.setCmdType("66-" + weatherResource.getWeatherType());
}
responseCmdInfo.setData(weatherMsgArray);
System.out.println(responseCmdInfo.toString());
Global.sendCmdInfoQueue.add(responseCmdInfo);
} catch (Exception ex) {
//获取到 reids中天气数据
//每个天气开始遍历
//获取天气开始的时间 ,最好加一个状态，提示天气任务的开始，运行
//想定的开始时间+想定目前持续的时间
log.error(ex.getMessage());
}
// try {
// QueryWrapper<WeatherResource> weatherResourceQueryWrapper = new QueryWrapper<>();
// weatherResourceQueryWrapper.eq("scenario_id", currentScenario.getId());
// List<WeatherResource> weatherResourceList = this.weatherResourceService.list(weatherResourceQueryWrapper);
// ResponseCmdInfo<JSONArray> responseCmdInfo = new ResponseCmdInfo();
// responseCmdInfo.setScenarioId(currentScenario.getId());
// responseCmdInfo.setRoom(roomId);
// JSONArray weatherMsgArray = new JSONArray();
// for (WeatherResource weatherResource : weatherResourceList) {
// JSONObject jsonObject = new JSONObject();
// DateTimeFormatter dtf = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
// jsonObject.putOnce("begTime", dtf.format(weatherResource.getLastBegTime()));
// jsonObject.putOnce("endTime", dtf.format(weatherResource.getLastEndTime()));
// weatherMsgArray.add(jsonObject);
// responseCmdInfo.setCmdType("66-" + weatherResource.getWeatherType());
// }
// responseCmdInfo.setData(weatherMsgArray);
// System.out.println(responseCmdInfo.toString());
// Global.sendCmdInfoQueue.add(responseCmdInfo);
// } catch (Exception ex) {
//
// log.error(ex.getMessage());
// }
}
/**
* 获取当前想定从开始到现在时间
* @param scenario
* @param roomId
* @return
*/
private int getCurrentDuringTime(Scenario scenario, String roomId) {
Object duringTime = redisUtil.hget(roomId + "_" + scenario.getId(), "duringTime");
if (duringTime != null) {
return (Integer) duringTime;
}
return 0;
}
private void taskTrigger(Scenario currentScenario, String roomId) {
try{
Object statusObj = redisUtil.hget(roomId + "_" + currentScenario.getId(), "status");

6
src/main/java/com/hivekion/scenario/service/impl/TaskLogicServiceImpl.java
View File

@ -191,6 +191,12 @@ public class TaskLogicServiceImpl implements TaskLogicService {
Global.sendCmdInfoQueue.add(cmdInfo);
}
/**
* 获取当前想定从开始到现在时间
* @param scenario
* @param roomId
* @return
*/
private int getCurrentDuringTime(Scenario scenario, String roomId) {
Object duringTime = redisUtil.hget(roomId + "_" + scenario.getId(), "duringTime");
if (duringTime != null) {

45
src/main/java/com/hivekion/statistic/service/impl/StatisticServiceImpl.java
View File

@ -2,7 +2,9 @@ package com.hivekion.statistic.service.impl;
import com.hivekion.baseData.entity.OrgSupplier;
import com.hivekion.baseData.service.OrgSupplierService;
import com.hivekion.scenario.entity.ScenarioOrgPost;
import com.hivekion.scenario.entity.ScenarioResource;
import com.hivekion.scenario.service.IScenarioOrgPostService;
import com.hivekion.scenario.service.ScenarioResourceService;
import com.hivekion.statistic.bean.StatisticBean;
import com.hivekion.statistic.service.StatisticService;
@ -10,7 +12,9 @@ import com.hivekion.supplier.entity.SuppliesDict;
import com.hivekion.supplier.service.SuppliesDictService;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import javax.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
/**
@ -23,6 +27,7 @@ import org.springframework.stereotype.Service;
* @since 2025/7/22
*/
@Service
@Slf4j
public class StatisticServiceImpl implements StatisticService {
@Resource
@ -31,7 +36,8 @@ public class StatisticServiceImpl implements StatisticService {
private OrgSupplierService orgSupplierService;
@Resource
private SuppliesDictService suppliesDictService;
@Resource
private IScenarioOrgPostService scenarioOrgPostService;
@Override
public StatisticBean statistic(String resourceId) {
StatisticBean statisticBean = new StatisticBean();
@ -40,37 +46,50 @@ public class StatisticServiceImpl implements StatisticService {
if(resourceInstance == null){
return statisticBean;
}
statisticBean.getTeam().setType(resourceInstance.getType());
statisticBean.getTeam().setLat(resourceInstance.getLat());
statisticBean.getTeam().setLng(resourceInstance.getLng());
statisticBean.getTeam().setTeamName(resourceInstance.getResourceName());
//获取关联的组织机构信息
ScenarioOrgPost post = new ScenarioOrgPost();
post.setResourceId(resourceId);
List<ScenarioOrgPost> orgPostList = scenarioOrgPostService.selectByCondition(post);
List<Integer> orgList = orgPostList.stream().map(ScenarioOrgPost::getOrgId)
.collect(Collectors.toList());
//获取人员信息
//获取物资信息
List<OrgSupplier> suppliers = orgSupplierService.getByOrgId(resourceInstance.getResourceId());
List<OrgSupplier> suppliers = orgSupplierService.selectByOrgIds(orgList);
log.info("suppliers size::{}",suppliers.size());
Map<String, SuppliesDict> supplierMap = suppliesDictService.supplierDictMap();
log.info("supplierMap::{}",supplierMap);
suppliers.forEach(supplier -> {
SuppliesDict dict = supplierMap.get(supplier.getId());
log.info("supplier:{},{}",supplier,supplier.getId());
SuppliesDict dict = supplierMap.get(supplier.getSupplierId());
log.info("dict:{}",dict);
if (dict != null) {
switch (dict.getCode()) {
case "FOOD":
statisticBean.getFood().setTotal(supplier.getAccount());
statisticBean.getFood().setCurrent(supplier.getAccount());
statisticBean.getFood().setTotal(statisticBean.getFood().getTotal()+supplier.getAccount());
statisticBean.getFood().setCurrent(statisticBean.getFood().getCurrent()+supplier.getAccount());
break;
case "WATER":
statisticBean.getWater().setTotal(supplier.getAccount());
statisticBean.getWater().setCurrent(supplier.getAccount());
statisticBean.getWater().setTotal(statisticBean.getWater().getTotal()+supplier.getAccount());
statisticBean.getWater().setCurrent(statisticBean.getWater().getCurrent()+supplier.getAccount());
break;
case "FUEL":
statisticBean.getFuel().setTotal(supplier.getAccount());
statisticBean.getFuel().setCurrent(supplier.getAccount());
statisticBean.getFuel().setTotal(statisticBean.getFuel().getTotal()+supplier.getAccount());
statisticBean.getFuel().setCurrent(statisticBean.getFuel().getCurrent()+supplier.getAccount());
break;
case "MEDICAL":
statisticBean.getMedical().setTotal(supplier.getAccount());
statisticBean.getMedical().setCurrent(supplier.getAccount());
statisticBean.getMedical().setTotal( statisticBean.getMedical().getTotal()+supplier.getAccount());
statisticBean.getMedical().setCurrent( statisticBean.getMedical().getCurrent()+supplier.getAccount());
break;
case "AMMUNITION":
statisticBean.getAmmunition().setTotal(supplier.getAccount());
statisticBean.getAmmunition().setCurrent(supplier.getAccount());
statisticBean.getAmmunition().setTotal( statisticBean.getAmmunition().getTotal()+supplier.getAccount());
statisticBean.getAmmunition().setCurrent(statisticBean.getAmmunition().getCurrent()+supplier.getAccount());
break;
default:
break;

27
src/main/java/com/hivekion/ws/WsServer.java
View File

@ -1,7 +1,10 @@
package com.hivekion.ws;
import com.alibaba.fastjson2.JSON;
import com.hivekion.Global;
import com.hivekion.common.entity.RequestCmdInfo;
import com.hivekion.common.entity.ResponseCmdInfo;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import javax.websocket.OnClose;
@ -57,6 +60,12 @@ public class WsServer {
}
try{
session.getBasicRemote().sendText(testWeatherJson());
}catch (Exception e){
log.error("error::",e);
}
}
}
@ -111,10 +120,24 @@ public class WsServer {
if (roomMap.containsKey(room)) {
Map<String, Session> singleRoomMap = roomMap.get(room);
singleRoomMap.forEach((sessionId, session) -> {
session.getAsyncRemote().sendText(message);
try{
session.getBasicRemote().sendText(message);
}catch (Exception e){
log.error("error::",e);
}
});
}
}
}
private String testWeatherJson(){
ResponseCmdInfo responseCmdInfo = new ResponseCmdInfo();
responseCmdInfo.setCmdType("66-raining");
responseCmdInfo.setRoom("123");
responseCmdInfo.setScenarioId(2746);
Map<String,Object> data = new HashMap<>();
data.put("begTime","2028-03-09 10:00:00");
data.put("endTime","2028-03-19 00:20:00");
return JSON.toJSONString(responseCmdInfo);
}
}

16
src/main/resources/mapper/tbl/OrgSupplierMapper.xml
View File

@ -1,5 +1,19 @@
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.hivekion.baseData.mapper.OrgSupplierMapper">
<resultMap id="DMOrgSupplier" type="com.hivekion.baseData.entity.OrgSupplier">
<result property="id" column="id"/>
<result property="supplierId" column="supplier_id"/>
<result property="account" column="account"/>
<result property="orgId" column="org_id"/>
<result property="name" column="name"/>
<result property="code" column="code"/>
<!-- 其他字段 -->
</resultMap>
<select id="selectByOrgIds" resultMap="DMOrgSupplier">
select * from TBL_ORG_SUPPLIER where org_id in
<foreach collection="ids" item="id" open="(" separator="," close=")">
#{id}
</foreach>
</select>
</mapper>