森林防火 供应商管理

rentaiming

2024-07-30 837da67d468fbcdd10ee27dcc44a94e7b1130a85

 ruoyi-modules/ruoyi-management/src/main/java/com/ruoyi/management/service/impl/SlEquipmentServiceImpl.java

@@ -1,10 +1,28 @@
package com.ruoyi.management.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.ruoyi.common.core.utils.page.PageDTO;
import com.ruoyi.management.domain.SlEquipment;
import com.ruoyi.management.domain.SlGoodsMaterials;
import com.ruoyi.management.domain.SlStoreManagement;
import com.ruoyi.management.domain.dto.SlEquipmentDTO;
import com.ruoyi.management.domain.dto.SlEquipmentQuery;
import com.ruoyi.management.domain.vo.SlEquipmentVO;
import com.ruoyi.management.domain.vo.SlGoodsMaterialsVO;
import com.ruoyi.management.mapper.SlEquipmentMapper;
import com.ruoyi.management.mapper.SlStoreManagementMapper;
import com.ruoyi.management.service.SlEquipmentService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.system.api.domain.SeBei;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

/**
 * <p>
@@ -17,4 +35,134 @@
@Service
public class SlEquipmentServiceImpl extends ServiceImpl<SlEquipmentMapper, SlEquipment> implements SlEquipmentService {

    @Resource
    private SlStoreManagementMapper slStoreManagementMapper;

    @Override
    public PageDTO<SlEquipmentVO> getSlEquipmentList(SlEquipmentQuery slEquipmentQuery) {
        Page<SlEquipment> page = new Page<>(slEquipmentQuery.getPageCurr(), slEquipmentQuery.getPageSize());
        LambdaQueryWrapper< SlEquipment> wrapper= Wrappers.lambdaQuery();
        if(slEquipmentQuery.getEquipmentName()!=null&&slEquipmentQuery.getEquipmentName()!=""){
            wrapper.like(SlEquipment::getEquipmentName,slEquipmentQuery.getEquipmentName());
        }
        if(slEquipmentQuery.getEquipmentName()!=null&&slEquipmentQuery.getType()!=0){
        wrapper.eq(SlEquipment::getType,slEquipmentQuery.getType());
        }
        wrapper.eq( SlEquipment::getDelFlag,0);
        wrapper.orderByDesc(SlEquipment::getCreateTime);
        Page<SlEquipment> page1 = this.page(page, wrapper);
        PageDTO<SlEquipmentVO> slEquipmentVOPageDTO = PageDTO.of(page1, SlEquipmentVO.class);
        List<SlEquipmentVO> list = slEquipmentVOPageDTO.getList();
        for (SlEquipmentVO sl:list){
            SlStoreManagement slStoreManagement = slStoreManagementMapper.selectById(sl.getManagementId());
            sl.setStoreManagementName(slStoreManagement.getStoreManagementName());
            sl.setCountyCode(slStoreManagement.getCountyCode());
            sl.setProvinceName(slStoreManagement.getProvinceName());
            sl.setProvinceCode(slStoreManagement.getProvinceCode());
            sl.setCountyName(slStoreManagement.getCountyName());
            sl.setCityCode(slStoreManagement.getCityCode());
            sl.setCityName(slStoreManagement.getCityName());
            sl.setTownName(slStoreManagement.getTownName());
            sl.setTownCode(slStoreManagement.getTownCode());
        }
        return slEquipmentVOPageDTO;
    }

    @Override
    public void addSlEquipment(SlEquipmentDTO slEquipmentDTO) {
        if (slEquipmentDTO.getId()==null){
            SlEquipment slEquipment=new SlEquipment();
            slEquipment.setEquipmentName(slEquipmentDTO.getEquipmentName());
            slEquipment.setEquipmentNum(slEquipmentDTO.getEquipmentNum());
            slEquipment.setType(slEquipmentDTO.getType());
            slEquipment.setManagementId(slEquipmentDTO.getManagementId());
            this.save(slEquipment);
        }else{
            SlEquipment byId = this.getById(slEquipmentDTO.getId());
            if (slEquipmentDTO.getEquipmentName()!=null){
                byId.setEquipmentName(slEquipmentDTO.getEquipmentName());
            }
            if (slEquipmentDTO.getEquipmentNum()!=null){
                byId.setEquipmentNum(slEquipmentDTO.getEquipmentNum());
            }
            if (slEquipmentDTO.getType()!=null){
                byId.setType(slEquipmentDTO.getType());
            }
            if (slEquipmentDTO.getManagementId()!=null){
                byId.setManagementId(slEquipmentDTO.getManagementId());
            }
            this.updateById(byId);
        }
    }

    @Override
    public SlEquipmentVO getSlEquipmentOne(SlEquipmentDTO slEquipmentDTO) {
        SlEquipment byId = this.getById(slEquipmentDTO.getId());
        SlEquipmentVO slEquipmentVO=new SlEquipmentVO();
        slEquipmentVO.setEquipmentName(byId.getEquipmentName());
        slEquipmentVO.setEquipmentNum(byId.getEquipmentNum());
        slEquipmentVO.setType(byId.getType());
        slEquipmentVO.setManagementId(byId.getManagementId());

        SlStoreManagement slStoreManagement = slStoreManagementMapper.selectById(byId.getManagementId());
        slEquipmentVO.setCountyCode(slStoreManagement.getCountyCode());
        slEquipmentVO.setStoreManagementName(slStoreManagement.getStoreManagementName());
        slEquipmentVO.setProvinceName(slStoreManagement.getProvinceName());
        slEquipmentVO.setProvinceCode(slStoreManagement.getProvinceCode());
        slEquipmentVO.setCountyName(slStoreManagement.getCountyName());
        slEquipmentVO.setCityCode(slStoreManagement.getCityCode());
        slEquipmentVO.setCityName(slStoreManagement.getCityName());
        slEquipmentVO.setTownName(slStoreManagement.getTownName());
        slEquipmentVO.setTownCode(slStoreManagement.getTownCode());

        return slEquipmentVO;
    }

    @Override
    public Integer getequipmente(SeBei seBei) {
        LambdaQueryWrapper< SlStoreManagement> wrapper= Wrappers.lambdaQuery();

        List<Long> arr=new ArrayList<>();
        wrapper.eq(SlStoreManagement::getDelFlag,0);
        wrapper.like(SlStoreManagement::getAdministratorId,seBei.getUserId());
        List<SlStoreManagement> page = slStoreManagementMapper.selectList(wrapper);
        for (SlStoreManagement a:page){
            arr.add(a.getId());
        }

        LambdaQueryWrapper< SlStoreManagement> wrapper1= Wrappers.lambdaQuery();
        wrapper1.eq(SlStoreManagement::getDelFlag,0);
        wrapper1.like(SlStoreManagement::getCompetentId,seBei.getUserId());
        List<SlStoreManagement> page1 = slStoreManagementMapper.selectList(wrapper1);

        for (SlStoreManagement a1:page1){
            arr.add(a1.getId());
        }

        LambdaQueryWrapper< SlStoreManagement> wrapper2= Wrappers.lambdaQuery();
        wrapper2.eq(SlStoreManagement::getDelFlag,0);
        wrapper2.like(SlStoreManagement::getDirectorId,seBei.getUserId());
        List<SlStoreManagement> page2 = slStoreManagementMapper.selectList(wrapper2);
        for (SlStoreManagement a2:page2){
            arr.add(a2.getId());
        }
        Set<Long> set = new HashSet<>(arr);
        arr.clear();
        arr.addAll(set);

        int i=1;
        for (Long arr1:arr){
            LambdaQueryWrapper< SlEquipment> wrapper3= Wrappers.lambdaQuery();
                wrapper3.eq(SlEquipment::getManagementId,arr1);
                wrapper3.eq(SlEquipment::getEquipmentNum,seBei.getSebei());
            wrapper3.eq( SlEquipment::getDelFlag,0);
            List<SlEquipment> page3 = this.list( wrapper3);
            if (page3.size()>0){
                i=2;
                break;
            }

        }
      return i;
    }
}