package cn.iocoder.yudao.module.hrm.service.employee.contract;
|
|
import cn.iocoder.yudao.module.hrm.dal.dataobject.employee.HrmEmployeeContractDO;
|
import cn.iocoder.yudao.module.hrm.dal.dataobject.employee.HrmEmployeeDO;
|
import cn.iocoder.yudao.module.hrm.dal.mysql.employee.HrmEmployeeContractMapper;
|
import cn.iocoder.yudao.module.hrm.dal.mysql.employee.HrmEmployeeMapper;
|
import com.deepoove.poi.XWPFTemplate;
|
import jakarta.annotation.Resource;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Service;
|
|
import java.io.ByteArrayOutputStream;
|
import java.io.InputStream;
|
import java.math.BigDecimal;
|
import java.time.LocalDate;
|
import java.time.format.DateTimeFormatter;
|
import java.time.temporal.ChronoUnit;
|
import java.util.HashMap;
|
import java.util.Map;
|
|
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
|
import static cn.iocoder.yudao.module.hrm.enums.ErrorCodeConstants.CONTRACT_TEMPLATE_LOAD_FAILED;
|
import static cn.iocoder.yudao.module.hrm.enums.ErrorCodeConstants.CONTRACT_TEMPLATE_TYPE_NOT_SUPPORT;
|
|
/**
|
* 员工合同文档生成 Service 实现类
|
*
|
* @author 超级管理员
|
*/
|
@Slf4j
|
@Service
|
public class ContractDocumentServiceImpl implements ContractDocumentService {
|
|
/** 劳动合同（简版·社保折现） */
|
public static final int TYPE_LABOR_SIMPLE = 1;
|
/** 劳动合同（标准·官方制式） */
|
public static final int TYPE_LABOR_STANDARD = 2;
|
/** 劳务合同 */
|
public static final int TYPE_SERVICE = 3;
|
|
private static final DateTimeFormatter DATE_FMT = DateTimeFormatter.ofPattern("yyyy年MM月dd日");
|
|
private static final Map<Integer, String> TEMPLATE_PATHS = Map.of(
|
TYPE_LABOR_SIMPLE, "template/contract/contract_labor_simple.docx",
|
TYPE_LABOR_STANDARD, "template/contract/contract_labor_standard.docx",
|
TYPE_SERVICE, "template/contract/contract_service.docx");
|
|
private static final Map<Integer, String> TEMPLATE_NAMES = Map.of(
|
TYPE_LABOR_SIMPLE, "劳动合同(简版)",
|
TYPE_LABOR_STANDARD, "劳动合同(标准)",
|
TYPE_SERVICE, "劳务合同");
|
|
@Resource
|
private HrmEmployeeContractMapper contractMapper;
|
@Resource
|
private HrmEmployeeMapper employeeMapper;
|
|
@Override
|
public byte[] generateDocx(Long contractId, Integer templateType) {
|
HrmEmployeeContractDO contract = contractMapper.selectById(contractId);
|
if (contract == null) {
|
throw exception(CONTRACT_TEMPLATE_LOAD_FAILED);
|
}
|
HrmEmployeeDO employee = contract.getEmployeeId() == null ? null
|
: employeeMapper.selectById(contract.getEmployeeId());
|
Map<String, Object> vars = buildVariables(contract, employee);
|
|
String path = TEMPLATE_PATHS.get(templateType);
|
if (path == null) {
|
throw exception(CONTRACT_TEMPLATE_TYPE_NOT_SUPPORT);
|
}
|
try (InputStream in = getClass().getClassLoader().getResourceAsStream(path)) {
|
if (in == null) {
|
throw exception(CONTRACT_TEMPLATE_LOAD_FAILED);
|
}
|
XWPFTemplate template = XWPFTemplate.compile(in).render(vars);
|
ByteArrayOutputStream out = new ByteArrayOutputStream();
|
template.write(out);
|
template.close();
|
return out.toByteArray();
|
} catch (Exception e) {
|
log.error("[generateDocx][合同({}) 模板({}) 生成失败]", contractId, templateType, e);
|
throw exception(CONTRACT_TEMPLATE_LOAD_FAILED);
|
}
|
}
|
|
@Override
|
public String buildFileName(Long contractId, Integer templateType) {
|
HrmEmployeeContractDO contract = contractMapper.selectById(contractId);
|
String no = contract != null && contract.getContractNo() != null ? contract.getContractNo() : String.valueOf(contractId);
|
String name = TEMPLATE_NAMES.getOrDefault(templateType, "合同");
|
return no + "-" + name + ".docx";
|
}
|
|
private Map<String, Object> buildVariables(HrmEmployeeContractDO contract, HrmEmployeeDO employee) {
|
Map<String, Object> vars = new HashMap<>();
|
// 甲方（公司）—— 目前取签约主体，其余留空后续可接入公司配置
|
vars.put("companyName", nvl(contract.getSignCompany()));
|
vars.put("companyAddress", "");
|
vars.put("legalPerson", "");
|
vars.put("companyPhone", "");
|
vars.put("creditCode", "");
|
// 乙方（员工）
|
vars.put("employeeName", employee != null ? nvl(employee.getName()) : "");
|
vars.put("gender", employee != null ? genderText(employee.getGender()) : "");
|
vars.put("age", employee != null ? nvl(employee.getAge()) : "");
|
vars.put("idCard", employee != null ? nvl(employee.getIdCard()) : "");
|
vars.put("householdAddress", employee != null ? nvl(employee.getAddress()) : "");
|
vars.put("address", employee != null ? nvl(employee.getAddress()) : "");
|
vars.put("phone", employee != null ? nvl(employee.getPhone()) : "");
|
vars.put("emergencyPhone", "");
|
// 合同
|
vars.put("contractNo", nvl(contract.getContractNo()));
|
vars.put("postName", "");
|
vars.put("postDuty", "");
|
vars.put("workPlace", "");
|
vars.put("contractStart", formatDate(contract.getStartDate()));
|
vars.put("contractEnd", formatDate(contract.getEndDate()));
|
vars.put("contractMonths", monthsBetween(contract.getStartDate(), contract.getEndDate()));
|
vars.put("probationStart", formatDate(contract.getProbationStartDate()));
|
vars.put("probationEnd", formatDate(contract.getProbationEndDate()));
|
vars.put("probationMonths", monthsBetween(contract.getProbationStartDate(), contract.getProbationEndDate()));
|
vars.put("probationDays", daysBetween(contract.getProbationStartDate(), contract.getProbationEndDate()));
|
vars.put("birthday", employee != null ? formatDate(employee.getBirthday()) : "");
|
// 年月日（供「___年___月___日」逐格填写）
|
vars.put("contractStartY", y(contract.getStartDate()));
|
vars.put("contractStartM", mo(contract.getStartDate()));
|
vars.put("contractStartD", d(contract.getStartDate()));
|
vars.put("contractEndY", y(contract.getEndDate()));
|
vars.put("contractEndM", mo(contract.getEndDate()));
|
vars.put("contractEndD", d(contract.getEndDate()));
|
vars.put("probationStartY", y(contract.getProbationStartDate()));
|
vars.put("probationStartM", mo(contract.getProbationStartDate()));
|
vars.put("probationStartD", d(contract.getProbationStartDate()));
|
vars.put("probationEndY", y(contract.getProbationEndDate()));
|
vars.put("probationEndM", mo(contract.getProbationEndDate()));
|
vars.put("probationEndD", d(contract.getProbationEndDate()));
|
vars.put("probationSalary", nvl(contract.getProbationSalary()));
|
vars.put("regularSalary", nvl(contract.getRegularSalary()));
|
vars.put("baseSalary", nvl(contract.getRegularSalary()));
|
vars.put("performanceSalary", "");
|
vars.put("socialSecuritySalary", "");
|
vars.put("signDate", formatDate(contract.getSignDate()));
|
// 劳务合同里的可选项
|
vars.put("payType", "一");
|
vars.put("payDay", "");
|
vars.put("workVolume", "");
|
vars.put("workAmount", "");
|
vars.put("piecePrice", "");
|
vars.put("noticeCompensation", "");
|
vars.put("copies", "两");
|
vars.put("eachCopies", "一");
|
return vars;
|
}
|
|
private static String nvl(Object o) {
|
return o == null ? "" : o.toString();
|
}
|
|
private static String formatDate(LocalDate d) {
|
return d == null ? "" : d.format(DATE_FMT);
|
}
|
|
private static String monthsBetween(LocalDate start, LocalDate end) {
|
if (start == null || end == null) {
|
return "";
|
}
|
long months = ChronoUnit.MONTHS.between(start, end);
|
return String.valueOf(Math.max(months, 0));
|
}
|
|
private static String daysBetween(LocalDate start, LocalDate end) {
|
if (start == null || end == null) {
|
return "";
|
}
|
long days = ChronoUnit.DAYS.between(start, end);
|
return String.valueOf(Math.max(days, 0));
|
}
|
|
private static String y(LocalDate date) {
|
return date == null ? "" : String.format("%04d", date.getYear());
|
}
|
|
private static String mo(LocalDate date) {
|
return date == null ? "" : String.format("%02d", date.getMonthValue());
|
}
|
|
private static String d(LocalDate date) {
|
return date == null ? "" : String.format("%02d", date.getDayOfMonth());
|
}
|
|
private static String genderText(Integer sex) {
|
if (sex == null) {
|
return "";
|
}
|
return sex == 1 ? "男" : sex == 2 ? "女" : "";
|
}
|
|
// 兼容 BigDecimal 直接 toString
|
private static String nvl(BigDecimal v) {
|
return v == null ? "" : v.toPlainString();
|
}
|
|
}
|
