巡查项接口新增调整

zhongrj

2023-04-03 ed0a71a11aeffcbb441d5a67e2a5663bd0f0e8ed

 skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/patrol/service/impl/PatrolGroupServiceImpl.java

@@ -1,6 +1,8 @@
package cn.gistack.sm.patrol.service.impl;


import cn.gistack.common.node.TreeNode;
import cn.gistack.common.utils.NodeTreeUtil;
import cn.gistack.sm.patrol.entity.PatrolGroup;
import cn.gistack.sm.patrol.mapper.PatrolGroupMapper;
import cn.gistack.sm.patrol.service.IPatrolGroupService;
@@ -11,6 +13,7 @@
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;

import java.util.List;
import java.util.Map;

/**
 * @Description: 巡查指标组
@@ -27,4 +30,18 @@
   public List<PatrolGroup> getPatrolGroupByProjectId(String projectId) {
      return baseMapper.getPatrolGroupByProjectId(projectId);
   }

   /**
    * 查询巡查项树数据
    * @return
    */
   @Override
   public Object getPatrolGroupTree() {
      // 查询父树目录
      List<TreeNode> patrolGroupTree = baseMapper.getPatrolGroupTree();
      // 查询孩子树目录
      List<TreeNode> patrolGroupItemTree = baseMapper.getPatrolGroupItemTree();
      // 数据处理
      return NodeTreeUtil.getNodeTree(patrolGroupTree,patrolGroupItemTree);
   }
}