simulation-backend/src/main/java/com/hivekion/room/bean/BattleRootTask.java

package com.hivekion.room.bean;

import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.extra.spring.SpringUtil;
import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONObject;
import com.hivekion.Global;
import com.hivekion.common.entity.ResponseCmdInfo;
import com.hivekion.common.redis.RedisUtil;
import com.hivekion.common.uuid.IdUtils;
import com.hivekion.scenario.entity.BattleConsume;
import com.hivekion.scenario.entity.ScenarioTask;
import com.hivekion.scenario.service.IBattleConsumeService;
import com.hivekion.statistic.bean.*;
import com.hivekion.statistic.service.StatisticService;
import com.hivekion.statistic.service.impl.StatisticServiceImpl;
import com.hivekion.supplier.entity.SupplierRequest;
import com.hivekion.supplier.service.ISupplierRequestService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.core.env.Environment;
import org.springframework.util.CollectionUtils;

import javax.swing.*;
import java.time.LocalDateTime;
import java.time.ZoneId;
import java.util.*;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;

/**
 * [类的简要说明]
 * <p>
 * [详细描述，可选]
 * <p>
 *
 * @author LiDongYU
 * @since 2025/7/22
 */
@Slf4j
public class BattleRootTask extends AbtParentTask  {

  private StatisticService statisticService = null;
  private RedisUtil redisUtil = null;

  private ISupplierRequestService supplierRequestService;

  private IBattleConsumeService battleConsumeService;

  private static final Integer DEATH_SPREED = 3;
  private static final Integer INJURED_SPREED = 3;
  private static final Double AMMUNITION_SPREED = 2.6D;
  private static final Double FOOD_SPREED = 2.3D;
  private static final Double WATER_SPREED = 3.6D;
  private static final Double FUEL_SPREED = 3.6D;
  private static final Double MEDICAL_SPREED = 1.6D;



  public BattleRootTask(ScenarioTask scenarioTask,String roomId) {
    super(scenarioTask,roomId);
  }

