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.Mould;
import com.hssx.cloudmodel.entity.MouldEquipment;
import com.hssx.cloudmodel.entity.User;
import com.hssx.cloudmodel.entity.vo.MouldEquipmentVO;
import com.hssx.cloudmodel.mapper.MouldEquipmentMapper;
import com.hssx.cloudmodel.mapper.MouldMapper;
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.springframework.stereotype.Service;
import org.springframework.web.bind.annotation.RequestParam;

import javax.annotation.Resource;
import java.time.LocalDateTime;
import java.time.LocalTime;
import java.util.ArrayList;
import java.util.List;

/**
 * <p>
 * 服务实现类
 * </p>
 *
 * @author 吴涛涛
 * @since 2019-08-02
 */
@Service
public class MouldEquipmentServiceImpl extends ServiceImpl<MouldEquipmentMapper, MouldEquipment> implements MouldEquipmentService {
    @Resource
    MouldEquipmentMapper mouldEquipmentMapper;
    @Resource
    MouldMapper mouldMapper;

    @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());
            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;
    }

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

    @Override
    public HttpRespMsg getListByCompanyId(MouldEquipmentVO mouldEquipmentVO) {
        HttpRespMsg msg = new HttpRespMsg();
        List<Integer> ides = new ArrayList<>();
        List<Mould> moulds = mouldMapper.selectList(new QueryWrapper<Mould>().eq("company_id", mouldEquipmentVO.getBelongCompanyId()));
        for (Mould mould : moulds) {
            ides.add(mould.getEquipmentId());
        }
        if (ides.size() == 0) {
            ides.add(-1);
        }
        List<MouldEquipmentVO> list = mouldEquipmentMapper.getListByCompanyId(mouldEquipmentVO.getBelongCompanyId(), ides);
        msg.data = list;
        return msg;
    }
}