yunsu/cloud-model/src/main/java/com/hssx/cloudmodel/service/impl/MouldEquipmentServiceImpl.java

package com.hssx.cloudmodel.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.hssx.cloudmodel.constant.Constant;
import com.hssx.cloudmodel.entity.*;
import com.hssx.cloudmodel.entity.vo.MouldEquipmentVO;
import com.hssx.cloudmodel.entity.vo.UserVO;
import com.hssx.cloudmodel.mapper.*;
import com.hssx.cloudmodel.service.MouldEquipmentService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.hssx.cloudmodel.util.HttpRespMsg;
import com.hssx.cloudmodel.util.PageUtil;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.stereotype.Service;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.time.LocalDateTime;
import java.time.LocalTime;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

/**
 * <p>
 * 服务实现类
 * </p>
 *
 * @author 吴涛涛
 * @since 2019-08-02
 */
@Service
public class MouldEquipmentServiceImpl extends ServiceImpl<MouldEquipmentMapper, MouldEquipment> implements MouldEquipmentService {
    @Resource
    MouldEquipmentMapper mouldEquipmentMapper;
    @Resource
    MouldMapper mouldMapper;
    @Resource
    UserMapper userMapper;
    @Resource
    NewsNoticeMapper newsNoticeMapper;
    @Resource
    NewsNoticeUserMapper newsNoticeUserMapper;
    @Resource
    PowerMapper powerMapper;
    @Resource
    ProjectMapper projectMapper;
    @Resource
    ProjectApproveMapper projectApproveMapper;
    @Resource
    CustomCompanyMapper customCompanyMapper;
    @Resource
    CompanyMapper companyMapper;
    @Resource
    ProjectUserMapper projectUserMapper;

    @Override
    public HttpRespMsg addAndUpdateMouldEquipment(MouldEquipment mouldEquipment, User user) {
        HttpRespMsg msg = new HttpRespMsg();
        if (user.getParentId() == 0) {
            if (mouldEquipment.getId() == null) {
                //添加设备
                //查询当前设备编号的模具是否存在
                Integer count = mouldEquipmentMapper.selectCount(new QueryWrapper<MouldEquipment>().eq("equipment_no", mouldEquipment.getEquipmentNo()));
                if (count > 0) {
                    msg.setError("当前设备编号已存在，请重新输入其他模具编号");
                } else {
                    mouldEquipmentMapper.insert(mouldEquipment);
                }
            } else {
                //修改设备
                MouldEquipment m = mouldEquipmentMapper.selectOne(new QueryWrapper<MouldEquipment>().eq("equipment_no", mouldEquipment.getEquipmentNo()));
                if ((m != null && m.getId() == mouldEquipment.getId()) || m == null) {
                    mouldEquipmentMapper.updateById(mouldEquipment);
                } else {
                    msg.setError("当前设备编号已存在，请重新输入其他模具编号");
                }
            }
        } else {
            msg.setError("对不起，您不是管理员，不具备设备创建或修改的权限");
        }
        return msg;
    }

    @Override
    public HttpRespMsg isUse(MouldEquipment mouldEquipment, User user) {
        HttpRespMsg msg = new HttpRespMsg();
        //判断是不是超级管理员，是才可操作
        if (user.getParentId() == 0) {
            //修改设备
            MouldEquipment newMouldEquipment = mouldEquipmentMapper.selectById(mouldEquipment.getId());
            if (null == newMouldEquipment.getBelongCompanyId()) {
                msg.setError("该设备没有所属公司，请添加所属公司后再启用");
                return msg;
            }
            newMouldEquipment.setIsUse(mouldEquipment.getIsUse());
            LocalDateTime now = LocalDateTime.now();
            newMouldEquipment.setStartTime(now);
            newMouldEquipment.setEndTime(now.plusYears(newMouldEquipment.getUseLife()));
            mouldEquipmentMapper.updateById(newMouldEquipment);
        } else {
            msg.setError("对不起，您不是管理员，不具备启用设备的权限");
            return msg;
        }
        return msg;
    }

