neim/neim-biz/src/main/java/com/jiayue/biz/service/impl/FengJiInfoServiceImpl.java

package com.jiayue.biz.service.impl;

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.jiayue.biz.domain.*;
import com.jiayue.biz.dto.FanTowerDto;
import com.jiayue.biz.mapper.WindTowerDataParentTableMapper;
import com.jiayue.biz.service.*;
import lombok.AllArgsConstructor;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.stereotype.Service;

import java.util.*;
import java.util.stream.Collectors;

/**
 * 风机信息Service业务层处理
 *
 * @author L.ym
 * @date 2022-05-11
 */
@Service
@AllArgsConstructor
public class FengJiInfoServiceImpl extends ServiceImpl<WindTowerDataParentTableMapper, WindTowerDataParentTable> implements FengJiInfoService {

    private final ProjectInfoService projectInfoService;

    private final StationInfoService stationInfoService;

    //查询项目风机
    public List<FanTowerDto> getFanTowerForProject(String id) {
        List<ProjectInfo> projectInfoList = new ArrayList<>();
        if (id.equals("")) {
            projectInfoList = projectInfoService.getProjectInfoList();
        } else {
            projectInfoList.add(projectInfoService.getOneProjectInfoById(id));
        }
        ArrayList<FanTowerDto> list = new ArrayList<>();
        for (ProjectInfo projectInfo : projectInfoList) {
            if (projectInfo.getFanTowerList() != null) {
                for (FanTower fanTower : projectInfo.getFanTowerList()) {
                    FanTowerDto fanTowerDto = new FanTowerDto();
                    fanTowerDto.setId(projectInfo.getId());
                    if (projectInfo.getProjectBasicInfo() != null) {
                        fanTowerDto.setName(projectInfo.getProjectBasicInfo().getProjectNameEasy());
                    }
                    this.getFanTowerDto(fanTowerDto, fanTower);

                    list.add(fanTowerDto);
                }
            } else {
                FanTowerDto fanTowerDto = new FanTowerDto();
                fanTowerDto.setId(projectInfo.getId());
                if (projectInfo.getProjectBasicInfo() != null) {
                    fanTowerDto.setName(projectInfo.getProjectBasicInfo().getProjectNameEasy());
                }
                list.add(fanTowerDto);
            }


        }
        return list;

    }

    //查询场站风机
    public List<FanTowerDto> getFanTowerForStation(String id) {
        List<StationInfo> stationInfos = new ArrayList<>();
        if (id.equals("")) {
            stationInfos = stationInfoService.selectStationInfo();
        } else {
            stationInfos.add(stationInfoService.getOneStationInfo(id));
        }

        ArrayList<FanTowerDto> list = new ArrayList<>();
        for (StationInfo stationInfo : stationInfos) {
            if (stationInfo.getFanTowerList() != null) {
                for (FanTower fanTower : stationInfo.getFanTowerList()) {
                    FanTowerDto fanTowerDto = new FanTowerDto();
                    fanTowerDto.setId(stationInfo.getId());
                    if (stationInfo.getStationBasicInfo() != null) {
                        fanTowerDto.setName(stationInfo.getStationBasicInfo().getStationName());
                    }
                    this.getFanTowerDto(fanTowerDto, fanTower);
                    list.add(fanTowerDto);

                }
            } else {
                if (stationInfo.getStationBasicInfo() != null) {
                    FanTowerDto fanTowerDto = new FanTowerDto();
                    fanTowerDto.setId(stationInfo.getId());
                    fanTowerDto.setName(stationInfo.getStationBasicInfo().getStationName());
                    list.add(fanTowerDto);
                }
            }
        }
        return list;

    }

    //项目风机更新
    public void updateFanTowerForProjectInfo(FanTowerDto fanTowerDto) {
        ProjectInfo projectInfoById = projectInfoService.getOneProjectInfoById(fanTowerDto.getId());
        List<FanTower> fanTowerList = projectInfoById.getFanTowerList().stream().peek(s -> {
            if (s.getId().equals(fanTowerDto.getFanId())) {
                this.getFanTower(fanTowerDto, s);
            }
        }).collect(Collectors.toList());
        projectInfoById.setFanTowerList(fanTowerList);
        projectInfoService.saveProjectInfo(projectInfoById);
    }


