package com.ruoyi.device.service.impl;
|
|
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.bean.BeanUtils;
|
import com.ruoyi.common.utils.poi.ExcelUtil;
|
import com.ruoyi.device.dto.DeviceLedgerDto;
|
import com.ruoyi.device.execl.DeviceLedgerExeclDto;
|
import com.ruoyi.device.mapper.DeviceLedgerMapper;
|
import com.ruoyi.device.pojo.DeviceLedger;
|
import com.ruoyi.device.service.IDeviceLedgerService;
|
import com.ruoyi.framework.web.domain.AjaxResult;
|
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 javax.servlet.http.HttpServletResponse;
|
import java.util.ArrayList;
|
import java.util.Arrays;
|
import java.util.List;
|
|
@Service
|
@AllArgsConstructor
|
@Slf4j
|
public class DeviceLedgerServiceImpl extends ServiceImpl<DeviceLedgerMapper, DeviceLedger> implements IDeviceLedgerService {
|
|
@Autowired
|
private DeviceLedgerMapper deviceLedgerMapper;
|
@Autowired
|
private SysUserMapper sysUserMapper;
|
|
@Override
|
public IPage<DeviceLedgerDto> queryPage(Page page, DeviceLedgerDto deviceLedger) {
|
|
return deviceLedgerMapper.queryPage(page, deviceLedger);
|
}
|
|
@Override
|
public AjaxResult saveDeviceLedger(DeviceLedger deviceLedger) {
|
boolean save = this.save(deviceLedger);
|
if (save){
|
return AjaxResult.success();
|
}
|
return AjaxResult.error();
|
}
|
|
@Override
|
public AjaxResult updateDeviceLedger(DeviceLedger deviceLedger) {
|
if (this.updateById(deviceLedger)) {
|
return AjaxResult.success();
|
}
|
return AjaxResult.error();
|
}
|
|
@Override
|
public void export(HttpServletResponse response, Long[] ids) {
|
|
if (ids == null || ids.length == 0) {
|
List<DeviceLedger> supplierManageList = this.list();
|
|
ArrayList<DeviceLedgerExeclDto> deviceLedgerExeclDtos = new ArrayList<>();
|
supplierManageList.stream().forEach(deviceLedger -> {
|
DeviceLedgerExeclDto deviceLedgerExeclDto = new DeviceLedgerExeclDto();
|
BeanUtils.copyProperties(deviceLedger,deviceLedgerExeclDto);
|
deviceLedgerExeclDto.setCreateUser(sysUserMapper.selectUserById(Long.valueOf(deviceLedger.getCreateUser().toString())).getUserName());
|
deviceLedgerExeclDtos.add(deviceLedgerExeclDto);
|
});
|
ExcelUtil<DeviceLedgerExeclDto> util = new ExcelUtil<DeviceLedgerExeclDto>(DeviceLedgerExeclDto.class);
|
util.exportExcel(response, deviceLedgerExeclDtos, "设备台账导出");
|
}else {
|
ArrayList<Long> arrayList = new ArrayList<>();
|
Arrays.stream(ids).map(id -> {
|
return arrayList.add( id);
|
});
|
List<DeviceLedger> supplierManageList = deviceLedgerMapper.selectBatchIds(arrayList);
|
ArrayList<DeviceLedgerExeclDto> deviceLedgerExeclDtos = new ArrayList<>();
|
supplierManageList.stream().forEach(deviceLedger -> {
|
DeviceLedgerExeclDto deviceLedgerExeclDto = new DeviceLedgerExeclDto();
|
BeanUtils.copyProperties(deviceLedger,deviceLedgerExeclDto);
|
deviceLedgerExeclDto.setCreateUser(sysUserMapper.selectUserById(Long.valueOf(deviceLedger.getCreateUser().toString())).getUserName());
|
deviceLedgerExeclDtos.add(deviceLedgerExeclDto);
|
});
|
ExcelUtil<DeviceLedgerExeclDto> util = new ExcelUtil<DeviceLedgerExeclDto>(DeviceLedgerExeclDto.class);
|
util.exportExcel(response, deviceLedgerExeclDtos, "设备台账导出");
|
}
|
|
}
|
}
|
