| | |
|---|
| | | package com.ruoyi.device.service.impl; |
|---|
| | | |
|---|
| | | import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; |
|---|
| | | import com.baomidou.mybatisplus.core.metadata.IPage; |
|---|
| | | import com.baomidou.mybatisplus.extension.plugins.pagination.Page; |
|---|
| | | import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; |
|---|
| | | import com.ruoyi.common.utils.SecurityUtils; |
|---|
| | | import com.ruoyi.common.utils.bean.BeanUtils; |
|---|
| | | import com.ruoyi.common.utils.poi.ExcelUtil; |
|---|
| | | import com.ruoyi.device.dto.DeviceLedgerDto; |
|---|
| | |
|---|
| | | import com.ruoyi.device.pojo.DeviceLedger; |
|---|
| | | import com.ruoyi.device.service.IDeviceLedgerService; |
|---|
| | | import com.ruoyi.framework.web.domain.AjaxResult; |
|---|
| | | import com.ruoyi.project.system.domain.SysUser; |
|---|
| | | import com.ruoyi.project.system.mapper.SysUserMapper; |
|---|
| | | import lombok.AllArgsConstructor; |
|---|
| | | import lombok.extern.slf4j.Slf4j; |
|---|
| | | import org.springframework.beans.factory.annotation.Autowired; |
|---|
| | | import org.springframework.stereotype.Service; |
|---|
| | | import org.springframework.web.multipart.MultipartFile; |
|---|
| | | |
|---|
| | | import javax.servlet.http.HttpServletResponse; |
|---|
| | | import java.io.IOException; |
|---|
| | | import java.util.ArrayList; |
|---|
| | | import java.util.Arrays; |
|---|
| | | import java.util.List; |
|---|
| | |
|---|
| | | |
|---|
| | | @Override |
|---|
| | | public AjaxResult saveDeviceLedger(DeviceLedger deviceLedger) { |
|---|
| | | LambdaQueryWrapper<DeviceLedger> deviceLedgerLambdaQueryWrapper = new LambdaQueryWrapper<>(); |
|---|
| | | deviceLedgerLambdaQueryWrapper.eq(DeviceLedger::getDeviceName,deviceLedger.getDeviceName()); |
|---|
| | | if (this.count(deviceLedgerLambdaQueryWrapper) > 0) { |
|---|
| | | return AjaxResult.error("设备名称已存在"); |
|---|
| | | } |
|---|
| | | boolean save = this.save(deviceLedger); |
|---|
| | | if (save){ |
|---|
| | | return AjaxResult.success(); |
|---|
| | |
|---|
| | | util.exportExcel(response, deviceLedgerExeclDtos, "设备台账导出"); |
|---|
| | | }else { |
|---|
| | | ArrayList<Long> arrayList = new ArrayList<>(); |
|---|
| | | Arrays.stream(ids).map(id -> { |
|---|
| | | return arrayList.add( id); |
|---|
| | | }); |
|---|
| | | Arrays.stream(ids).map(arrayList::add); |
|---|
| | | List<DeviceLedger> supplierManageList = deviceLedgerMapper.selectBatchIds(arrayList); |
|---|
| | | ArrayList<DeviceLedgerExeclDto> deviceLedgerExeclDtos = new ArrayList<>(); |
|---|
| | | supplierManageList.stream().forEach(deviceLedger -> { |
|---|
| | |
|---|
| | | } |
|---|
| | | |
|---|
| | | } |
|---|
| | | |
|---|
| | | @Override |
|---|
| | | public Boolean importData(MultipartFile file) throws IOException { |
|---|
| | | ExcelUtil<DeviceLedgerExeclDto> util = new ExcelUtil<DeviceLedgerExeclDto>(DeviceLedgerExeclDto.class); |
|---|
| | | List<DeviceLedgerExeclDto> userList = util.importExcel(file.getInputStream()); |
|---|
| | | userList.forEach(c -> { |
|---|
| | | DeviceLedger deviceLedger = new DeviceLedger(); |
|---|
| | | SysUser sysUser = sysUserMapper.selectUserByUserName(c.getCreateUser()); |
|---|
| | | if (sysUser!=null) { |
|---|
| | | deviceLedger.setCreateUser(sysUser.getUserId().intValue()); |
|---|
| | | }else { |
|---|
| | | deviceLedger.setCreateUser(SecurityUtils.getUserId().intValue()); |
|---|
| | | } |
|---|
| | | BeanUtils.copyProperties(c,deviceLedger); |
|---|
| | | deviceLedgerMapper.insert(deviceLedger); |
|---|
| | | }); |
|---|
| | | |
|---|
| | | return true; |
|---|
| | | } |
|---|
| | | } |
|---|