    @Override
    public HttpRespMsg getList(User user, PageUtil page, Integer companyId, String keyName) {
        HttpRespMsg msg = new HttpRespMsg();
        if (user != null) {
            if (Constant.SYS_PARENT_ID == user.getParentId()) {
                PageHelper.startPage(page.getPageNum(), page.getPageSize());
                List<MouldEquipmentVO> mouldEquipments = mouldEquipmentMapper.getList(companyId, keyName);
                PageInfo<MouldEquipmentVO> pageInfo = new PageInfo<>(mouldEquipments);
                msg.data = pageInfo;
            } else {
                msg.setError("对不起！您不含有查看该列表的权利。");
            }
        }
        return msg;
    }

    @Override
    public HttpRespMsg getListByCompanyId(MouldEquipmentVO mouldEquipmentVO) {
        HttpRespMsg msg = new HttpRespMsg();
        List<Integer> mouldsEquipmentIds = mouldMapper.selectList(new QueryWrapper<Mould>().eq("company_id", mouldEquipmentVO.getBelongCompanyId()).isNotNull("equipment_id")).stream().map(Mould::getEquipmentId).collect(Collectors.toList());
        mouldsEquipmentIds.add(-1);
        List<Integer> equipmentIds = mouldEquipmentMapper.selectList(new QueryWrapper<MouldEquipment>().eq("belong_company_id", mouldEquipmentVO.getBelongCompanyId())).stream().map(MouldEquipment::getId).collect(Collectors.toList());
        equipmentIds.add(-1);
        equipmentIds.removeAll(mouldsEquipmentIds);//把已经使用的设备剔除
        List<MouldEquipmentVO> list = mouldEquipmentMapper.getListByCompanyId(mouldEquipmentVO.getBelongCompanyId(), equipmentIds);
        msg.data = list;
        return msg;
    }