    //场站风机更新
    public void updateFanTowerForStationInfo(FanTowerDto fanTowerDto) {
        StationInfo oneStationInfo = stationInfoService.getOneStationInfo(fanTowerDto.getId());
        List<FanTower> fanTowerList = oneStationInfo.getFanTowerList().stream().peek(s -> {
            if (s.getId().equals(fanTowerDto.getFanId())) {
                this.getFanTower(fanTowerDto, s);
            }
        }).collect(Collectors.toList());

        oneStationInfo.setFanTowerList(fanTowerList);
        stationInfoService.saveStation(oneStationInfo);
    }

    //项目新增风机信息
    public void saveFanTowerForProjectInfo(FanTowerDto fanTowerDto) {
        ProjectInfo projectInfoById = projectInfoService.getOneProjectInfoById(fanTowerDto.getId());
        if (projectInfoById.getFanTowerList() == null) {
            projectInfoById.setFanTowerList(new ArrayList<>());
        }
        FanTower fanTower = new FanTower();
        this.getFanTower(fanTowerDto, fanTower);
        projectInfoById.getFanTowerList().add(fanTower);
        projectInfoService.saveProjectInfo(projectInfoById);


    }

    //场站新增风机信息
    public void saveFanTowerForStationInfo(FanTowerDto fanTowerDto) {
        StationInfo oneStationInfo = stationInfoService.getOneStationInfo(fanTowerDto.getId());
        if (oneStationInfo.getFanTowerList() == null) {
            oneStationInfo.setFanTowerList(new ArrayList<>());
        }
        FanTower fanTower = new FanTower();
        this.getFanTower(fanTowerDto, fanTower);
        oneStationInfo.getFanTowerList().add(fanTower);
        stationInfoService.saveStation(oneStationInfo);


    }

    //项目删除风机信息
    public void deleteFanTowerForProjectInfo(FanTowerDto fanTowerDto) {
        ProjectInfo projectInfoById = projectInfoService.getOneProjectInfoById(fanTowerDto.getId());
        List<FanTower> fanTowerList = projectInfoById.getFanTowerList().stream().filter(s -> !s.getId().equals(fanTowerDto.getFanId())).collect(Collectors.toList());
        projectInfoById.setFanTowerList(fanTowerList);
        projectInfoService.saveProjectInfo(projectInfoById);
    }


    //场站删除风机信息
    public void deleteFanTowerForStationInfo(FanTowerDto fanTowerDto) {
        StationInfo oneStationInfo = stationInfoService.getOneStationInfo(fanTowerDto.getId());
        List<FanTower> fanTowerList = oneStationInfo.getFanTowerList().stream().filter(s -> !s.getId().equals(fanTowerDto.getFanId())).collect(Collectors.toList());
        oneStationInfo.setFanTowerList(fanTowerList);
        stationInfoService.saveStation(oneStationInfo);
    }


    //组装风机信息
    public void getFanTowerDto(FanTowerDto fanTowerDto, FanTower fanTower) {
        fanTowerDto.setFanId(fanTower.getId());
        fanTowerDto.setFanModel(fanTower.getFanModel());
        fanTowerDto.setFanName(fanTower.getFanName());
        fanTowerDto.setLatitudeFan(fanTower.getLatitudeFan());
        fanTowerDto.setLongitudeFan(fanTower.getLongitudeFan());
        fanTowerDto.setHubHeight(fanTower.getHubHeight());
    }

    //重组组风机信息
    public void getFanTower(FanTowerDto fanTowerDto, FanTower fanTower) {
        fanTower.setId(fanTowerDto.getFanId() == null ? UUID.randomUUID().toString() : fanTowerDto.getFanId());
        fanTower.setFanModel(fanTowerDto.getFanModel());
        fanTower.setFanName(fanTowerDto.getFanName());
        fanTower.setLatitudeFan(fanTowerDto.getLatitudeFan());
        fanTower.setLongitudeFan(fanTowerDto.getLongitudeFan());
        fanTower.setHubHeight(fanTowerDto.getHubHeight());
    }


}