|
|
|
@ -15,6 +15,7 @@ import com.hnac.hzims.operational.station.vo.AreaTypeVo;
|
|
|
|
|
import com.hnac.hzims.operational.station.vo.DeptTreeVo; |
|
|
|
|
import com.hnac.hzims.operational.station.vo.WaterDeptTreeVo; |
|
|
|
|
import lombok.AllArgsConstructor; |
|
|
|
|
import org.springblade.core.secure.utils.AuthUtil; |
|
|
|
|
import org.springblade.core.tool.api.R; |
|
|
|
|
import org.springblade.core.tool.utils.BeanUtil; |
|
|
|
|
import org.springblade.core.tool.utils.CollectionUtil; |
|
|
|
@ -199,7 +200,7 @@ public class ImsDeptServiceImpl extends ServiceImpl<ImsDeptMapper, ImsDeptEntity
|
|
|
|
|
@Override |
|
|
|
|
public List<DeptTreeVo> getHydropowerTree(Integer stationType, Integer serveType) { |
|
|
|
|
// 获取所有机构
|
|
|
|
|
R<List<Dept>> D = sysClient.getDeptByType("200000", null, null); |
|
|
|
|
R<List<Dept>> D = sysClient.getDeptByType(AuthUtil.getTenantId(), null, null); |
|
|
|
|
if (!D.isSuccess() || CollectionUtil.isEmpty(D.getData())) { |
|
|
|
|
return null; |
|
|
|
|
} |
|
|
|
@ -210,10 +211,8 @@ public class ImsDeptServiceImpl extends ServiceImpl<ImsDeptMapper, ImsDeptEntity
|
|
|
|
|
} |
|
|
|
|
// 所有站点
|
|
|
|
|
List<Long> value = D.getData().stream().filter(o -> o.getDeptCategory().equals(4)).map(Dept::getId).collect(Collectors.toList()); |
|
|
|
|
List<Integer> type = ObjectUtil.isNotEmpty(stationType) ? Collections.singletonList(stationType) : null; |
|
|
|
|
if (HomePageConstant.WATER_CONSERVANCY.equals(stationType)) { |
|
|
|
|
type = HomePageConstant.waterList; |
|
|
|
|
} |
|
|
|
|
List<Integer> type = ObjectUtil.isNotEmpty(stationType) ? Collections.singletonList(stationType) : HomePageConstant.SERVETYPE_LIST; |
|
|
|
|
|
|
|
|
|
List<StationEntity> stationEntityList = stationService.getStationType(serveType, type, value, true); |
|
|
|
|
if (CollectionUtil.isEmpty(stationEntityList)) { |
|
|
|
|
return null; |
|
|
|
|