    @Override
    public HttpRespMsg MouldEquipmentAlarm() {
        HttpRespMsg msg = new HttpRespMsg();
        List<MouldEquipment> mouldEquipments = mouldEquipmentMapper.selectList(new QueryWrapper<MouldEquipment>().eq("is_use", 1));
        for (MouldEquipment mouldEquipment : mouldEquipments) {
            Mould mould = mouldMapper.selectOne(new QueryWrapper<Mould>().eq("equipment_id", mouldEquipment.getId()));
            if (Constant.ELECTRICITY_THRESHOLD >= Double.parseDouble(mouldEquipment.getHillNumber())) {
                //公众号的推送to do

                Project project = projectMapper.selectById(mould.getProjectId());
                //查询到参与该项目的人
                List<Integer> userIds = new ArrayList<>();
                userIds.add(-1);
                userIds.add(project.getCreatorId());
                userIds.add(project.getManagerId());
                List<ProjectUser> projectUsers = projectUserMapper.selectList(new QueryWrapper<ProjectUser>().eq("project_id", project.getId()));
                for (ProjectUser projectUser : projectUsers) {
                    userIds.add(projectUser.getUserId());
                }
                List<ProjectApprove> projectApproves = projectApproveMapper.selectList(new QueryWrapper<ProjectApprove>().eq("project_id", project.getId()));
                for (ProjectApprove projectApprove : projectApproves) {
                    userIds.add(projectApprove.getApproverId());
                }
                List<User> users = userMapper.selectList(new QueryWrapper<User>().in("id", userIds));
                //查询到之前模具保养的消息
                NewsNotice oldNews = newsNoticeMapper.selectOne(new QueryWrapper<NewsNotice>().eq("ref_id", mould.getId()).eq("notice_type", 2));
                NewsNotice newsNotice = new NewsNotice();
                if (oldNews != null) {
                    NewsNotice newNews = new NewsNotice();
                    newNews.setId(oldNews.getId());
                    oldNews.setIndate(LocalDateTime.now());
                    newsNoticeMapper.updateById(oldNews);
                    newsNoticeUserMapper.updateNewsNoticeUserByNewsNoticeId(oldNews.getId());
//                    for (User u : users) {
//                        //添加通知的消息
//                        NewsNoticeUser newsNoticeUser = new NewsNoticeUser();
//                        newsNoticeUser.setNewsId(oldNews.getId());
//                        newsNoticeUser.setUserId(u.getId());
//                        newsNoticeUser.setIsRead(0);
//                        newsNoticeUserMapper.updateById(newsNoticeUser);
//                    }
                } else {
                    newsNotice.setNoticeType(Constant.EMERGENCY_TYPE);
                    newsNotice.setProjectId(project.getId());
                    newsNotice.setProjectName(project.getProjectName() + "-" + mould.getModelName());
//                    newsNotice.setRefId(mould.getId());
                    newsNotice.setContent(mouldEquipment.getEquipmentNo() + "-" + Constant.EMERGENCY_ELECTRICITY_NOTICE);
                    newsNoticeMapper.insert(newsNotice);
                    for (User u : users) {
                        //添加通知的消息
                        NewsNoticeUser newsNoticeUser = new NewsNoticeUser();
                        newsNoticeUser.setNewsId(newsNotice.getId());
                        newsNoticeUser.setUserId(u.getId());
                        newsNoticeUserMapper.insert(newsNoticeUser);
                    }
                }
            }
            if (Constant.TEMPERATURE_THRESHOLD <= mouldEquipment.getTemperature()) {
                //公众号的推送to do

                Project project = projectMapper.selectById(mould.getProjectId());
                //查询到参与该项目的人
                List<Integer> userIds = new ArrayList<>();
                userIds.add(-1);
                userIds.add(project.getCreatorId());
                userIds.add(project.getManagerId());
                List<ProjectUser> projectUsers = projectUserMapper.selectList(new QueryWrapper<ProjectUser>().eq("project_id", project.getId()));
                for (ProjectUser projectUser : projectUsers) {
                    userIds.add(projectUser.getUserId());
                }
                List<ProjectApprove> projectApproves = projectApproveMapper.selectList(new QueryWrapper<ProjectApprove>().eq("project_id", project.getId()));
                for (ProjectApprove projectApprove : projectApproves) {
                    userIds.add(projectApprove.getApproverId());
                }
                List<User> users = userMapper.selectList(new QueryWrapper<User>().in("id", userIds));
                //查询到之前模具保养的消息
                NewsNotice oldNews = newsNoticeMapper.selectOne(new QueryWrapper<NewsNotice>().eq("ref_id", mould.getId()).eq("notice_type", 2));
                NewsNotice newsNotice = new NewsNotice();
                if (oldNews != null) {
                    NewsNotice newNews = new NewsNotice();
                    newNews.setId(oldNews.getId());
                    oldNews.setIndate(LocalDateTime.now());
                    newsNoticeMapper.updateById(oldNews);
                    newsNoticeUserMapper.updateNewsNoticeUserByNewsNoticeId(oldNews.getId());
//                    for (User u : users) {
//                        //添加通知的消息
//                        NewsNoticeUser newsNoticeUser = new NewsNoticeUser();
//                        newsNoticeUser.setNewsId(oldNews.getId());
//                        newsNoticeUser.setUserId(u.getId());
//                        newsNoticeUser.setIsRead(0);
//                        newsNoticeUserMapper.updateById(newsNoticeUser);
//                    }
                } else {
                    newsNotice.setNoticeType(Constant.EMERGENCY_TYPE);
                    newsNotice.setProjectId(project.getId());
                    newsNotice.setProjectName(project.getProjectName() + "-" + mould.getModelName());
//                    newsNotice.setRefId(mould.getId());
                    newsNotice.setContent(mouldEquipment.getEquipmentNo() + "-" + Constant.EMERGENCY_TEMPERATURE_NOTICE);
                    newsNoticeMapper.insert(newsNotice);
                    for (User u : users) {
                        //添加通知的消息
                        NewsNoticeUser newsNoticeUser = new NewsNoticeUser();
                        newsNoticeUser.setNewsId(newsNotice.getId());
                        newsNoticeUser.setUserId(u.getId());
                        newsNoticeUserMapper.insert(newsNoticeUser);
                    }
                }
            }
        }
        return msg;
    }

