yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/wm/stocktaking/task/MesWmStockTakingTaskServiceImpl.java
@@ -2,6 +2,7 @@
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjUtil;
import cn.hutool.core.util.StrUtil;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.mes.controller.admin.wm.stocktaking.task.vo.MesWmStockTakingTaskPageReqVO;
@@ -20,15 +21,11 @@
import cn.iocoder.yudao.module.mes.service.wm.stocktaking.plan.MesWmStockTakingPlanService;
import cn.iocoder.yudao.module.mes.service.wm.transaction.MesWmTransactionService;
import cn.iocoder.yudao.module.mes.service.wm.transaction.dto.MesWmTransactionSaveReqDTO;
import cn.iocoder.yudao.module.system.api.approval.ApprovalConfigApi;
import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
import cn.iocoder.yudao.module.bpm.api.task.BpmProcessInstanceApi;
import cn.iocoder.yudao.module.bpm.api.task.dto.BpmProcessInstanceCreateReqDTO;
import cn.iocoder.yudao.module.bpm.dal.dataobject.definition.BpmProcessDefinitionInfoDO;
import cn.iocoder.yudao.module.bpm.enums.task.BpmTaskStatusEnum;
import cn.iocoder.yudao.module.bpm.service.definition.BpmProcessDefinitionService;
import cn.iocoder.yudao.module.system.api.user.dto.AdminUserRespDTO;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.flowable.engine.repository.ProcessDefinition;
import org.springframework.context.annotation.Lazy;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder;
@@ -38,11 +35,14 @@
import java.math.BigDecimal;
import java.time.LocalDateTime;
import java.util.*;
import java.util.List;
import java.util.Map;
import java.util.Set;
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertList;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertMap;
import static cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils.getLoginUserId;
import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.*;
/**
@@ -55,13 +55,15 @@
@Slf4j
public class MesWmStockTakingTaskServiceImpl implements MesWmStockTakingTaskService {
    @Resource
    private MesWmStockTakingTaskMapper stockTakingTaskMapper;
    /**
     * 盘点结果审批的业务类型编码
     *
     * 与 system_approval_config.biz_type 对应,审核人在「系统管理 - 审批配置」中维护。
     */
    private static final String STOCK_TAKING_RESULT_APPROVE_BIZ_TYPE = "stock_taking_result_approve";
    @Resource
    private BpmProcessInstanceApi processInstanceApi;
    @Resource
    private BpmProcessDefinitionService bpmProcessDefinitionService;
    private MesWmStockTakingTaskMapper stockTakingTaskMapper;
    @Resource
    private MesWmMaterialStockService materialStockService;
@@ -82,6 +84,9 @@
    @Resource
    private AdminUserApi adminUserApi;
    @Resource
    private ApprovalConfigApi approvalConfigApi;
    @Override
    @Transactional(rollbackFor = Exception.class)
