package com.gkhy.exam.system.service.impl;
|
|
import com.alibaba.excel.EasyExcel;
|
import com.alibaba.excel.ExcelReader;
|
import com.alibaba.excel.context.AnalysisContext;
|
import com.alibaba.excel.read.listener.ReadListener;
|
import com.alibaba.excel.read.metadata.ReadSheet;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.gkhy.exam.common.api.CommonPage;
|
import com.gkhy.exam.common.api.CommonResult;
|
import com.gkhy.exam.common.api.ImportResult;
|
import com.gkhy.exam.common.domain.entity.SysDept;
|
import com.gkhy.exam.common.domain.entity.SysUser;
|
import com.gkhy.exam.common.utils.PageUtils;
|
import com.gkhy.exam.common.utils.SecurityUtils;
|
import com.gkhy.exam.system.domain.DTO.StandingBookImportDTO;
|
import com.gkhy.exam.system.domain.StandingBook;
|
import com.gkhy.exam.system.domain.vo.DeptVo;
|
import com.gkhy.exam.system.mapper.StandingBookMapper;
|
import com.gkhy.exam.system.mapper.SysDeptMapper;
|
import com.gkhy.exam.system.mapper.SysUserMapper;
|
import com.gkhy.exam.system.service.StandingBookService;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.time.LocalDateTime;
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
/**
|
* <p>
|
* 台账 服务实现类
|
* </p>
|
*
|
* @author hh
|
* @since 2025-07-31 16:12:02
|
*/
|
@Service
|
public class StandingBookServiceImpl extends ServiceImpl<StandingBookMapper, StandingBook> implements StandingBookService {
|
|
@Autowired
|
private StandingBookMapper standingBookMapper;
|
|
@Autowired
|
private SysDeptMapper sysDeptMapper;
|
|
@Autowired
|
|
private SysUserMapper sysUserMapper;
|
@Override
|
public CommonPage selectStandingBookList(StandingBook standingBook) {
|
PageUtils.startPage();
|
List<StandingBook> standingBooks = standingBookMapper.getStandingBooks(standingBook);
|
return CommonPage.restPage(standingBooks);
|
}
|
|
@Override
|
public CommonResult insertStandingBook(StandingBook standingBook) {
|
standingBook.setCreateBy(SecurityUtils.getUsername());
|
standingBook.setCreateTime(LocalDateTime.now());
|
int insert = standingBookMapper.insert(standingBook);
|
if (insert > 0){
|
return CommonResult.success();
|
}
|
return CommonResult.failed();
|
}
|
|
@Override
|
public CommonResult updateStandingBook(StandingBook standingBook) {
|
standingBook.setUpdateBy(SecurityUtils.getUsername());
|
standingBook.setUpdateTime(LocalDateTime.now());
|
int insert = standingBookMapper.updateById(standingBook);
|
if (insert > 0){
|
return CommonResult.success();
|
}
|
return CommonResult.failed();
|
}
|
|
@Override
|
public CommonResult deletedStandingBook(Integer id) {
|
|
StandingBook standingBook = new StandingBook();
|
standingBook.setId(id.longValue());
|
standingBook.setDelFlag(1);
|
standingBook.setUpdateBy(SecurityUtils.getUsername());
|
standingBook.setUpdateTime(LocalDateTime.now());
|
int update = standingBookMapper.updateById(standingBook);
|
if (update > 0){
|
return CommonResult.success();
|
}
|
return CommonResult.failed();
|
}
|
|
@Override
|
@Transactional
|
public ImportResult importStandingBooks(Long companyId, MultipartFile file) {
|
ImportResult result = new ImportResult();
|
result.setSuccessCount(0);
|
result.setFailCount(0);
|
result.setErrorMessages(new ArrayList<>());
|
|
SysDept sysDept = new SysDept();
|
sysDept.setCompanyId(companyId);
|
List<DeptVo> deptVos = sysDeptMapper.selectDeptList(sysDept);
|
Map<String, Long> deptNameIdMap = deptVos.stream()
|
.collect(Collectors.toMap(DeptVo::getDeptName, DeptVo::getDeptId));
|
|
LambdaQueryWrapper<SysUser> queryWrapper = new LambdaQueryWrapper<>();
|
queryWrapper.eq(SysUser::getCompanyId, companyId);
|
queryWrapper.eq(SysUser::getDelFlag, 0);
|
List<SysUser> sysUsers = sysUserMapper.selectList(queryWrapper);
|
Map<String, Long> userNameIdMap = sysUsers.stream()
|
.collect(Collectors.toMap(SysUser::getName, SysUser::getId));
|
|
try {
|
// 使用EasyExcel读取文件
|
ReadListener<StandingBookImportDTO> listener = new ReadListener<StandingBookImportDTO>() {
|
@Override
|
public void invoke(StandingBookImportDTO dto, AnalysisContext context) {
|
try {
|
// 转换为实体对象
|
StandingBook standingBook = new StandingBook();
|
standingBook.setCompanyId(companyId);
|
standingBook.setDeptId(deptNameIdMap.get(dto.getDeptName()) == null ? 0L : deptNameIdMap.get(dto.getDeptName()));
|
standingBook.setName(dto.getName());
|
standingBook.setModel(dto.getModel());
|
standingBook.setPersonResponsible(userNameIdMap.get(dto.getPersonResponsible()) == null ? 0L : userNameIdMap.get(dto.getPersonResponsible()));
|
String deviceType = dto.getDeviceType();
|
switch (deviceType) {
|
case "计算机设备":
|
standingBook.setDeviceType(1);
|
break;
|
case "办公自动化设备":
|
standingBook.setDeviceType(2);
|
break;
|
case "外部设备":
|
standingBook.setDeviceType(3);
|
break;
|
case "生产设备":
|
standingBook.setDeviceType(5);
|
break;
|
default:
|
standingBook.setDeviceType(4);
|
break;
|
}
|
standingBook.setNumber(dto.getNumber());
|
standingBook.setBrand(dto.getBrand());
|
standingBook.setConfidentiality(dto.getConfidentiality());
|
switch (dto.getStatus()) {
|
case "完好":
|
standingBook.setStatus(1);
|
break;
|
case "需整改":
|
standingBook.setStatus(2);
|
break;
|
default:
|
standingBook.setStatus(3);
|
break;
|
}
|
standingBook.setPurpose(dto.getPurpose());
|
standingBook.setLocation(dto.getLocation());
|
standingBook.setUsed(dto.getUsed());
|
standingBook.setRemark(dto.getRemark());
|
standingBook.setCreateTime(LocalDateTime.now());
|
standingBook.setCreateBy(SecurityUtils.getUsername());
|
standingBookMapper.insert(standingBook);
|
result.setSuccessCount(result.getSuccessCount() + 1);
|
} catch (Exception e) {
|
result.getErrorMessages().add("行" + context.readRowHolder().getRowIndex() + "导入失败：" + e.getMessage());
|
result.setFailCount(result.getFailCount() + 1);
|
}
|
}
|
|
|
|
|
@Override
|
public void doAfterAllAnalysed(AnalysisContext context) {
|
// 导入完成后的处理
|
}
|
};
|
|
// 创建读取器
|
ExcelReader excelReader = EasyExcel.read(file.getInputStream(), StandingBookImportDTO.class, listener).build();
|
ReadSheet readSheet = EasyExcel.readSheet(0).build();
|
excelReader.read(readSheet);
|
|
} catch (Exception e) {
|
result.getErrorMessages().add("导入失败：" + e.getMessage());
|
}
|
|
return result;
|
}
|
|
|
|
|
}
|
