计费规则指标处理

xuhy

3 天以前 fe1bb1affe25ba41a0f01c3c07347603a69f31cd

计费规则指标处理

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

@@ -2,25 +2,28 @@


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.TAppUser;
import com.ruoyi.system.model.TMission;
import com.ruoyi.system.model.*;
import com.ruoyi.system.query.TMissionQuery;
import com.ruoyi.system.service.TMissionReassignService;
import com.ruoyi.system.service.TMissionService;
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.List;
import java.util.Map;
import java.util.*;
import java.util.stream.Collectors;

/**
 * <p>
@@ -39,11 +42,21 @@
    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) {
    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;
    }

    /**
@@ -110,7 +123,221 @@
    @ApiOperation(value = "通过任务计算规则分数获取队伍")
    @PostMapping(value = "/api/t-mission/getTeamScores")
    public R<Map<String, List<TAppUser>>> getTeamScores(@RequestParam(value = "missionId") String missionId) {
        return R.ok();
        // 查询任务
        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);
    }

}

 ruoyi-system/src/main/java/com/ruoyi/system/model/TAppUser.java

@@ -10,7 +10,6 @@
import lombok.Data;
import lombok.EqualsAndHashCode;

import java.math.BigDecimal;
import java.util.List;

/**
@@ -91,12 +90,12 @@
    @ApiModelProperty(value = "用户专业")
    private List<TUserMajor> userMajors;

    @ApiModelProperty(value = "距离")
    @ApiModelProperty(value = "距离 km")
    @TableField(exist = false)
    private String userDistance;
    private Double userDistance;

    @ApiModelProperty(value = "评分")
    @TableField(exist = false)
    private BigDecimal userScore;
    private Double scoreTotal;

}

 ruoyi-system/src/main/java/com/ruoyi/system/utils/GDMapGeocodingUtil.java

New file
@@ -0,0 +1,37 @@
package com.ruoyi.system.utils;

/**
 * 高德地图的地理编码工具类
 */
public class GDMapGeocodingUtil {

    /**
     * 地球半径，单位m
     */
    private static final double EARTH_RADIUS = 6371000;
    private static double rad(double d) {
        return d * Math.PI / 180.0;
    }
    /**
     * @param lon1 第一点的精度
     * @param lat1 第一点的纬度
     * @param lon2 第二点的精度
     * @param lat2 第二点的纬度
     * @return 返回的距离，单位m
     * */
    public static double getDistance(double lon1,double lat1,double lon2, double lat2) {
        double radLat1 = rad(lat1);
        double radLat2 = rad(lat2);
        double a = radLat1 - radLat2;
        double b = rad(lon1) - rad(lon2);
        double s = 2 * Math.asin(Math.sqrt(Math.pow(Math.sin(a / 2), 2) + Math.cos(radLat1) * Math.cos(radLat2) * Math.pow(Math.sin(b / 2), 2)));
        s = s * EARTH_RADIUS;
        s = Math.round(s * 10000) / 10000;
        return s;
    }

    public static void main(String[] args) {
        System.err.println(getDistance(104.165799,30.586865,104.155819,31.585843));
    }

}