package com.yuanchu.mom.service.impl;
|
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.deepoove.poi.XWPFTemplate;
|
import com.deepoove.poi.config.Configure;
|
import com.deepoove.poi.config.ConfigureBuilder;
|
import com.deepoove.poi.data.Pictures;
|
import com.yuanchu.mom.common.GetLook;
|
import com.yuanchu.mom.dto.PersonPersonnelCapacityDto;
|
import com.yuanchu.mom.dto.PersonPersonnelCapacityExportDto;
|
import com.yuanchu.mom.dto.PersonSupervisionControlSheetExportDto;
|
import com.yuanchu.mom.exception.ErrorException;
|
import com.yuanchu.mom.mapper.UserMapper;
|
import com.yuanchu.mom.pojo.PersonPersonnelCapacity;
|
import com.yuanchu.mom.mapper.PersonPersonnelCapacityMapper;
|
import com.yuanchu.mom.service.PersonPersonnelCapacityService;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.yuanchu.mom.utils.DateImageUtil;
|
import org.springframework.beans.factory.annotation.Value;
|
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;
|
|
/**
|
* <p>
|
* 人员能力 服务实现类
|
* </p>
|
*
|
* @author
|
* @since 2024-10-10 11:26:18
|
*/
|
@Service
|
public class PersonPersonnelCapacityServiceImpl extends ServiceImpl<PersonPersonnelCapacityMapper, PersonPersonnelCapacity> implements PersonPersonnelCapacityService {
|
|
@Resource
|
private UserMapper userMapper;
|
|
@Resource
|
private GetLook getLook;
|
|
@Value("${file.path}")
|
private String imgUrl;
|
|
@Override
|
public IPage<PersonPersonnelCapacityDto> personPersonnelCapacityPage(Page page, Integer departLimsId, Integer userId, String userName) {
|
return baseMapper.personPersonnelCapacityPage(page, departLimsId, userId, userName);
|
}
|
|
/**
|
* 导出人员能力
|
* @param id
|
* @param response
|
*/
|
@Override
|
public void exportPersonnelCapacity(Integer id, HttpServletResponse response) {
|
PersonPersonnelCapacityExportDto capacityExportDto = baseMapper.selectExportPersonnelCapacity(id);
|
|
// 确认人
|
String confirmUrl = null;
|
if (capacityExportDto.getConfirmOperatingPersonnelId() != null) {
|
confirmUrl = userMapper.selectById(capacityExportDto.getConfirmOperatingPersonnelId()).getSignatureUrl();
|
if (StringUtils.isBlank(confirmUrl)) {
|
throw new ErrorException("确认人");
|
}
|
}
|
|
// 获取路径
|
InputStream inputStream = this.getClass().getResourceAsStream("/static/personnel-capacity.docx");
|
ConfigureBuilder builder = Configure.builder();
|
builder.useSpringEL(true);
|
String finalConfirmUrl = confirmUrl;
|
XWPFTemplate template = XWPFTemplate.compile(inputStream, builder.build()).render(
|
new HashMap<String, Object>() {{
|
put("capacity", capacityExportDto);
|
put("confirmUrl", StringUtils.isNotBlank(finalConfirmUrl) ? Pictures.ofLocal(imgUrl + "/" + finalConfirmUrl).create() : null);
|
put("confirmDateUrl", capacityExportDto.getConfirmDate() != null ?
|
Pictures.ofStream(DateImageUtil.createDateImage(capacityExportDto.getConfirmDate())).create() : null);
|
}});
|
|
try {
|
response.setContentType("application/msword");
|
String fileName = URLEncoder.encode(
|
"人员能力", "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("导出失败");
|
}
|
}
|
|
/**
|
* 人员能力确认
|
* @param id
|
* @return
|
*/
|
@Override
|
public boolean confirmPersonnelCapacity(Integer id) {
|
// 查询人员能力
|
PersonPersonnelCapacity personPersonnelCapacity = baseMapper.selectById(id);
|
if (personPersonnelCapacity.getConfirmOperatingPersonnelId() != null) {
|
throw new ErrorException("该人员能力已确认过, 无需重复确认");
|
}
|
Integer userId = getLook.selectPowerByMethodAndUserId(null).get("userId");
|
baseMapper.update(null, Wrappers.<PersonPersonnelCapacity>lambdaUpdate()
|
.eq(PersonPersonnelCapacity::getId, id)
|
.set(PersonPersonnelCapacity::getConfirmOperatingPersonnelId, userId)
|
.set(PersonPersonnelCapacity::getConfirmDate, LocalDateTime.now()));
|
|
return true;
|
}
|
}
|
