134 lines
5.2 KiB
Java
134 lines
5.2 KiB
Java
package com.hivekion.statistic.service.impl;
|
|
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
|
import com.hivekion.baseData.entity.Fightpowerstaff;
|
|
import com.hivekion.baseData.entity.OrgSupplier;
|
|
import com.hivekion.baseData.service.FightpowerstaffService;
|
|
import com.hivekion.baseData.service.OrgSupplierService;
|
|
import com.hivekion.scenario.entity.ScenarioOrgPost;
|
|
import com.hivekion.scenario.entity.ScenarioResource;
|
|
import com.hivekion.scenario.entity.ScenarioTask;
|
|
import com.hivekion.scenario.service.IScenarioOrgPostService;
|
|
import com.hivekion.scenario.service.ScenarioResourceService;
|
|
import com.hivekion.scenario.service.ScenarioTaskService;
|
|
import com.hivekion.statistic.bean.ScenarioInfo;
|
|
import com.hivekion.statistic.service.ScenarioService;
|
|
import com.hivekion.supplier.entity.SupplierRequest;
|
|
import com.hivekion.supplier.entity.SuppliesDict;
|
|
import com.hivekion.supplier.service.ISupplierRequestService;
|
|
import com.hivekion.supplier.service.SuppliesDictService;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.stream.Collectors;
|
|
import javax.annotation.Resource;
|
|
import org.springframework.stereotype.Component;
|
|
|
|
@Component("WebsocketScenarioService")
|
|
public class ScenarioServiceImpl implements ScenarioService {
|
|
|
|
@Resource
|
|
private ScenarioResourceService scenarioResourceService;
|
|
@Resource
|
|
private OrgSupplierService orgSupplierService;
|
|
@Resource
|
|
private SuppliesDictService suppliesDictService;
|
|
@Resource
|
|
private IScenarioOrgPostService scenarioOrgPostService;
|
|
@Resource
|
|
private FightpowerstaffService fightpowerstaffService;
|
|
|
|
@Resource
|
|
private ScenarioTaskService scenarioTaskService;
|
|
@Resource
|
|
private ISupplierRequestService supplierRequestService;
|
|
|
|
@Override
|
|
public ScenarioInfo listScenarioInfo(Integer scenarioId, String roomId, String resourceId) {
|
|
ScenarioInfo scenarioInfo = new ScenarioInfo();
|
|
//获取分队信息
|
|
ScenarioResource resourceInstance = scenarioResourceService.getById(resourceId);
|
|
//获取关联的组织机构信息
|
|
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<Fightpowerstaff> staffList = null;
|
|
if (!orgList.isEmpty()) {
|
|
staffList = fightpowerstaffService.queryByOrgIds(orgList);
|
|
} else {
|
|
staffList = new ArrayList<>();
|
|
}
|
|
if (orgList.isEmpty()) {
|
|
return new ScenarioInfo();
|
|
}
|
|
//获取物资信息
|
|
List<OrgSupplier> suppliers = orgSupplierService.selectByOrgIds(orgList);
|
|
if (suppliers == null || suppliers.isEmpty()) {
|
|
return new ScenarioInfo();
|
|
}
|
|
Map<String, SuppliesDict> supplierMap = suppliesDictService.supplierDictMap();
|
|
|
|
suppliers.forEach(supplier -> {
|
|
|
|
SuppliesDict dict = supplierMap.get(supplier.getSupplierId());
|
|
|
|
if (dict != null) {
|
|
switch (dict.getCode()) {
|
|
case "FOOD":
|
|
scenarioInfo.getFood()
|
|
.setTotal(scenarioInfo.getFood().getTotal() + supplier.getAccount());
|
|
scenarioInfo.getFood()
|
|
.setCurrent(scenarioInfo.getFood().getCurrent() + supplier.getAccount());
|
|
break;
|
|
case "WATER":
|
|
scenarioInfo.getWater()
|
|
.setTotal(scenarioInfo.getWater().getTotal() + supplier.getAccount());
|
|
scenarioInfo.getWater()
|
|
.setCurrent(scenarioInfo.getWater().getCurrent() + supplier.getAccount());
|
|
break;
|
|
case "FUEL":
|
|
scenarioInfo.getFuel()
|
|
.setTotal(scenarioInfo.getFuel().getTotal() + supplier.getAccount());
|
|
scenarioInfo.getFuel()
|
|
.setCurrent(scenarioInfo.getFuel().getCurrent() + supplier.getAccount());
|
|
break;
|
|
case "MEDICAL":
|
|
scenarioInfo.getMedical()
|
|
.setTotal(scenarioInfo.getMedical().getTotal() + supplier.getAccount());
|
|
scenarioInfo.getMedical()
|
|
.setCurrent(scenarioInfo.getMedical().getCurrent() + supplier.getAccount());
|
|
break;
|
|
|
|
case "AMMUNITION":
|
|
scenarioInfo.getAmmunition()
|
|
.setTotal(scenarioInfo.getAmmunition().getTotal() + supplier.getAccount());
|
|
scenarioInfo.getAmmunition()
|
|
.setCurrent(scenarioInfo.getAmmunition().getCurrent() + supplier.getAccount());
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
});
|
|
|
|
ScenarioTask scenarioTask = new ScenarioTask();
|
|
scenarioTask.setScenarioId(scenarioId);
|
|
scenarioTask.setRoomId(roomId);
|
|
scenarioTask.setResourceId(resourceId);
|
|
|
|
List<ScenarioTask> scenarioTasks = scenarioTaskService.queryTaskList(scenarioTask);
|
|
|
|
List<SupplierRequest> supplierRequests = supplierRequestService.list(
|
|
new QueryWrapper<SupplierRequest>().eq("FROM_RESOURCE_ID", resourceId));
|
|
|
|
scenarioInfo.setScenarioTasks(scenarioTasks);
|
|
scenarioInfo.setStaffList(staffList);
|
|
scenarioInfo.setOrgPostList(orgPostList);
|
|
scenarioInfo.setSuppliers(suppliers);
|
|
scenarioInfo.setSupplierRequests(supplierRequests);
|
|
return scenarioInfo;
|
|
}
|
|
}
|