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

package com.jiayue.biz.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.jiayue.biz.domain.*;
import com.jiayue.biz.dto.ProjectDto;
import com.jiayue.biz.mapper.ProjectMapper;
import com.jiayue.biz.service.ProjectInfoService;
import com.jiayue.biz.service.ProjectService;
import com.jiayue.biz.service.WindTowerInfoService;
import com.jiayue.biz.util.CalculationUtil;
import lombok.AllArgsConstructor;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;
import java.util.UUID;
import java.util.stream.Collectors;

/**
 * 项目Service业务层处理
 *
 * @author L.ym
 * @date 2022-12-22
 */
@AllArgsConstructor
@Service
public class ProjectServiceImpl extends ServiceImpl<ProjectMapper, Project> implements ProjectService {

    private final ProjectInfoService projectInfoService;

    private final WindTowerInfoService windTowerInfoService;


    private final MongoTemplate mongoTemplate;

    /*
     * 根据项目名称模糊查询
     */
    @Override
    public Page getByProjectName(Integer current, Integer size, String projectName) {
        QueryWrapper<Project> wrapper = new QueryWrapper<>();
        wrapper.like(projectName != null, "project_name", projectName);
        Page page = new Page(current, size);
        return page(page, wrapper);
    }


    //查询组装项目列表
    public List<ProjectDto> getProject(Project project) {
        List<ProjectInfo> projectInfoList = projectInfoService.getProjectInfoList();
        ArrayList<ProjectDto> projectList = new ArrayList<>();
        for (ProjectInfo projectInfo : projectInfoList) {
            StringBuilder builder = new StringBuilder();
            ProjectDto projectDto = new ProjectDto();
            projectDto.setId(projectInfo.getId());
            if (projectInfo.getProjectBasicInfo() != null) {
                projectDto.setProjectSort(projectInfo.getProjectBasicInfo().getProjectSort());
                projectDto.setProjectState(projectInfo.getProjectBasicInfo().getProjectState());
                projectDto.setProjectType(projectInfo.getProjectBasicInfo().getProjectType());
                projectDto.setProjectName(projectInfo.getProjectBasicInfo().getProjectName());
                projectDto.setProjectOverview(projectInfo.getProjectBasicInfo().getProjectOverview());
                projectDto.setResourcesOverview(projectInfo.getProjectBasicInfo().getResourcesOverview());
                projectDto.setProjectNameEasy(projectInfo.getProjectBasicInfo().getProjectNameEasy());
                projectDto.setWd(projectInfo.getProjectBasicInfo().getWd());
                projectDto.setTAvailableHours(projectInfo.getProjectBasicInfo().getTAvailableHours());
                projectDto.setTPowerGeneration(projectInfo.getProjectBasicInfo().getTPowerGeneration());
                projectDto.setProjectNo(projectInfo.getProjectBasicInfo().getProjectNo());
                if (projectInfo.getProjectBasicInfo().getPosition() != null && !projectInfo.getProjectBasicInfo().getPosition().equals("")) {
                    String[] split = projectInfo.getProjectBasicInfo().getPosition().split(",");
                    projectDto.setLongitude(split[0]);
                    projectDto.setLatitude(split[1]);
                }
            }
            if (projectInfo.getEquipment() != null) {
                List<String> collect = projectInfo.getEquipment().stream().map(Equipment::getName).collect(Collectors.toList());
                if (projectInfo.getEquipment() != null && !projectInfo.getEquipment().equals("")) {
                    for (String s : collect) {
                        builder.append(s).append(",");
                    }
                    if (builder.length() > 0) {
                        builder.delete(builder.length() - 1, builder.length());
                    }
                    projectDto.setEquipment(builder.toString());
                }
            }

            StringBuilder stringBuilder = new StringBuilder();
            List<List<Coordinates>> coordinates = projectInfo.getCoordinates();
            if (coordinates != null && coordinates.size() > 0) {
                for (List<Coordinates> coordinate : coordinates) {
                    for (Coordinates coordinates1 : coordinate) {
                        stringBuilder.append(coordinates1.getLongitude()).append(",").append(coordinates1.getLatitude()).append(";");
                    }
                    if (stringBuilder.length() > 0) {
                        stringBuilder.delete(stringBuilder.length() - 1, stringBuilder.length());
                        stringBuilder.append("。");
                    }

                }
            }
            projectDto.setCoordinates(stringBuilder.toString());
            projectList.add(projectDto);
        }
        if (project.getProjectName() != null) {
            return projectList.stream().filter(p -> p.getProjectName().contains(project.getProjectName())).collect(Collectors.toList());
        }
        return projectList;
    }

    @Override
    public ProjectInfo getProjectById(String projectId) {
        return projectInfoService.getOneProjectInfoById(projectId);
    }


    public void delProject(String id) {
        projectInfoService.delProjectInfo(id);
    }


