package com.ruoyi.equipmentenergyconsumption.service.impl;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.ruoyi.basic.dto.ProductTreeDto;
|
import com.ruoyi.basic.pojo.Product;
|
import com.ruoyi.common.utils.bean.BeanUtils;
|
import com.ruoyi.equipmentenergyconsumption.dto.ElectricityConsumptionAreaTreeDto;
|
import com.ruoyi.equipmentenergyconsumption.mapper.ElectricityConsumptionAreaMapper;
|
import com.ruoyi.equipmentenergyconsumption.pojo.ElectricityConsumptionArea;
|
import com.ruoyi.equipmentenergyconsumption.service.ElectricityConsumptionAreaService;
|
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
|
|
@Service
|
@Slf4j
|
public class ElectricityConsumptionAreaServiceImpl extends ServiceImpl<ElectricityConsumptionAreaMapper, ElectricityConsumptionArea> implements ElectricityConsumptionAreaService {
|
|
@Autowired
|
private ElectricityConsumptionAreaMapper electricityConsumptionAreaMapper;
|
|
@Override
|
public IPage<ElectricityConsumptionArea> listPage(Page page, ElectricityConsumptionArea electricityConsumptionArea) {
|
return electricityConsumptionAreaMapper.listPage(page,electricityConsumptionArea);
|
}
|
|
|
//左侧树结构
|
@Override
|
public List<ElectricityConsumptionAreaTreeDto> selectElectricityConsumptionAreaList() {
|
// 查询根节点（parentId 为 null）
|
LambdaQueryWrapper<ElectricityConsumptionArea> queryWrapper = new LambdaQueryWrapper<>();
|
queryWrapper.isNull(ElectricityConsumptionArea::getFuId);
|
// 查询根节点列表
|
List<ElectricityConsumptionArea> rootProducts = electricityConsumptionAreaMapper.selectList(queryWrapper);
|
|
// 转换为树节点并递归构建子树
|
List<ElectricityConsumptionAreaTreeDto> tree = new ArrayList<>();
|
for (ElectricityConsumptionArea electricityConsumptionArea : rootProducts) {
|
ElectricityConsumptionAreaTreeDto node = convertToTreeDto(electricityConsumptionArea);
|
node.setChildren(buildChildrenNodes(electricityConsumptionArea.getId()));
|
tree.add(node);
|
}
|
return tree;
|
}
|
|
// 将 Product 转换为 ProductTreeDto
|
private ElectricityConsumptionAreaTreeDto convertToTreeDto(ElectricityConsumptionArea electricityConsumptionArea) {
|
ElectricityConsumptionAreaTreeDto dto = new ElectricityConsumptionAreaTreeDto();
|
BeanUtils.copyProperties(electricityConsumptionArea, dto);
|
dto.setLabel(electricityConsumptionArea.getAreaName()); // 设置 label 为产品名称
|
dto.setChildren(new ArrayList<>());
|
return dto;
|
}
|
|
// 递归构建子节点
|
private List<ElectricityConsumptionAreaTreeDto> buildChildrenNodes(Long parentId) {
|
// 查询当前父节点的子节点
|
LambdaQueryWrapper<ElectricityConsumptionArea> queryWrapper = new LambdaQueryWrapper<>();
|
queryWrapper.eq(ElectricityConsumptionArea::getFuId, parentId);
|
List<ElectricityConsumptionArea> childProducts = electricityConsumptionAreaMapper.selectList(queryWrapper);
|
|
// 转换子节点并递归构建它们的子树
|
List<ElectricityConsumptionAreaTreeDto> children = new ArrayList<>();
|
for (ElectricityConsumptionArea child : childProducts) {
|
ElectricityConsumptionAreaTreeDto childNode = convertToTreeDto(child);
|
childNode.setChildren(buildChildrenNodes(child.getId()));
|
children.add(childNode);
|
}
|
return children;
|
}
|
}
|
