|
@@ -6,17 +6,20 @@ import com.huaxu.common.Reflections;
|
|
|
import com.huaxu.common.StringUtils;
|
|
|
import com.huaxu.dao.SceneImageMapper;
|
|
|
import com.huaxu.dao.SceneMapper;
|
|
|
+import com.huaxu.dto.DeviceDto;
|
|
|
import com.huaxu.entity.SceneEntity;
|
|
|
import com.huaxu.entity.SceneImageEntity;
|
|
|
import com.huaxu.entity.SceneTypeEntity;
|
|
|
import com.huaxu.model.AjaxMessage;
|
|
|
import com.huaxu.model.LoginUser;
|
|
|
import com.huaxu.model.ResultStatus;
|
|
|
+import com.huaxu.util.OrgInfoUtil;
|
|
|
import com.huaxu.util.Snowflake;
|
|
|
import com.huaxu.util.UserUtil;
|
|
|
import javafx.scene.Scene;
|
|
|
import org.apache.ibatis.annotations.Param;
|
|
|
import org.springframework.beans.BeanUtils;
|
|
|
+import org.springframework.beans.factory.annotation.Autowired;
|
|
|
import org.springframework.stereotype.Service;
|
|
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
|
@@ -46,6 +49,8 @@ public class SceneService extends ServiceImpl<SceneMapper, SceneEntity> {
|
|
|
private SceneMapper sceneMapper;
|
|
|
@Resource
|
|
|
private SceneImageService sceneImageService;
|
|
|
+ @Autowired
|
|
|
+ private OrgInfoUtil orgInfoUtil;
|
|
|
|
|
|
/**
|
|
|
* 自定义分页查询,含关联实体对像
|
|
@@ -62,7 +67,11 @@ public class SceneService extends ServiceImpl<SceneMapper, SceneEntity> {
|
|
|
//查询子节点的信息
|
|
|
List<SceneEntity> sceneEntities = newArrayList();
|
|
|
for (SceneEntity item : sceneEntityPage.getRecords()) {
|
|
|
- sceneEntities.addAll(findByParentLike(item.getId()));
|
|
|
+ sceneEntities.addAll(findByParentLike(item.getId()));
|
|
|
+ }
|
|
|
+ for (SceneEntity item : sceneEntities) {
|
|
|
+ item.setCompanyOrgName(orgInfoUtil.getOrgName(item.getCompanyOrgId().intValue()));
|
|
|
+ item.setDeptOrgName(orgInfoUtil.getOrgName(item.getDeptOrgId().intValue()));
|
|
|
}
|
|
|
List<SceneEntity> sceneTree = getSceneTree(sceneEntities, Long.valueOf(0), 1);
|
|
|
sceneEntityPage.setRecords(sceneTree);
|