zengh
2021-01-15 58d92eba8c87fdc91abd928d1b5d3da5a9ef52e8
src/main/java/org/springblade/modules/equipment/controller/EquipmentController.java
@@ -40,7 +40,9 @@
import org.springblade.core.boot.ctrl.BladeController;
import springfox.documentation.annotations.ApiIgnore;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;
@@ -93,31 +95,89 @@
   @GetMapping("/page")
   @ApiOperationSupport(order = 3)
   @ApiOperation(value = "分页", notes = "传入equipment")
   public R<List<EquipmentVO>> page(EquipmentVO equipment, Query query, String pid, HttpServletResponse response) {
   public R<List<EquipmentVO>> page(EquipmentVO equipment, Query query, String pid, HttpServletResponse response) throws ParseException {
      response.setHeader("Access-Control-Allow-Origin", "*");
      response.setHeader("Access-Control-Allow-Methods", "POST, GET, OPTIONS, DELETE");
      response.setHeader("Access-Control-Allow-Credentials", "true");
      List<EquipmentVO> pages;
      if (pid == null) {
         pages = equipmentService.selectEquipmentPage(equipment.getDeviceName(), equipment.getAddvcd(), equipment.getDtype(), equipment.getDevicestate());
      } else {
         String s = catalogService.selectCatalogEqNUmber(pid);
         if (s == null) {
            pages = equipmentService.selectEquipmentPage(equipment.getDeviceName(), equipment.getAddvcd(), equipment.getDtype(), equipment.getDevicestate());
         } else {
            String[] split = s.split(",");
            String strArrays = "";
            for (int i = 0; i < split.length; i++) {
               strArrays += "'" + split[i] + "',";
      //清空全局参数
      childMenu = new ArrayList();
      //根据查询条件得到所有设备
      List<EquipmentVO> pages = equipmentService.selectEquipmentPage(equipment.getDeviceName(), equipment.getAddvcd(), equipment.getDtype(), equipment.getDevicestate());
      List<EquipmentVO> datas = new ArrayList<>();
      if (pid == null || pid.equals("")) {
         for (int j = 0; j < pages.size(); j++) {
            String expireTime = pages.get(j).getExpireTime();
            //1.获取当前时间
            Date dd = new Date();
            SimpleDateFormat sim = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
            String end = sim.format(dd);
            //把string类型转换为long类型的
            long st = sim.parse(expireTime).getTime();
            long en = sim.parse(end).getTime();
            //计算天数
            int day = (int) ((st - en) / 86400000);
            if (day <= 30) {
               pages.get(j).setType(1);
            } else {
               pages.get(j).setType(0);
            }
            String substring = strArrays.substring(0, strArrays.length() - 1);
            equipment.setDeviceNumber(substring);
            pages = equipmentService.selectEquipmentPage(equipment.getDeviceName(), equipment.getAddvcd(), equipment.getDtype(), equipment.getDevicestate());
         }
         datas = pages;
      } else {
         List<Map<String, Object>> treedatas = catalogService.selectCatalogEqNUmber(pid);
         //根据传入的id递归出该节点下所有的设备
         List<Map<String, Object>> childList = treeMenuList(treedatas, Integer.parseInt(pid));
         //将得到的设备与查询所得设备进行匹配
         for (int i = 0; i < childList.size(); i++) {
            for (int j = 0; j < pages.size(); j++) {
               String expireTime = pages.get(j).getExpireTime();
               //1.获取当前时间
               Date dd = new Date();
               SimpleDateFormat sim = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
               String end = sim.format(dd);
               //把string类型转换为long类型的
               long st = sim.parse(expireTime).getTime();
               long en = sim.parse(end).getTime();
               //计算天数
               int day = (int) ((st - en) / 86400000);
               if (day <= 30) {
                  pages.get(j).setType(1);
               } else {
                  pages.get(j).setType(0);
               }
               if (childList.get(i).get("eqid") != null) {
                  if (childList.get(i).get("eqid").equals(pages.get(j).getDeviceNumber())) {
                     datas.add(pages.get(j));
                  }
               }
            }
         }
      }
      return R.data(datas);
   }
   //子节点
   private static List childMenu = new ArrayList();
   public static List treeMenuList(List<Map<String, Object>> menuList, int pid) {
      for (int i = 0; i < menuList.size(); i++) {
         //遍历出父id等于参数的id,add进子节点集合
         if (pid != 0) {
            if (menuList.get(i).get("pId").equals(pid)) {
               //递归遍历下一级
               treeMenuList(menuList, Integer.valueOf((Integer) menuList.get(i).get("id")));
               childMenu.add(menuList.get(i));
            }
         }
      }
      return R.data(pages);
      return childMenu;
   }
   /**
@@ -147,7 +207,10 @@
   @PostMapping("/update")
   @ApiOperationSupport(order = 5)
   @ApiOperation(value = "修改", notes = "传入equipment")
   public R update(@Valid @RequestBody EquipmentVOS equipment) {
   public R update(@Valid @RequestBody EquipmentVOS equipment, HttpServletResponse response) {
      response.setHeader("Access-Control-Allow-Origin", "*");
      response.setHeader("Access-Control-Allow-Methods", "POST, GET, OPTIONS, DELETE");
      response.setHeader("Access-Control-Allow-Credentials", "true");
      Date day = new Date();
      SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
      equipment.setStime(df.format(day));
@@ -156,6 +219,20 @@
   }
   /**
    * 修改
    */
   @PostMapping("/updateClinent")
   @ApiOperationSupport(order = 5)
   @ApiOperation(value = "修改", notes = "传入equipment")
   public R updateClinent(EquipmentVO equipment, HttpServletResponse response) {
      response.setHeader("Access-Control-Allow-Origin", "*");
      response.setHeader("Access-Control-Allow-Methods", "POST, GET, OPTIONS, DELETE");
      response.setHeader("Access-Control-Allow-Credentials", "true");
      equipmentService.updateClinent(equipment);
      return R.success("修改成功");
   }
   /**
    * 新增或修改
    */
   @PostMapping("/submit")