package com.ruoyi.web.controller.api;
|
|
|
import com.alibaba.fastjson.JSON;
|
import com.alibaba.fastjson.JSONArray;
|
import com.alibaba.fastjson.JSONObject;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.ruoyi.common.annotation.Log;
|
import com.ruoyi.common.basic.PageInfo;
|
import com.ruoyi.common.core.domain.R;
|
import com.ruoyi.common.enums.BusinessType;
|
import com.ruoyi.framework.web.service.TokenService;
|
import com.ruoyi.system.model.*;
|
import com.ruoyi.system.query.TMissionQuery;
|
import com.ruoyi.system.service.*;
|
import com.ruoyi.system.utils.GDMapGeocodingUtil;
|
import com.ruoyi.system.vo.TMissionReassignVO;
|
import com.ruoyi.system.vo.TMissionVO;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.util.CollectionUtils;
|
import org.springframework.web.bind.annotation.*;
|
|
import java.util.*;
|
import java.util.stream.Collectors;
|
|
/**
|
* <p>
|
* 任务管理 前端控制器
|
* </p>
|
*
|
* @author xiaochen
|
* @since 2025-09-28
|
*/
|
@Api(tags = "任务管理")
|
@RestController
|
@RequestMapping("")
|
public class TMissionController {
|
|
|
private final TMissionService missionService;
|
private final TMissionReassignService missionReassignService;
|
private final TokenService tokenService;
|
private final TMissionAssignRuleService missionAssignRuleService;
|
private final TMissionAssignRuleIndicatorService missionAssignRuleIndicatorService;
|
private final TAppUserService appUserService;
|
private final TAppUserEquipmentService appUserEquipmentService;
|
private final TUserMajorService userMajorService;
|
@Autowired
|
public TMissionController(TMissionService missionService, TMissionReassignService missionReassignService, TokenService tokenService, TMissionAssignRuleService missionAssignRuleService, TMissionAssignRuleIndicatorService missionAssignRuleIndicatorService, TAppUserService appUserService, TAppUserEquipmentService appUserEquipmentService, TUserMajorService userMajorService) {
|
this.missionService = missionService;
|
this.missionReassignService = missionReassignService;
|
this.tokenService = tokenService;
|
this.missionAssignRuleService = missionAssignRuleService;
|
this.missionAssignRuleIndicatorService = missionAssignRuleIndicatorService;
|
this.appUserService = appUserService;
|
this.appUserEquipmentService = appUserEquipmentService;
|
this.userMajorService = userMajorService;
|
}
|
|
/**
|
* 任务管理列表
|
*/
|
//@PreAuthorize("@ss.hasPermi('system:mission:pageListAssigned')")
|
@ApiOperation(value = "任务管理分页列表-待分配任务池", response = TMissionQuery.class)
|
@PostMapping(value = "/api/t-mission/pageListAssigned")
|
public R<PageInfo<TMissionVO>> pageListAssigned(@RequestBody String param) {
|
TMissionQuery query = JSON.parseObject(param, TMissionQuery.class);
|
return R.ok(missionService.pageListAssigned(query));
|
}
|
|
/**
|
* 任务管理列表
|
*/
|
//@PreAuthorize("@ss.hasPermi('system:mission:pageListReassignAudit')")
|
@ApiOperation(value = "任务管理分页列表-改派支援审核", response = TMissionReassignVO.class)
|
@PostMapping(value = "/api/t-mission/pageListReassignAudit")
|
public R<PageInfo<TMissionReassignVO>> pageListReassignAudit(@RequestBody String param) {
|
TMissionQuery query = JSON.parseObject(param, TMissionQuery.class);
|
return R.ok(missionService.pageListReassignAudit(query));
|
}
|
|
/**
|
* 任务管理列表
|
*/
|
//@PreAuthorize("@ss.hasPermi('system:mission:pageListMissionPool')")
|
@ApiOperation(value = "任务管理分页列表-抢单任务池", response = TMissionVO.class)
|
@PostMapping(value = "/api/t-mission/pageListMissionPool")
|
public R<PageInfo<TMissionVO>> pageListMissionPool(@RequestBody String param) {
|
TMissionQuery query = JSON.parseObject(param, TMissionQuery.class);
|
return R.ok(missionService.pageListMissionPool(query));
|
}
|
|
/**
|
* 添加任务管理管理
|
*/
|
//@PreAuthorize("@ss.hasPermi('system:mission:add')")
|
@Log(title = "任务管理信息-新增任务管理", businessType = BusinessType.INSERT)
|
@ApiOperation(value = "添加任务管理",response = TMission.class)
|
@PostMapping(value = "/api/t-mission/add")
|
public R<Boolean> add(@RequestBody String param) {
|
TMission dto = JSON.parseObject(param,TMission.class);
|
missionService.save(dto);
|
return R.ok();
|
}
|
|
/**
|
* 查看任务管理详情
|
*/
|
//@PreAuthorize("@ss.hasPermi('system:mission:detail')")
|
@ApiOperation(value = "查看任务管理详情")
|
@GetMapping(value = "/open/t-mission/getDetailById")
|
public R<TMission> getDetailById(@RequestParam String id) {
|
TMission equipment = missionService.getById(id);
|
return R.ok(equipment);
|
}
|
|
/**
|
* 通过任务计算规则分数获取队伍
|
*/
|
//@PreAuthorize("@ss.hasPermi('system:mission:detail')")
|
@ApiOperation(value = "通过任务计算规则分数获取队伍")
|
@PostMapping(value = "/api/t-mission/getTeamScores")
|
public R<Map<String, List<TAppUser>>> getTeamScores(@RequestParam(value = "missionId") String missionId) {
|
// 查询任务
|
TMission mission = missionService.getById(missionId);
|
if (mission == null) {
|
return R.fail("任务不存在");
|
}
|
// 通过任务类型获取规则信息
|
TMissionAssignRule missionAssignRule = missionAssignRuleService.getById(mission.getMissionTypeId());
|
if (missionAssignRule == null) {
|
return R.fail("任务类型不存在");
|
}
|
// 查询任务类型下的指标信息
|
List<TMissionAssignRuleIndicator> missionAssignRuleIndicators = missionAssignRuleIndicatorService.list(Wrappers.lambdaQuery(TMissionAssignRuleIndicator.class)
|
.eq(TMissionAssignRuleIndicator::getRuleId,missionAssignRule.getId()));
|
if (CollectionUtils.isEmpty(missionAssignRuleIndicators)) {
|
return R.fail("任务类型下没有指标");
|
}
|
// 查询所有用户
|
List<TAppUser> appUsers = appUserService.list(Wrappers.lambdaQuery(TAppUser.class)
|
.eq(TAppUser::getState,1)
|
.eq(TAppUser::getStatus,1));
|
|
if(CollectionUtils.isEmpty(appUsers)){
|
return R.ok(new HashMap<>());
|
}
|
|
List<String> appUserIds = appUsers.stream().map(TAppUser::getId).collect(Collectors.toList());
|
// 查询装备信息
|
List<TAppUserEquipment> appUserEquipmentList = appUserEquipmentService.list(Wrappers.lambdaQuery(TAppUserEquipment.class)
|
.in(TAppUserEquipment::getAppUserId,appUserIds));
|
// 查询专业信息
|
List<TUserMajor> userMajorList = userMajorService.list(Wrappers.lambdaQuery(TUserMajor.class)
|
.in(TUserMajor::getAppUserId,appUserIds));
|
|
for (TAppUser appUser : appUsers) {
|
double scoreTotal = 0;
|
// 匹配装备信息
|
List<TAppUserEquipment> userEquipments = appUserEquipmentList.stream().filter(appEquipment -> appUser.getId().equals(appEquipment.getAppUserId())).collect(Collectors.toList());
|
// 匹配专业信息
|
List<TUserMajor> userMajors = userMajorList.stream().filter(userMajor -> appUser.getId().equals(userMajor.getAppUserId())).collect(Collectors.toList());
|
|
// 计算用户到任务的距离
|
double distance = GDMapGeocodingUtil.getDistance(Double.parseDouble(mission.getMissionLon()), Double.parseDouble(mission.getMissionLat()), Double.parseDouble(appUser.getLon()), Double.parseDouble(appUser.getLat()));
|
distance = distance/1000;
|
appUser.setUserDistance(distance);
|
|
for (TMissionAssignRuleIndicator missionAssignRuleIndicator : missionAssignRuleIndicators) {
|
switch (missionAssignRuleIndicator.getIndicatorType()) {
|
case 1:
|
JSONArray equipmentIndicators = JSONObject.parseArray(missionAssignRuleIndicator.getContent());
|
// 装备指标分数计算
|
if(!CollectionUtils.isEmpty(userEquipments)){
|
// 获取人员装备信息
|
TAppUserEquipment appUserEquipment = userEquipments.get(0);
|
// 装备信息
|
JSONArray jsonArray = JSONObject.parseArray(appUserEquipment.getEquipmentCount());
|
for (Object userEquipment : jsonArray) {
|
// 人员单个装备
|
JSONObject jsonObject = JSONObject.parseObject(JSONObject.toJSONString(userEquipment));
|
String equipmentId = jsonObject.getString("id");
|
Double count = jsonObject.getDouble("count");
|
// 匹配装备指标得分
|
List<Object> objectList = equipmentIndicators.stream().filter(content -> equipmentId.equals(JSONObject.parseObject(JSONObject.toJSONString(content)).getString("equipmentId"))).collect(Collectors.toList());
|
if(!CollectionUtils.isEmpty(objectList)){
|
String jsonString = JSONObject.toJSONString(objectList.get(0));
|
JSONObject rule = JSONObject.parseObject(jsonString);
|
Integer value = rule.getInteger("value");
|
Integer score = rule.getInteger("score");
|
if(count >= value){
|
// 得基准分值
|
scoreTotal += score;
|
}else{
|
// 计算分值 数量 / 基准值 * 基准分值
|
scoreTotal += count / value * score;
|
}
|
}
|
}
|
}
|
break;
|
case 2:
|
JSONArray personIndicators = JSONObject.parseArray(missionAssignRuleIndicator.getContent());
|
// 人员指标分数计算
|
if(!CollectionUtils.isEmpty(userEquipments)){
|
TAppUserEquipment appUserEquipment = userEquipments.get(0);
|
Integer personCount = appUserEquipment.getPersonCount();
|
// 实训次数
|
Integer practicalTrainCount = appUser.getPracticalTrainCount();
|
for (int i = 0; i < 3; i++) {
|
JSONObject jsonObject = JSONObject.parseObject(JSONObject.toJSONString(personIndicators.get(i)));
|
Integer value = jsonObject.getInteger("value");
|
Integer score = jsonObject.getInteger("score");
|
// 人员数量
|
if(0 == i){
|
if(personCount >= value){
|
// 得基准分值
|
scoreTotal += score;
|
}else{
|
// 人员数量 / 基准值 * 基准分值
|
scoreTotal += (double)personCount / value * score;
|
}
|
}
|
// 训练次数
|
if(1 == i){
|
if(practicalTrainCount >= value){
|
// 得基准分值
|
scoreTotal += score;
|
}else{
|
// 训练次数 / 基准值 * 基准分值
|
scoreTotal += (double)practicalTrainCount / value * score;
|
}
|
}
|
// 技能等级
|
if(2 == i){
|
// TODO 技能等级
|
}
|
}
|
}
|
break;
|
case 3:
|
// 专业指标
|
JSONArray personMajorIndicators = JSONObject.parseArray(missionAssignRuleIndicator.getContent());
|
// 人员专业指标分数计算
|
if(!CollectionUtils.isEmpty(userMajors)){
|
// 人员专业
|
for (TUserMajor userMajor : userMajors) {
|
List<Object> personMajorIndicatorList = personMajorIndicators.stream().filter(content -> userMajor.getMajorId().equals(JSONObject.parseObject(JSONObject.toJSONString(content)).getString("majorId"))).collect(Collectors.toList());
|
if(!CollectionUtils.isEmpty(personMajorIndicatorList)){
|
String jsonString = JSONObject.toJSONString(personMajorIndicatorList.get(0));
|
JSONObject rule = JSONObject.parseObject(jsonString);
|
Integer value = rule.getInteger("value");
|
Integer score = rule.getInteger("score");
|
Integer userNumber = userMajor.getUserNumber();
|
if(userNumber >= value){
|
// 得基准分值
|
scoreTotal += score;
|
}else{
|
// 计算分值 数量 / 基准值 * 基准分值
|
scoreTotal += (double)userNumber / value * score;
|
}
|
}
|
}
|
}
|
break;
|
case 4:
|
JSONArray trainingIndicators = JSONObject.parseArray(missionAssignRuleIndicator.getContent());
|
// 训练指标分数计算
|
// 实训次数
|
Integer practicalTrainCount = appUser.getPracticalTrainCount();
|
for (int i = 0; i < 3; i++) {
|
JSONObject jsonObject = JSONObject.parseObject(JSONObject.toJSONString(trainingIndicators.get(i)));
|
Integer value = jsonObject.getInteger("value");
|
Integer score = jsonObject.getInteger("score");
|
// 侦察专业成绩
|
if(0 == i){
|
// TODO
|
}
|
// 训练次数
|
if(1 == i){
|
if(practicalTrainCount >= value){
|
// 得基准分值
|
scoreTotal += score;
|
}else{
|
// 训练次数 / 基准值 * 基准分值
|
scoreTotal += (double)practicalTrainCount / value * score;
|
}
|
}
|
}
|
break;
|
case 5:
|
JSONArray distanceIndicators = JSONObject.parseArray(missionAssignRuleIndicator.getContent());
|
// 距离指标分数计算
|
for (Object distanceIndicator : distanceIndicators) {
|
JSONObject jsonObject = JSONObject.parseObject(JSONObject.toJSONString(distanceIndicator));
|
Double value1 = jsonObject.getDouble("value1");
|
Double value2 = jsonObject.getDouble("value2");
|
Integer score = jsonObject.getInteger("score");
|
// 判断value1等于0
|
if(0 == value1 && distance < value2){
|
scoreTotal += score;
|
}else if(0 == value2 && distance > value1){
|
scoreTotal += (double)score / distance;
|
}else {
|
scoreTotal += (double)score / 2;
|
}
|
}
|
break;
|
}
|
}
|
appUser.setScoreTotal(scoreTotal);
|
}
|
|
Map<String, List<TAppUser>> result = new HashMap<>();
|
appUsers = appUsers.stream().sorted(Comparator.comparing(TAppUser::getScoreTotal).reversed()).collect(Collectors.toList());
|
result.put("teamList", appUsers);
|
|
List<TAppUser> teamRecommendList = new ArrayList<>();
|
|
switch (missionAssignRule.getRuleType()){
|
case 1:
|
teamRecommendList = appUsers.stream().filter(appUser -> appUser.getScoreTotal()>missionAssignRule.getRuleValue()).collect(Collectors.toList());
|
break;
|
case 2:
|
teamRecommendList = appUsers.stream().filter(appUser -> appUser.getScoreTotal()>=missionAssignRule.getRuleValue()).collect(Collectors.toList());
|
break;
|
case 3:
|
teamRecommendList = appUsers.stream().filter(appUser -> Objects.equals(appUser.getScoreTotal(), missionAssignRule.getRuleValue())).collect(Collectors.toList());
|
break;
|
case 4:
|
teamRecommendList = appUsers.stream().filter(appUser -> appUser.getScoreTotal()<=missionAssignRule.getRuleValue()).collect(Collectors.toList());
|
break;
|
case 5:
|
teamRecommendList = appUsers.stream().filter(appUser -> appUser.getScoreTotal()<missionAssignRule.getRuleValue()).collect(Collectors.toList());
|
break;
|
}
|
result.put("teamRecommendList",teamRecommendList);
|
return R.ok(result);
|
}
|
|
}
|