    //新增
    public void saveProject(ProjectDto projectDto) {
        List<WindTowerInfo> windTowerInfoList = windTowerInfoService.list();
        ProjectInfo projectInfo = new ProjectInfo();
        projectInfo.setId(UUID.randomUUID().toString());
        ProjectBasicInfo projectBasicInfo = projectInfo.getProjectBasicInfo();
        if (projectBasicInfo == null) {
            projectBasicInfo = new ProjectBasicInfo();
            this.getProjectBasicInfo(projectBasicInfo, projectDto);
        }
        projectInfo.setProjectBasicInfo(projectBasicInfo);

        if (projectDto.getEquipment() != null && !projectDto.getEquipment().equals("")) {
            List<Equipment> equipment = CalculationUtil.getEquipment(windTowerInfoList, projectDto.getEquipment());
            projectInfo.setEquipment(equipment);
        } else {
            projectInfo.setEquipment(new ArrayList<>());
        }

        ArrayList<List<Coordinates>> cooList = new ArrayList<>();
        if (projectDto.getCoordinates() != null) {
            if (projectDto.getCoordinates().contains("。")) {
                String[] split = projectDto.getCoordinates().split("。");
                for (String s : split) {
                    List<Coordinates> coordinatesList = this.getCoordinates(s);
                    cooList.add(coordinatesList);
                }
            } else {
                List<Coordinates> coordinatesList1 = this.getCoordinates(projectDto.getCoordinates());
                cooList.add(coordinatesList1);
            }
        } else {
            projectInfo.setCoordinates(new ArrayList<>());
        }

        mongoTemplate.save(projectInfo);


    }


    //修改
    public void updateProject(ProjectDto projectDto) {
        List<WindTowerInfo> windTowerInfoList = windTowerInfoService.list();
        List<List<Coordinates>> cooList = new ArrayList<>();
        ProjectInfo projectInfo = projectInfoService.getOneProjectInfoById(projectDto.getId());

        ProjectBasicInfo projectBasicInfo = projectInfo.getProjectBasicInfo();
        //组装基本信息
        this.getProjectBasicInfo(projectBasicInfo, projectDto);

        if (projectDto.getEquipment() != null && !projectDto.getEquipment().equals("")) {
            List<Equipment> equipment = CalculationUtil.getEquipment(windTowerInfoList, projectDto.getEquipment());
            projectInfo.setEquipment(equipment);
        } else {
            projectInfo.setEquipment(new ArrayList<>());
        }

        if (projectDto.getCoordinates().contains("。")) {
            String[] split = projectDto.getCoordinates().split("。");
            for (String s : split) {
                List<Coordinates> coordinatesList = this.getCoordinates(s);
                cooList.add(coordinatesList);
            }
        } else {
            if (!projectDto.getCoordinates().equals("")) {
                List<Coordinates> coordinatesList1 = this.getCoordinates(projectDto.getCoordinates());
                cooList.add(coordinatesList1);
            }
        }
        projectInfo.setCoordinates(cooList);
        mongoTemplate.save(projectInfo);
    }

    //修改
    public void updateProjectInfo(ProjectInfo projectInfo) {

        mongoTemplate.save(projectInfo);
    }

    public List<Coordinates> getCoordinates(String coordinates) {
        ArrayList<Coordinates> coordinatesList = new ArrayList<>();
        if (coordinates.length() > 0) {
            String[] splits = coordinates.split(";");
            for (String s : splits) {
                String[] split = s.split(",");
                Coordinates coordinates1 = new Coordinates();
                coordinates1.setLongitude(split[0]);
                coordinates1.setLatitude(split[1]);
                coordinatesList.add(coordinates1);
            }

        }
        return coordinatesList;
    }


    public void getProjectBasicInfo(ProjectBasicInfo projectBasicInfo, ProjectDto projectDto) {
        projectBasicInfo.setProjectName(projectDto.getProjectName());
        projectBasicInfo.setProjectSort(projectDto.getProjectSort());
        projectBasicInfo.setProjectState(projectDto.getProjectState());
        projectBasicInfo.setProjectType(projectDto.getProjectType());
        projectBasicInfo.setProjectNameEasy(projectDto.getProjectNameEasy());
        projectBasicInfo.setWd(projectDto.getWd());
        projectBasicInfo.setTAvailableHours(projectDto.getTAvailableHours());
        projectBasicInfo.setTPowerGeneration(projectDto.getTPowerGeneration());
        projectBasicInfo.setProjectNo(projectDto.getProjectNo());
        if (projectDto.getLongitude() != null && projectDto.getLatitude() != null) {
            if (projectDto.getLongitude().equals("") && projectDto.getLatitude().equals("")) {
                projectBasicInfo.setPosition("");
            } else {
                projectBasicInfo.setPosition(projectDto.getLongitude() + "," + projectDto.getLatitude());
            }
        } else {
            projectBasicInfo.setPosition("");
        }
        projectBasicInfo.setProjectOverview(projectDto.getProjectOverview());
        projectBasicInfo.setResourcesOverview(projectDto.getResourcesOverview());
    }

}