@@ -186,9 +191,8 @@
            throw exception(WM_STOCK_TAKING_TASK_NO_LINE);
        }
        // 2. 更新任务状态为审批中
        stockTakingTaskMapper.updateById(new MesWmStockTakingTaskDO().setId(id)
                .setStatus(MesWmStockTakingTaskStatusEnum.APPROVING.getStatus()));
        // 2. 更新任务状态为盘点中,并清空审核标记
        stockTakingTaskMapper.submitAndResetAuditInfo(id, MesWmStockTakingTaskStatusEnum.APPROVING.getStatus());
        // 3. 根据冻结标识,对物资进行冻结
        if (Boolean.TRUE.equals(task.getFrozen())) {
@@ -198,11 +202,11 @@
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void submitStockTakingResultApproval(Long id, String processDefinitionKey) {
    public void submitStockTakingResultApproval(Long id) {
        // 1.1 校验任务存在且处于盘点中状态
        MesWmStockTakingTaskDO task = validateStockTakingTaskExistsAndApproving(id);
        // 1.2 校验结果审批未提交过(避免重复提交)
        if (task.getProcessInstanceId() != null) {
        // 1.2 校验本轮结果审批未提交过(避免重复提交)
        if (isResultApprovalPending(task)) {
            throw exception(WM_STOCK_TAKING_TASK_ALREADY_SUBMIT_APPROVAL);
        }
        // 1.3 校验至少有一条盘点结果
@@ -210,94 +214,80 @@
        if (CollUtil.isEmpty(results)) {
            throw exception(WM_STOCK_TAKING_TASK_NO_RESULT);
        }
        // 1.4 校验审批配置已启用且配置了有效审核人(未配置时抛出带明确提示的业务异常)
        approvalConfigApi.validateApprovalEnabledAndGetApprovers(STOCK_TAKING_RESULT_APPROVE_BIZ_TYPE);
        // 2. 创建 BPM 流程实例
        String processInstanceId = processInstanceApi.createProcessInstance(getCurrentUserId(),
                new BpmProcessInstanceCreateReqDTO()
                        .setProcessDefinitionKey(processDefinitionKey)
                        .setBusinessKey(String.valueOf(id)));
        // 3. 回写流程实例编号(任务保持盘点中,等待审批)
        stockTakingTaskMapper.updateById(new MesWmStockTakingTaskDO()
                .setId(id).setProcessInstanceId(processInstanceId));
        // 2. 记录提交时间并清空上一轮审核结果(任务保持盘点中,等待审核)
        stockTakingTaskMapper.markResultApprovalSubmitted(id);
    }
    @Override
    public List<Map<String, Object>> getStockTakingTaskResultApprovalProcessDefinitionList() {
        return getProcessDefinitionListByCategory(APPROVE_CATEGORY_CODE);
    public Set<Long> getStockTakingTaskApproverUserIds() {
        return approvalConfigApi.getApproverUserIds(STOCK_TAKING_RESULT_APPROVE_BIZ_TYPE);
    }
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void updateStockTakingTaskResultAuditStatus(Long id, Integer bpmResult) {
    public void auditStockTakingResult(Long id, Boolean pass, String reviewRemark) {
        // 1. 校验任务存在
        MesWmStockTakingTaskDO task = validateStockTakingTaskExists(id);
        // 2. 校验处于盘点中(含结果审批中)状态
        if (!MesWmStockTakingTaskStatusEnum.APPROVING.getStatus().equals(task.getStatus())) {
            log.warn("[updateStockTakingTaskResultAuditStatus] 盘点任务({}) 不处于盘点中状态,忽略审批事件", id);
            return;
            throw exception(WM_STOCK_TAKING_TASK_NOT_APPROVING);
        }
        // 2. 审批通过:先置为已完成,再解冻并同事务自动核销差异
        if (BpmTaskStatusEnum.APPROVE.getStatus().equals(bpmResult)) {
            // 2.1 先置为已完成:任务状态变更后,后续重复的审批事件读取到 FINISHED 即被忽略,避免重复核销
            stockTakingTaskMapper.updateById(new MesWmStockTakingTaskDO().setId(id)
                    .setStatus(MesWmStockTakingTaskStatusEnum.FINISHED.getStatus()));
            // 2.2 解冻库存:须在差异核销前解冻,否则核销事务会被库存冻结校验拦截
            if (Boolean.TRUE.equals(task.getFrozen())) {
                updateMaterialStockFrozen(stockTakingTaskLineService.getStockTakingTaskLineListByTaskId(id), false);
        // 3. 校验本轮结果审批已提交且尚未审核
        if (!isResultApprovalPending(task)) {
            throw exception(WM_STOCK_TAKING_TASK_RESULT_NOT_SUBMITTED);
        }
        // 4. 校验当前登录用户是该业务类型的审核人(或签:任一人均可审核)
        Long userId = getLoginUserId();
        approvalConfigApi.validateApprover(STOCK_TAKING_RESULT_APPROVE_BIZ_TYPE, userId);
        // 5. 审核不通过:必须填写原因,任务保持盘点中,可修改实盘后重新提交结果审批
        if (!Boolean.TRUE.equals(pass)) {
            if (StrUtil.isBlank(reviewRemark)) {
                throw exception(WM_STOCK_TAKING_TASK_REJECT_REASON_REQUIRED);
            }
            // 2.3 自动核销盘点差异:盘盈入库 / 盘亏出库,同事务内更新库存台账;任一步失败整体回滚,任务回退为盘点中
            doProcessStockTakingDifference(task);
            stockTakingTaskMapper.auditStockTakingTask(id, MesWmStockTakingTaskStatusEnum.APPROVING.getStatus(),
                    userId, getUserNickname(userId), reviewRemark);
            return;
        }
        // 3. 审批驳回:清空流程实例编号,任务保持盘点中,可修改实盘后重新提交结果审批
        if (BpmTaskStatusEnum.REJECT.getStatus().equals(bpmResult)) {
            stockTakingTaskMapper.clearProcessInstanceId(id);
            return;
        // 6. 审核通过:先置为已完成并记录审核人与审核意见
        stockTakingTaskMapper.auditStockTakingTask(id, MesWmStockTakingTaskStatusEnum.FINISHED.getStatus(),
                userId, getUserNickname(userId), reviewRemark);
        // 7. 解冻库存:须在差异核销前解冻,否则核销事务会被库存冻结校验拦截
        if (Boolean.TRUE.equals(task.getFrozen())) {
            updateMaterialStockFrozen(stockTakingTaskLineService.getStockTakingTaskLineListByTaskId(id), false);
        }
        // 4. 审批取消:解冻库存并置为已取消
        if (BpmTaskStatusEnum.CANCEL.getStatus().equals(bpmResult)) {
            if (Boolean.TRUE.equals(task.getFrozen())) {
                updateMaterialStockFrozen(stockTakingTaskLineService.getStockTakingTaskLineListByTaskId(id), false);
            }
            stockTakingTaskMapper.clearProcessInstanceId(id);
            stockTakingTaskMapper.updateById(new MesWmStockTakingTaskDO().setId(id)
                    .setStatus(MesWmStockTakingTaskStatusEnum.CANCELED.getStatus()));
        }
        // 8. 自动核销盘点差异:盘盈入库 / 盘亏出库,同事务内更新库存台账;任一步失败整体回滚,任务回退为盘点中
        doProcessStockTakingDifference(task);
    }
    /**
     * 获取分类下的可用 BPM 流程定义(过滤挂起流程,同 Key 保留最新版本)
     * 判断本轮盘点结果审批是否已提交且尚未审核
     *
     * @param categoryCode 流程分类编码
     * @return 流程定义列表(id/key/name)
     * 任务状态 2(盘点中)同时承载「盘点中」与「盘点结果审批中」两个阶段,
     * 以 reviewTime 非空且 reviewerId 为空作为「已提交、等待审核」的判定条件;
     * 审核不通过后 reviewerId 非空,任务回到可重新提交结果审批的状态。
     *
     * @param task 盘点任务
     * @return 是否处于结果审批中
     */
    private List<Map<String, Object>> getProcessDefinitionListByCategory(String categoryCode) {
        List<BpmProcessDefinitionInfoDO> definitionInfoList = bpmProcessDefinitionService
                .getProcessDefinitionInfoListByCategory(categoryCode);
        if (CollUtil.isEmpty(definitionInfoList)) {
            return Collections.emptyList();
    private boolean isResultApprovalPending(MesWmStockTakingTaskDO task) {
        return task.getReviewerId() == null && task.getReviewTime() != null;
    }
    /**
     * 获取用户昵称
     *
     * @param userId 用户编号
     * @return 昵称,用户不存在时返回 null
     */
    private String getUserNickname(Long userId) {
        if (userId == null) {
            return null;
        }
        // 过滤激活状态,保留最新版本
        Map<String, ProcessDefinition> latestVersionMap = new HashMap<>();
        for (BpmProcessDefinitionInfoDO info : definitionInfoList) {
            ProcessDefinition pd = bpmProcessDefinitionService.getProcessDefinition(info.getProcessDefinitionId());
            if (pd == null || pd.isSuspended()) {
                continue;
            }
            ProcessDefinition existing = latestVersionMap.get(pd.getKey());
            if (existing == null || pd.getVersion() > existing.getVersion()) {
                latestVersionMap.put(pd.getKey(), pd);
            }
        }
        List<Map<String, Object>> result = new ArrayList<>();
        for (ProcessDefinition pd : latestVersionMap.values()) {
            Map<String, Object> item = new HashMap<>();
            item.put("id", pd.getId());
            item.put("key", pd.getKey());
            item.put("name", pd.getName());
            result.add(item);
        }
        return result;
        AdminUserRespDTO user = adminUserApi.getUser(userId);
        return user == null ? null : user.getNickname();
    }
    private Long getCurrentUserId() {
@@ -318,8 +308,8 @@
        }
        task.setStatus(MesWmStockTakingTaskStatusEnum.CANCELED.getStatus());
        stockTakingTaskMapper.updateById(task);
        // 取消后清空流程实例编号,避免残留审批标记影响后续重新开始盘点
        stockTakingTaskMapper.clearProcessInstanceId(id);
        // 取消后清空审核标记,避免残留审批标记影响后续重新开始盘点
        stockTakingTaskMapper.clearAuditInfo(id);
        if (Boolean.TRUE.equals(task.getFrozen())) {
            updateMaterialStockFrozen(stockTakingTaskLineService.getStockTakingTaskLineListByTaskId(id), false);
        }
@@ -384,7 +374,7 @@
    /**
     * 自动核销盘点差异:按盘点结果逐行生成 IN/OUT 库存事务更新库存台账
     *
     * <p>仅由 {@link #updateStockTakingTaskResultAuditStatus} 的审批通过分支在同事务内调用;
     * <p>仅由 {@link #auditStockTakingResult} 的审核通过分支在同事务内调用;
     * 失败整体回滚,任务回退为盘点中状态,避免重复调整。
     *
     * @param task 已置为已完成、已解冻的盘点任务