From fec50a92029593e89cea21e8b8edfd8d4f479c23 Mon Sep 17 00:00:00 2001
From: tanghaihao <850814414@qq.com>
Date: Mon, 14 Aug 2023 15:25:59 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AF=BC=E5=85=A5=E8=AE=BE=E5=A4=87=E4=BF=A1?=
 =?UTF-8?q?=E6=81=AF=E7=9A=84=E6=97=B6=E5=80=99=E4=BB=8E=E5=AE=9A=E4=B9=89?=
 =?UTF-8?q?=E4=B8=AD=E8=8E=B7=E5=8F=96=E6=A8=A1=E5=9E=8B=E6=A0=87=E8=AF=86?=
 =?UTF-8?q?=E5=A1=AB=E5=85=85?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../hzims/equipment/service/impl/EmInfoServiceImpl.java    | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/hzims-service/equipment/src/main/java/com/hnac/hzims/equipment/service/impl/EmInfoServiceImpl.java b/hzims-service/equipment/src/main/java/com/hnac/hzims/equipment/service/impl/EmInfoServiceImpl.java
index 25cd987..c3ece74 100644
--- a/hzims-service/equipment/src/main/java/com/hnac/hzims/equipment/service/impl/EmInfoServiceImpl.java
+++ b/hzims-service/equipment/src/main/java/com/hnac/hzims/equipment/service/impl/EmInfoServiceImpl.java
@@ -91,6 +91,8 @@ public class EmInfoServiceImpl extends BaseServiceImpl<EmInfoMapper, EmInfoEntit
 	@Autowired
 	private IEmTreeInspectService emTreeInspectService;
 	@Autowired
+	private IEmDefinitionService emDefinitionService;
+	@Autowired
 	private IStationClient stationClient;
 	@Autowired
 	private IAnalyseDataSearchClient searchClient;
@@ -493,7 +495,8 @@ public class EmInfoServiceImpl extends BaseServiceImpl<EmInfoMapper, EmInfoEntit
 		if (Func.isEmpty(station)) {
 			throw new ServiceException("站点不存在");
 		}
-		copyDeviceTree(station, 0L,deviceTreeDTO.getDefinitionId() , deviceTreeDTO.getTreeIds());
+		EmDefinitionEntity emDefinition = emDefinitionService.getById(deviceTreeDTO.getDefinitionId());
+		copyDeviceTree(station, 0L, deviceTreeDTO.getDefinitionId(), emDefinition.getModelSignage(), deviceTreeDTO.getTreeIds());
 		return true;
 	}
 
@@ -508,8 +511,8 @@ public class EmInfoServiceImpl extends BaseServiceImpl<EmInfoMapper, EmInfoEntit
 			throw new ServiceException("设备负责部门未绑定站点");
 		}
 		EmTreeEntity emTree = emTreeService.getById(emInfo.getTreeId());
-
-		copyDeviceTree(station, deviceTreeDTO.getInfoId(), emTree.getDefinitionId(), deviceTreeDTO.getTreeIds());
+		EmDefinitionEntity emDefinition = emDefinitionService.getById(deviceTreeDTO.getDefinitionId());
+		copyDeviceTree(station, deviceTreeDTO.getInfoId(), emTree.getDefinitionId(), emDefinition.getModelSignage(), deviceTreeDTO.getTreeIds());
 		return true;
 	}
 
@@ -781,7 +784,7 @@ public class EmInfoServiceImpl extends BaseServiceImpl<EmInfoMapper, EmInfoEntit
 		return resultList;
 	}
 
-	private void copyDeviceTree(StationEntity station, Long pid, Long definitionId, List<EmTreeIdsDTO> treeIds) {
+	private void copyDeviceTree(StationEntity station, Long pid, Long definitionId, String modelSignage, List<EmTreeIdsDTO> treeIds) {
 		List<ProjectDTO> projectDTOList = new ArrayList<>();
 		LambdaQueryWrapper<EmTreeEntity> wrapper = new LambdaQueryWrapper();
 		wrapper.eq(EmTreeEntity::getDefinitionId, definitionId);
@@ -814,6 +817,7 @@ public class EmInfoServiceImpl extends BaseServiceImpl<EmInfoMapper, EmInfoEntit
 			emInfo.setRunStatus("1");
 			emInfo.setStatus(1);
 			emInfo.setHomePageDisplay("1");
+			emInfo.setModelSignage(modelSignage);
 			emInfo.setRunDate(new Date());
 			emInfo.setCreateDept(station.getRefDept());
 			emInfo.setCreateTime(new Date());
@@ -883,7 +887,7 @@ public class EmInfoServiceImpl extends BaseServiceImpl<EmInfoMapper, EmInfoEntit
 			projectDTOList.add(projectDTO);
 
 			if (Func.isNotEmpty(emTreeIdsDTO.getChildren())) {
-				copyDeviceTree(station, infoId, definitionId, emTreeIdsDTO.getChildren());
+				copyDeviceTree(station, infoId, definitionId, modelSignage, emTreeIdsDTO.getChildren());
 			}
 		}
 		if (Func.isNotEmpty(projectDTOList)) {