package com.yuanchu.mom.service.impl;
|
|
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.deepoove.poi.XWPFTemplate;
|
import com.deepoove.poi.config.Configure;
|
import com.yuanchu.mom.common.GetLook;
|
import com.yuanchu.mom.dto.DeviceTraceabilityManagementDetailsDto;
|
import com.yuanchu.mom.dto.DeviceTraceabilityManagementDto;
|
import com.yuanchu.mom.mapper.DeviceTraceabilityManagementDetailsMapper;
|
import com.yuanchu.mom.mapper.DeviceTraceabilityManagementMapper;
|
import com.yuanchu.mom.mapper.UserMapper;
|
import com.yuanchu.mom.pojo.DeviceTraceabilityManagement;
|
import com.yuanchu.mom.pojo.DeviceTraceabilityManagementDetails;
|
import com.yuanchu.mom.pojo.User;
|
import com.yuanchu.mom.service.DeviceTraceabilityManagementDetailsService;
|
import com.yuanchu.mom.service.DeviceTraceabilityManagementService;
|
import com.yuanchu.mom.utils.HackLoopTableRenderPolicy;
|
import com.yuanchu.mom.utils.UserUtils;
|
import com.yuanchu.mom.utils.QueryWrappers;
|
import com.yuanchu.mom.vo.Result;
|
import org.apache.commons.collections4.CollectionUtils;
|
import org.springframework.beans.BeanUtils;
|
import org.springframework.stereotype.Service;
|
|
import javax.annotation.Resource;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.InputStream;
|
import java.io.OutputStream;
|
import java.net.URLEncoder;
|
import java.time.LocalDateTime;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
/**
|
* <p>
|
* 设备量值溯源计划表 服务实现类
|
* </p>
|
*
|
* @author 江苏鵷雏网络科技有限公司
|
* @since 2024-12-20 02:27:44
|
*/
|
@Service
|
public class DeviceTraceabilityManagementServiceImpl extends ServiceImpl<DeviceTraceabilityManagementMapper, DeviceTraceabilityManagement> implements DeviceTraceabilityManagementService {
|
|
@Resource
|
private DeviceTraceabilityManagementDetailsService deviceTraceabilityManagementDetailsService;
|
|
@Resource
|
private DeviceTraceabilityManagementDetailsMapper deviceTraceabilityManagementDetailsMapper;
|
|
@Resource
|
private GetLook getLook;
|
|
@Resource
|
private UserMapper userMapper;
|
|
/**
|
* 分页查询设备量值溯源计划
|
*
|
* @param page
|
* @return
|
*/
|
@Override
|
public Result<IPage<DeviceTraceabilityManagement>> selectDeviceTraceabilityManagementByPage(IPage page, DeviceTraceabilityManagementDto deviceTraceabilityManagementDto) {
|
IPage<DeviceTraceabilityManagement> iPage = baseMapper.selectDeviceParameterPage(page, QueryWrappers.queryWrappers(deviceTraceabilityManagementDto));
|
return Result.success(iPage);
|
}
|
|
|
/**
|
* 新增设备量值溯源计划
|
*
|
* @param deviceTraceabilityManagementDto 设备量值溯源计划
|
*/
|
@Override
|
public Result addTraceabilityManagement(DeviceTraceabilityManagementDto deviceTraceabilityManagementDto) {
|
Integer userId = getLook.selectPowerByMethodAndUserId(null).get("userId");
|
User user = userMapper.selectById(userId);
|
deviceTraceabilityManagementDto.setCompilerId(userId);
|
deviceTraceabilityManagementDto.setCompiler(user.getName());
|
deviceTraceabilityManagementDto.setDatePreparation(LocalDateTime.now());
|
|
// 查询审核人id
|
if (deviceTraceabilityManagementDto.getAuditId() != null) {
|
User auditUser = userMapper.selectById(deviceTraceabilityManagementDto.getAuditId());
|
deviceTraceabilityManagementDto.setAudit(auditUser.getName());
|
}
|
this.saveOrUpdate(deviceTraceabilityManagementDto);
|
|
// 详情赋值并保存
|
List<DeviceTraceabilityManagementDetailsDto> deviceTraceabilityManagementDetails = deviceTraceabilityManagementDto.getDeviceTraceabilityManagementDetails();
|
if (CollectionUtils.isNotEmpty(deviceTraceabilityManagementDetails)) { // 详情不为空
|
List<DeviceTraceabilityManagementDetails> collect = deviceTraceabilityManagementDetails.stream().map(detailsDto -> {
|
detailsDto.setTraceabilityManagementId(deviceTraceabilityManagementDto.getTraceabilityManagementId()); // 量值溯源计划ID
|
DeviceTraceabilityManagementDetails details = new DeviceTraceabilityManagementDetails();
|
BeanUtils.copyProperties(detailsDto, details);
|
return details;
|
}).collect(Collectors.toList());
|
deviceTraceabilityManagementDetailsService.saveBatch(collect);
|
}
|
|
return Result.success();
|
}
|
|
/**
|
* 修改设备量值溯源计划
|
*
|
* @param deviceTraceabilityManagementDto 设备量值溯源计划
|
*/
|
@Override
|
public Result updateTraceabilityManagement(DeviceTraceabilityManagementDto deviceTraceabilityManagementDto) {
|
// 查询审核人id
|
if (deviceTraceabilityManagementDto.getAuditId() != null) {
|
User auditUser = userMapper.selectById(deviceTraceabilityManagementDto.getAuditId());
|
deviceTraceabilityManagementDto.setAudit(auditUser.getName());
|
}
|
this.saveOrUpdate(deviceTraceabilityManagementDto);
|
|
// 删除原本的详情
|
deviceTraceabilityManagementDetailsService.remove(Wrappers.<DeviceTraceabilityManagementDetails>lambdaQuery().eq(DeviceTraceabilityManagementDetails::getTraceabilityManagementId, deviceTraceabilityManagementDto.getTraceabilityManagementId()));
|
// 详情赋值并保存
|
List<DeviceTraceabilityManagementDetailsDto> deviceTraceabilityManagementDetails = deviceTraceabilityManagementDto.getDeviceTraceabilityManagementDetails();
|
if (CollectionUtils.isNotEmpty(deviceTraceabilityManagementDetails)) { // 详情不为空
|
List<DeviceTraceabilityManagementDetails> collect = deviceTraceabilityManagementDetails.stream().map(detailsDto -> {
|
detailsDto.setTraceabilityManagementId(deviceTraceabilityManagementDto.getTraceabilityManagementId()); // 量值溯源计划ID
|
DeviceTraceabilityManagementDetails details = new DeviceTraceabilityManagementDetails();
|
BeanUtils.copyProperties(detailsDto, details);
|
return details;
|
}).collect(Collectors.toList());
|
deviceTraceabilityManagementDetailsService.saveBatch(collect);
|
}
|
return Result.success();
|
}
|
|
/**
|
* 删除设备量值溯源计划
|
*
|
* @param deviceTraceabilityManagementDto 设备量值溯源计划
|
*/
|
@Override
|
public Result deleteTraceabilityManagement(DeviceTraceabilityManagementDto deviceTraceabilityManagementDto) {
|
this.removeById(deviceTraceabilityManagementDto);
|
deviceTraceabilityManagementDetailsService.remove(Wrappers.<DeviceTraceabilityManagementDetails>lambdaQuery().eq(DeviceTraceabilityManagementDetails::getTraceabilityManagementId, deviceTraceabilityManagementDto.getTraceabilityManagementId()));
|
return Result.success();
|
}
|
|
/**
|
* 查询设备量值溯源计划详情
|
*
|
* @param traceabilityManagementId 设备量值溯源计划id
|
*/
|
@Override
|
public Result<DeviceTraceabilityManagementDto> getTraceabilityManagementDetail(Integer traceabilityManagementId) {
|
// 查询设备量值溯源计划
|
DeviceTraceabilityManagement deviceTraceabilityManagement = baseMapper.selectById(traceabilityManagementId);
|
// 查询详情
|
DeviceTraceabilityManagementDto deviceTraceabilityManagementDto = new DeviceTraceabilityManagementDto();
|
BeanUtils.copyProperties(deviceTraceabilityManagement, deviceTraceabilityManagementDto);
|
deviceTraceabilityManagementDto.setDeviceTraceabilityManagementDetails(deviceTraceabilityManagementDetailsMapper.deviceTraceabilityManagementDetailsList(traceabilityManagementId));
|
return Result.success(deviceTraceabilityManagementDto);
|
}
|
|
/**
|
* 审核设备量值溯源计划
|
*
|
* @param deviceTraceabilityManagementDto 设备量值溯源计划
|
*/
|
@Override
|
public Result reviewTraceabilityManagementStatus(DeviceTraceabilityManagementDto deviceTraceabilityManagementDto) {
|
Integer userId = getLook.selectPowerByMethodAndUserId(null).get("userId");
|
User user = userMapper.selectById(userId);
|
|
LambdaUpdateWrapper<DeviceTraceabilityManagement> wrapper = Wrappers.<DeviceTraceabilityManagement>lambdaUpdate()
|
.eq(DeviceTraceabilityManagement::getTraceabilityManagementId, deviceTraceabilityManagementDto.getTraceabilityManagementId())
|
.set(DeviceTraceabilityManagement::getAuditDate, LocalDateTime.now())
|
.set(DeviceTraceabilityManagement::getAuditId, user.getId())
|
.set(DeviceTraceabilityManagement::getAudit, user.getName())
|
.set(DeviceTraceabilityManagement::getStatus, deviceTraceabilityManagementDto.getStatus()) // 审核状态
|
.set(DeviceTraceabilityManagement::getAuditRemark, deviceTraceabilityManagementDto.getAuditRemark());// 审核备注
|
|
// 为0清除审核人
|
if (deviceTraceabilityManagementDto.getStatus().equals(0)) {
|
wrapper.set(DeviceTraceabilityManagement::getAuditId, null)
|
.set(DeviceTraceabilityManagement::getAudit, null);
|
}
|
this.update(wrapper); // 更新
|
return Result.success();
|
}
|
|
/**
|
* 导出设备量值溯源计划
|
*
|
* @param traceabilityManagementId 设备量值溯源计划id
|
*/
|
@Override
|
public Result exportDeviceTraceabilityManagementDto(Integer traceabilityManagementId, HttpServletResponse response) {
|
// 查询设备保养计划
|
DeviceTraceabilityManagementDto deviceTraceabilityManagement = baseMapper.selectDeviceTraceabilityManagementById(traceabilityManagementId);
|
|
// 查询设备保养计划详情
|
List<DeviceTraceabilityManagementDetailsDto> deviceTraceabilityManagementDetailsDtoList = deviceTraceabilityManagementDetailsMapper.deviceTraceabilityManagementDetailsList(traceabilityManagementId);
|
// 设置序号
|
deviceTraceabilityManagementDetailsDtoList.forEach(deviceTraceabilityManagementDetails -> {
|
deviceTraceabilityManagementDetails.setIndex(deviceTraceabilityManagementDetailsDtoList.indexOf(deviceTraceabilityManagementDetails) + 1);
|
});
|
|
// 获取路径
|
InputStream inputStream = this.getClass().getResourceAsStream("/static/word/traceability-management-details.docx");
|
Configure configure = Configure.builder()
|
.bind("deviceTraceabilityManagementDetailsDtoList", new HackLoopTableRenderPolicy())
|
.build();
|
XWPFTemplate template = XWPFTemplate.compile(inputStream, configure).render(
|
new HashMap<String, Object>() {{
|
put("deviceTraceabilityManagement", deviceTraceabilityManagement);
|
put("deviceTraceabilityManagementDetailsDtoList", deviceTraceabilityManagementDetailsDtoList);
|
// 编制人签名地址
|
put("compilerUrl", UserUtils.getFinalUserSignatureUrl(deviceTraceabilityManagement.getCompilerId()));
|
// 审核人签名地址
|
put("auditUrl", UserUtils.getFinalUserSignatureUrl(deviceTraceabilityManagement.getAuditId()));
|
}});
|
|
try {
|
response.setContentType("application/msword");
|
String finalFileName = deviceTraceabilityManagement.getFileName() == null ? "" : deviceTraceabilityManagement.getFileName() + "_";
|
String fileName = URLEncoder.encode(
|
finalFileName+ "设备量值溯源计划", "UTF-8");
|
response.setHeader("Content-disposition",
|
"attachment;filename=" + fileName + ".docx");
|
OutputStream os = response.getOutputStream();
|
template.write(os);
|
os.flush();
|
os.close();
|
} catch (Exception e) {
|
e.printStackTrace();
|
throw new RuntimeException("导出失败");
|
}
|
return Result.success();
|
}
|
|
}
|
