计费规则指标处理

xuhy

4 天以前 fe1bb1affe25ba41a0f01c3c07347603a69f31cd

 ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TMissionController.java

@@ -1,8 +1,29 @@
package com.ruoyi.web.controller.api;


import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
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>
@@ -12,9 +33,312 @@
 * @author xiaochen
 * @since 2025-09-28
 */
@Api(tags = "任务管理")
@RestController
@RequestMapping("/t-mission")
@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);
    }

}