    @Override
    public HttpRespMsg importMouldEquipmentExcel(MultipartFile file, UserVO userVO) {
        HttpRespMsg msg = new HttpRespMsg();
        User user = userMapper.selectOne(new QueryWrapper<User>().eq("head_imgurl", userVO.getToken()));
        if (user != null) {
            List<MouldEquipment> mouldEquipments = mouldEquipmentMapper.selectList(new QueryWrapper<MouldEquipment>());
            try {
                File f = null;
                if ("".equals(file) || file.getSize() <= 0) {
                    file = null;
                } else {
                    //获取输入流
                    InputStream ins = file.getInputStream();
                    //新建一个文件
                    f = new File(file.getOriginalFilename());
                    //输入流转file
                    inputStreamToFile(ins, f);
                }
                //根据文件创建工作簿
                XSSFWorkbook wookbook = new XSSFWorkbook(f);
                XSSFSheet sheet = wookbook.getSheetAt(0);
                int s = sheet.getLastRowNum();
                // 遍历当前sheet中的所有行，第一行是数据对应的字段，不是数据，
                // 故从第二行开始遍历拿数据（如果有标题的话，则从第三行开始拿数据）
                for (int j = 1; j < sheet.getLastRowNum() + 1; j++) {
                    XSSFRow row = sheet.getRow(j);
                    //新建云模盒对象
                    MouldEquipment mouldEquipment = new MouldEquipment();
                    // 遍历所有的列，下面的10是excle表格里共有10列即对应了10个字段
                    for (int y = 0; y < 3; y++) {
                        XSSFCell cell = row.getCell(y);
                        cell.setCellType(Cell.CELL_TYPE_STRING);
                        //取出当前列的值
                        String value = cell.getStringCellValue();
                        //判断第几列插入数据，后面就是从列中取数据往对象里放，然后插入到数据库里
                        if (value == null && "".equals(value)) {
                            log.error("数据不可为空");
                            msg.setError("数据不可为空");
                            return msg;
                        } else if (y == 0) {
                            //云模编号
                            for (MouldEquipment p : mouldEquipments) {
                                if (value.equals(p.getEquipmentNo())) {
                                    msg.setError("第" + j + "行的云模盒编号:" + value + "已被占用，请修改后重新上传");
                                    return msg;
                                }
                            }
                            mouldEquipment.setEquipmentNo(value);
                        } else if (y == 1) {
                            //使用年限
                            mouldEquipment.setUseLife(Integer.parseInt(value));
                        } else if (y == 2) {
                            //代理商
                            mouldEquipment.setAgent(value);
                        }
                    }
                    mouldEquipmentMapper.insert(mouldEquipment);
                }
                //用完后删除临时文件
                if (!f.isDirectory()) {
                    f.delete();
                }
            } catch (Exception e) {
                log.error(e.getMessage(), e);
                msg.setError(e.getMessage());
                return msg;
            }
        } else {
            msg.setError("用户不存在或者未登录");
        }
        return msg;
    }

    /**
     * 输入流转file
     *
     * @param ins
     * @param file
     */
    public static void inputStreamToFile(InputStream ins, File file) {
        try {
            OutputStream os = new FileOutputStream(file);
            int bytesRead = 0;
            byte[] buffer = new byte[8192];
            while ((bytesRead = ins.read(buffer, 0, 8192)) != -1) {
                os.write(buffer, 0, bytesRead);
            }
            os.close();
            ins.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}