From db0b070e51db46e779b8d2ad1d4d213b712602cd Mon Sep 17 00:00:00 2001
From: zengh <123456>
Date: Mon, 11 Jan 2021 17:04:18 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master' into master

---
 src/main/java/org/springblade/modules/equipment/controller/EquipmentController.java |   81 +++++++++++++++++++++++++++-------------
 1 files changed, 54 insertions(+), 27 deletions(-)

diff --git a/src/main/java/org/springblade/modules/equipment/controller/EquipmentController.java b/src/main/java/org/springblade/modules/equipment/controller/EquipmentController.java
index 5c17f5a..9ac58ec 100644
--- a/src/main/java/org/springblade/modules/equipment/controller/EquipmentController.java
+++ b/src/main/java/org/springblade/modules/equipment/controller/EquipmentController.java
@@ -19,8 +19,10 @@
 import io.swagger.annotations.*;
 import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
 import lombok.AllArgsConstructor;
+
 import javax.servlet.http.HttpServletResponse;
 import javax.validation.Valid;
+
 import org.springblade.core.mp.support.Condition;
 import org.springblade.core.mp.support.Query;
 import org.springblade.core.tool.api.R;
@@ -44,7 +46,7 @@
 import java.util.Map;
 
 /**
- *  控制器
+ * 控制器
  *
  * @author BladeX
  * @since 2020-07-01
@@ -65,10 +67,10 @@
 	@GetMapping("/detail")
 	@ApiOperationSupport(order = 1)
 	@ApiOperation(value = "详情", notes = "传入equipment")
-	public R<EquipmentVOS> detail(Equipment equipment,HttpServletResponse response) {
+	public R<EquipmentVOS> detail(Equipment 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");
+		response.setHeader("Access-Control-Allow-Credentials", "true");
 		/*Equipment detail = equipmentService.getOne(Condition.getQueryWrapper(equipment));*/
 
 		return R.data(equipmentService.selectInfo(equipment));
@@ -91,32 +93,56 @@
 	@GetMapping("/page")
 	@ApiOperationSupport(order = 3)
 	@ApiOperation(value = "分页", notes = "传入equipment")
-	public R<IPage<EquipmentVO>> page(EquipmentVO equipment, Query query,String pid, HttpServletResponse response) {
+	public R<List<Equipment>> page(EquipmentVO equipment, Query query, String pid, 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");
-		IPage<EquipmentVO> pages;
-		if (pid==null){
-			 pages = equipmentService.selectEquipmentPage(Condition.getPage(query), equipment);
-		}
-		else {
-			String s = catalogService.selectCatalogEqNUmber(pid);
-			if(s==null){
-				pages = equipmentService.selectEquipmentPage(Condition.getPage(query), equipment);
-			}else{
-				String[] split = s.split(",");
-				String strArrays="";
-				for(int i=0;i<split.length;i++){
-					strArrays+="'"+split[i]+"',";
+		response.setHeader("Access-Control-Allow-Credentials", "true");
+
+		//清空全局参数
+		childMenu = new ArrayList();
+
+		//根据查询条件得到所有设备
+		List<Equipment> pages = equipmentService.selectEquipmentPage(equipment.getDeviceName(), equipment.getAddvcd(), equipment.getDtype(), equipment.getDevicestate());
+
+		List<Equipment> datas = new ArrayList<>();
+
+		if (pid == null) {
+			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++) {
+					if (childList.get(i).get("eqid") != null) {
+						if (childList.get(i).get("eqid").equals(pages.get(j).getDeviceNumber())) {
+							datas.add(pages.get(j));
+						}
+					}
 				}
-				String substring = strArrays.substring(0,strArrays.length()-1);
-				equipment.setDeviceNumber(substring);
-				pages = equipmentService.selectEquipmentPage(Condition.getPage(query), equipment);
+			}
+		}
+		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;
 	}
 
 	/**
@@ -135,8 +161,8 @@
 		equipmentService.s(equipment);
 		String deviceNumber = equipment.getDeviceNumber();
 		String deviceName = equipment.getDeviceName();
-		String bType="0";
-		iDeployService.insterb(deviceNumber,deviceName,bType);
+		String bType = "0";
+		iDeployService.insterb(deviceNumber, deviceName, bType);
 		return R.success("成功");
 	}
 
@@ -204,14 +230,15 @@
 
 	/**
 	 * 设备列表
-	 * @param deviceType  设备类型
+	 *
+	 * @param deviceType 设备类型
 	 * @return
 	 */
 	@GetMapping("/selectList")
 	public R<List<EquipmentVOS>> selectList(String deviceType, 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");
+		response.setHeader("Access-Control-Allow-Credentials", "true");
 		List<EquipmentVOS> list = equipmentService.selectList(deviceType);
 		return R.data(list);
 	}

--
Gitblit v1.9.3