  private void initBean(){
    if(statisticService == null) {
      statisticService = SpringUtil.getBean(StatisticService.class);
    }
    if(redisUtil == null) {
      redisUtil = SpringUtil.getBean(RedisUtil.class);
    }
    if(supplierRequestService == null){
      supplierRequestService = SpringUtil.getBean(ISupplierRequestService.class);
    }
    if(battleConsumeService == null){
      battleConsumeService = SpringUtil.getBean(IBattleConsumeService.class);
    }
  }
  //执行一次
  @Override
  public void doSomeThing() {
    log.info("===============begin BattleRootTask=========={}==========",this.getRoomStatus());
    this.initBean();
    if(this.getRoomStatus()) {
      long initDuringTime = this.getDuringTime();
      redisUtil.hset(scenarioTask.getScenarioId() + "-" + roomId + "-" + scenarioTask.getResourceId(),"duringTime",String.valueOf(initDuringTime));
      String jsonStr = (String)redisUtil.hget(scenarioTask.getScenarioId() + "-" + roomId + "-" + scenarioTask.getResourceId(),"scenarioInfo");
      ScenarioInfo scenarioInfo =JSONObject.parseObject(jsonStr,ScenarioInfo.class);
      HashMap<String,Object> battleParams = new HashMap<>();
      //初始化本次战斗任务各种资源数
      battleParams.put("ammunition",Double.valueOf(scenarioInfo.getAmmunition().getTotal()).toString());
      battleParams.put("food",Double.valueOf(scenarioInfo.getFood().getTotal()).toString());
      battleParams.put("fuel",Double.valueOf(scenarioInfo.getFuel().getTotal()).toString());
      battleParams.put("medical",Double.valueOf(scenarioInfo.getMedical().getTotal()).toString());
      battleParams.put("water",Double.valueOf(scenarioInfo.getWater().getTotal()).toString());
      battleParams.put("person",Integer.valueOf(scenarioInfo.getPerson().getTotal()).toString());
      battleParams.put("death",Integer.valueOf(scenarioInfo.getPerson().getDeath()).toString());
      battleParams.put("injured",Integer.valueOf(scenarioInfo.getPerson().getInjured()).toString());
      battleParams.put("teamLat",scenarioInfo.getTeam().getLat().toString());
      battleParams.put("teamLng",scenarioInfo.getTeam().getLng().toString());
      battleParams.put("duringTime",Long.valueOf(initDuringTime).toString());
      log.info("===============================初始化本次战斗任务各种资源数====================================");
      double suppleAmount =scenarioInfo.getAmmunition().getTotal();
      int suppleDeath =scenarioInfo.getPerson().getDeath();
      int suppleInjured =scenarioInfo.getPerson().getInjured();
      final  Map<String,Boolean>  suppleFlagMap = new HashMap<>();
      suppleFlagMap.put("ammunition",false);
      suppleFlagMap.put("death",false);
      suppleFlagMap.put("injured",false);
      //定时检查统计各种资源消耗量
      this.createBattleTaskOnTimingHandle(new BizTaskOnTiming() {
        @Override
        public void execTask() {
          log.info("===============================定时检查统计各种资源消耗量 begin====================================");
          Long deathConsume = null;
          Long injuredConsume = null;
          Double ammunitionConsume = null;
          Double foodConsume = null;
          Double waterConsume = null;
          Double fuelConsume = null;
          Double medicalConsume = null;
          String teamLat = null;
          String teamLng = null;
          JSONObject jsonObject = new JSONObject();
          try {
            long duringTime = getDuringTime();
            long lastDuringTime = Long.valueOf(redisUtil.hget(scenarioTask.getScenarioId() + "-" + roomId + "-" + scenarioTask.getResourceId(),"duringTime").toString());
            long intervalDuringTime = duringTime - lastDuringTime;
            double ammunition = Double.valueOf(battleParams.get("ammunition").toString());
            double food = Double.valueOf(battleParams.get("food").toString());
            double fuel = Double.valueOf(battleParams.get("fuel").toString());
            double medical = Double.valueOf(battleParams.get("medical").toString());
            double water = Double.valueOf(battleParams.get("water").toString());
            long death = Long.valueOf(battleParams.get("death").toString());
            long injured = Long.valueOf(battleParams.get("injured").toString());
            teamLat = battleParams.get( "teamLat").toString();
            teamLng = battleParams.get( "teamLng").toString();
            //
            deathConsume =  DEATH_SPREED * intervalDuringTime;
            injuredConsume = INJURED_SPREED * intervalDuringTime;
            ammunitionConsume = intervalDuringTime * AMMUNITION_SPREED;
            foodConsume = intervalDuringTime * FOOD_SPREED;
            waterConsume = intervalDuringTime * WATER_SPREED;
            fuelConsume = intervalDuringTime * FUEL_SPREED;
            medicalConsume = intervalDuringTime * MEDICAL_SPREED;

            battleParams.put("ammunition",Double.valueOf(ammunition - ammunitionConsume).toString());
            battleParams.put("food",Double.valueOf(food - foodConsume).toString());
            battleParams.put("fuel",Double.valueOf(fuel - fuelConsume).toString());
            battleParams.put("medical",Double.valueOf(medical - medicalConsume).toString());
            battleParams.put("water",Double.valueOf(water - waterConsume).toString());
            battleParams.put("death",Long.valueOf(death+deathConsume).toString());
            battleParams.put("injured",Long.valueOf(injured +injuredConsume).toString());
            battleParams.put("duringTime",Long.valueOf(duringTime).toString());

            scenarioInfo.getAmmunition().setCurrent(Double.valueOf(ammunition - ammunitionConsume));
            scenarioInfo.getFood().setCurrent(Double.valueOf(food - foodConsume));
            scenarioInfo.getFuel().setCurrent(Double.valueOf(fuel - fuelConsume));
            scenarioInfo.getMedical().setCurrent(Double.valueOf(medical - medicalConsume));
            scenarioInfo.getWater().setCurrent(Double.valueOf(water - waterConsume));
            scenarioInfo.getPerson().setDeath(Long.valueOf(death+deathConsume).intValue());
            scenarioInfo.getPerson().setInjured(Long.valueOf(injured +injuredConsume).intValue());
            redisUtil.hset(scenarioTask.getScenarioId() + "-" + roomId + "-" + scenarioTask.getResourceId(),"duringTime",String.valueOf(duringTime));
            redisUtil.hset(scenarioTask.getScenarioId() + "-" + roomId + "-" + scenarioTask.getResourceId(), "scenarioInfo",JSONObject.toJSONString(scenarioInfo));
          }catch (Exception ex){
            log.error("==============================设置消耗信息失败=============================================",ex.getMessage());
          }
          try {
            //推送消耗數據
            ResponseCmdInfo<String> sendConsumeMsg = new ResponseCmdInfo<>();
            jsonObject.put("deathConsume", deathConsume);
            jsonObject.put("injuredConsume", injuredConsume);
            jsonObject.put("ammunitionConsume", ammunitionConsume);
            jsonObject.put("foodConsume", foodConsume);
            jsonObject.put("waterConsume", waterConsume);
            jsonObject.put("fuelConsume", fuelConsume);
            jsonObject.put("medicalConsume", medicalConsume);
            jsonObject.put("teamLat",teamLat);
            jsonObject.put("teamLng",teamLng);
            jsonObject.put("resourceId",scenarioTask.getResourceId());
            LocalDateTime currentDateTime = new Date().toInstant().atZone(ZoneId.systemDefault()).toLocalDateTime();
            jsonObject.put("consumeDate",currentDateTime);
            sendConsumeMsg.setData(jsonObject.toString());
            sendConsumeMsg.setRoom(roomId);
            sendConsumeMsg.setScenarioId(scenarioTask.getScenarioId());
            sendConsumeMsg.setCmdType("battleConsume");
            Global.sendCmdInfoQueue.add(sendConsumeMsg);
            BattleConsume battleConsume = new BattleConsume();
            battleConsume.setLat(teamLat);
            battleConsume.setLng(teamLng);
            battleConsume.setId(IdUtils.simpleUUID());
            battleConsume.setAmmunition(ammunitionConsume);
            battleConsume.setDeath(Integer.valueOf(Double.valueOf(deathConsume).intValue()));
            battleConsume.setInjured(Integer.valueOf(Double.valueOf(injuredConsume).intValue()));
            battleConsume.setFood(foodConsume);
            battleConsume.setFuel(fuelConsume);
            battleConsume.setMedical(medicalConsume);
            battleConsume.setWater(waterConsume);
            battleConsume.setResourceId(scenarioTask.getResourceId());
            battleConsume.setConsumeDate(currentDateTime);
            battleConsumeService.save(battleConsume);
          }catch (Exception ex){
            ex.printStackTrace();
            log.error("==================推送消耗數據 失败============================================",ex.getMessage());
          }

          try {
            LocalDateTime currentDateTime = new Date().toInstant().atZone(ZoneId.systemDefault()).toLocalDateTime();
            String death = SpringUtil.getBean(Environment.class).getProperty("death.warn");
            String injured = SpringUtil.getBean(Environment.class).getProperty("injured.warn");
            String ammunition = SpringUtil.getBean(Environment.class).getProperty("ammunition.warn");
            log.info("===========person ammunition==={}===={}====={}========",death,injured,ammunition);
            HashMap<String,Object> battleConsumeMap = (HashMap<String, Object>) redisUtil.hget(scenarioTask.getScenarioId() + "-" + roomId + "-" + scenarioTask.getResourceId());
            Double restAmmunition = Double.valueOf(battleConsumeMap.get("ammunition").toString());
            StatisticBean battleResourceStat = statisticService.statistic(scenarioTask.getResourceId());
            Double ammunitionConsumeRate = restAmmunition/battleResourceStat.getAmmunition().getTotal()*100;
            if(Double.valueOf(ammunitionConsumeRate) <= Double.valueOf(ammunition) && suppleFlagMap.get("ammunition") == false){
              SupplierRequest supplierRequest = new SupplierRequest();
              supplierRequest.setId(IdUtils.simpleUUID());
              supplierRequest.setFromResourceId(scenarioTask.getResourceId());
              supplierRequest.setSupplierNum(String.valueOf(suppleAmount));
              supplierRequest.setSupplierType("ammunition");
              supplierRequest.setGeneralTime(currentDateTime);
              supplierRequest.setLat(jsonObject.get("teamLat").toString());
              supplierRequest.setLng(jsonObject.get("teamLng").toString());
              supplierRequestService.save(supplierRequest);
              ResponseCmdInfo<String> respObj = new ResponseCmdInfo<>();
              respObj.setData(JSON.toJSONString(supplierRequest));
              respObj.setRoom(roomId);
              respObj.setScenarioId(scenarioTask.getScenarioId());
              respObj.setCmdType("");
              Global.sendCmdInfoQueue.add(respObj);
              suppleFlagMap.put("ammunition",true) ;
            }
            Long restDeath = Long.valueOf(battleConsumeMap.get("death").toString());
            Long deathConsumeRate = 0L;
            if(battleResourceStat.getPerson().getTotal() !=0) {
              deathConsumeRate = restDeath * 100 / battleResourceStat.getPerson().getTotal();
            }
            if(deathConsumeRate >= Long.valueOf(death) && suppleFlagMap.get("death") == false){
              SupplierRequest supplierRequest = new SupplierRequest();
              supplierRequest.setId(IdUtils.simpleUUID());
              supplierRequest.setFromResourceId(scenarioTask.getResourceId());
              supplierRequest.setSupplierNum(String.valueOf(suppleDeath));
              supplierRequest.setSupplierType("death");
              supplierRequest.setGeneralTime(currentDateTime);
              supplierRequest.setLat(jsonObject.get("teamLat").toString());
              supplierRequest.setLng(jsonObject.get("teamLng").toString());
              supplierRequestService.save(supplierRequest);
              ResponseCmdInfo<String> respObj = new ResponseCmdInfo<>();
              respObj.setData(JSON.toJSONString(supplierRequest));
              respObj.setRoom(roomId);
              respObj.setScenarioId(scenarioTask.getScenarioId());
              Global.sendCmdInfoQueue.add(respObj);
              suppleFlagMap.put("death",true) ;
            }
            Long restInjured = Long.valueOf(battleConsumeMap.get("injured").toString());
            Long injuredConsumeRate = restInjured*100/battleResourceStat.getPerson().getTotal();
            if(Long.valueOf(injuredConsumeRate) <= Long.valueOf(injured) && suppleFlagMap.get("injured") == false){
              SupplierRequest supplierRequest = new SupplierRequest();
              supplierRequest.setId(IdUtils.simpleUUID());
              supplierRequest.setFromResourceId(scenarioTask.getResourceId());
              supplierRequest.setSupplierNum(String.valueOf(suppleInjured));
              supplierRequest.setSupplierType("injured");
              supplierRequest.setGeneralTime(currentDateTime);
              supplierRequest.setLat(jsonObject.get("teamLat").toString());
              supplierRequest.setLng(jsonObject.get("teamLng").toString());
              supplierRequestService.save(supplierRequest);
              ResponseCmdInfo<String> respObj = new ResponseCmdInfo<>();
              respObj.setData(JSON.toJSONString(supplierRequest));
              respObj.setRoom(roomId);
              respObj.setScenarioId(scenarioTask.getScenarioId());
              Global.sendCmdInfoQueue.add(respObj);
              suppleFlagMap.put("injured",true) ;
            }
          }catch (Exception ex){
            ex.printStackTrace();
            log.error("===========BattleRootTask supplierRequestService.saveBatch error====================",ex.getMessage());
          }
          log.info("===============================定时检查统计各种资源消耗量 end====================================");
        }
      });

    }

  }




}