PAS/src/main/java/cn/cslg/pas/service/business/AssoProjectPersonService.java

package cn.cslg.pas.service.business;

import cn.cslg.pas.common.dto.AddProjectPersonDTO;
import cn.cslg.pas.common.dto.QueryProjectPersonDTO;
import cn.cslg.pas.common.model.cronModel.Personnel;
import cn.cslg.pas.common.model.cronModel.PersonnelVO;
import cn.cslg.pas.common.model.cronModel.Records;
import cn.cslg.pas.common.utils.CacheUtils;
import cn.cslg.pas.common.utils.LoginUtils;
import cn.cslg.pas.common.vo.ProjectPersonVO;
import cn.cslg.pas.domain.business.AssoProductPerson;
import cn.cslg.pas.domain.business.AssoProjectPerson;
import cn.cslg.pas.exception.XiaoShiException;
import cn.cslg.pas.mapper.AssoProductPersonMapper;
import cn.cslg.pas.mapper.AssoProjectPersonMapper;
import cn.cslg.pas.service.permissions.PermissionService;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

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

/**
 * 项目与可见人员关联表Service层
 *
 * @Author lrj
 * @Date 2023/11/10
 */
@Service
public class AssoProjectPersonService extends ServiceImpl<AssoProjectPersonMapper, AssoProjectPerson> {
    @Autowired
    private PermissionService permissionService;
    @Autowired
    private CacheUtils cacheUtils;
    @Autowired
    private LoginUtils loginUtils;
    /**
     * 添加分享人
     *
     * @param addProjectPersonDTO
     * @return
     */
    public List<Integer> addProjectShare(AddProjectPersonDTO addProjectPersonDTO) {
        PersonnelVO personnelVO =cacheUtils.getLoginUser(loginUtils.getId());

        List<Integer> reIds = new ArrayList<>();
        List<String> personIds = addProjectPersonDTO.getPersonIds();
        Integer projectId = addProjectPersonDTO.getProjectId();
        //根据ids查询是否存在
        LambdaQueryWrapper<AssoProjectPerson> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.select(AssoProjectPerson::getPersonId);
        queryWrapper.in(AssoProjectPerson::getPersonId, personIds);
        List<AssoProjectPerson> assoProjectPersonList = this.list(queryWrapper);
        List<String> ids = assoProjectPersonList.stream().map(AssoProjectPerson::getPersonId).collect(Collectors.toList());
        if (ids != null) {
            personIds.removeAll(ids);
        }
        if (personIds.size() != 0) {
            for (String personId : personIds) {
                AssoProjectPerson assoProjectPerson = new AssoProjectPerson();
                assoProjectPerson.setProjectId(projectId);
                assoProjectPerson.setPersonId(personId);
                assoProjectPerson.setCreateId(personnelVO.getId());
                assoProjectPerson.insert();
                reIds.add(assoProjectPerson.getId());
            }
        }
        return reIds;
    }

    /**
     * 批量删除分享记录
     *
     * @param ids
     * @return
     */
    public List<Integer> removeProjectShare(List<Integer> ids) {
        if (ids == null || ids.size() == 0) {
            throw new XiaoShiException("请选择分享记录");
        }
        this.removeBatchByIds(ids);
        return ids;
    }

    /**
     * 查询分享记录
     *
     * @param queryProjectPersonDTO
     * @return
     */
    public Records queryProjectShare(QueryProjectPersonDTO queryProjectPersonDTO) {
        Records records = new Records();
        Integer projectId = queryProjectPersonDTO.getProjectId();
        String personName = queryProjectPersonDTO.getPersonName();
        Long current = queryProjectPersonDTO.getCurrent();
        Long size = queryProjectPersonDTO.getSize();
        LambdaQueryWrapper<AssoProjectPerson> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(AssoProjectPerson::getProjectId, projectId);
        List<String> ids = new ArrayList<>();
        if (personName != null && personName.trim() != "") {
            ids = permissionService.getPersonIdsByName(personName);
        }
        if (ids.size() != 0) {
            queryWrapper.in(AssoProjectPerson::getProjectId, ids);
        }
        Page<AssoProjectPerson> page = this.page(new Page<>(current, size), queryWrapper);

        List<AssoProjectPerson> assoProjectPersonList = page.getRecords();
        List<ProjectPersonVO> projectPersonVOS = this.loadData(assoProjectPersonList);
        records.setData(projectPersonVOS);
        records.setTotal(page.getTotal());
        records.setCurrent(current);
        records.setSize(size);
        return records;
    }

    private List<ProjectPersonVO> loadData(List<AssoProjectPerson> assoProjectPersonList) {

        List<ProjectPersonVO> projectPersonVOS = new ArrayList<>();
        if (assoProjectPersonList == null || assoProjectPersonList.size() == 0) {
            return projectPersonVOS;
        }
        List<String> personIds = new ArrayList<>();
        assoProjectPersonList.forEach(item -> {
            if (item.getPersonId() != null) {
                personIds.add(item.getPersonId());
            }
            if (item.getCreateId() != null) {
                personIds.add(item.getCreateId());
            }
        });
        List<Personnel> personnels = new ArrayList<>();
        if (personIds.size() != 0) {
            try {
                String res = permissionService.getPersonnelByIdsFromPCS(personIds);
                JSONObject jsonObject = JSONObject.parseObject(res);
                personnels = JSONObject.parseArray(jsonObject.getString("data"), Personnel.class);
            } catch (Exception e) {

            }
        }
        for (AssoProjectPerson assoProjectPerson : assoProjectPersonList) {
            ProjectPersonVO projectPersonVO = new ProjectPersonVO();
            Personnel personnel1 = personnels.stream().filter(item -> item.getId().equals(assoProjectPerson.getPersonId())).findFirst().orElse(null);
            Personnel personnel2 = personnels.stream().filter(item -> item.getId().equals(assoProjectPerson.getCreateId())).findFirst().orElse(null);
            BeanUtils.copyProperties(assoProjectPerson, projectPersonVO);
            if (personnel1 != null) {
                projectPersonVO.setPersonName(personnel1.getPersonnelName());
            }
            if (personnel2 != null) {
                projectPersonVO.setCreateName(personnel2.getPersonnelName());
            }
            projectPersonVOS.add(projectPersonVO);
        }
        return projectPersonVOS;
    }
}