feat: 从长辉移植新高缺失的六个业务模块（安全生产/能耗/数采/备件/质检台账/售后问题回访）

xiaoyi

4 天以前 b0978a62cf2050170793e87fbd01db1760808043

 yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/dal/mysql/pro/task/MesProTaskMapper.java

@@ -8,16 +8,19 @@
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.route.MesProRouteProcessDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.task.MesProTaskDO;
import cn.iocoder.yudao.module.mes.enums.pro.MesProTaskStatusEnum;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import org.apache.ibatis.annotations.Mapper;

import java.math.BigDecimal;
import java.time.LocalDateTime;
import java.util.List;
import java.util.Map;

/**
 * MES 生产任务 Mapper
 *
 * @author 芋道源码
 * @author 超级管理员
 */
@Mapper
public interface MesProTaskMapper extends BaseMapperX<MesProTaskDO> {
@@ -49,7 +52,9 @@
        MPJLambdaWrapperX<MesProTaskDO> query = new MPJLambdaWrapperX<>();
        query.selectAll(MesProTaskDO.class);
        // 待报工台账：进行中的任务（排除已完成/已取消），且仍有待报工量
        query.eq(MesProTaskDO::getStatus, MesProTaskStatusEnum.PREPARE.getStatus());
        query.in(MesProTaskDO::getStatus, MesProTaskStatusEnum.PREPARE.getStatus(),
                MesProTaskStatusEnum.PRODUCING.getStatus(), MesProTaskStatusEnum.PARTIAL_FINISHED.getStatus(),
                MesProTaskStatusEnum.PAUSED.getStatus());
        query.apply("(quantity - IFNULL(produced_quantity, 0)) > 0");
        query.likeIfPresent(MesProTaskDO::getCode, reqVO.getCode())
                .likeIfPresent(MesProTaskDO::getName, reqVO.getName())
@@ -62,16 +67,54 @@
        return selectPage(reqVO, query);
    }

    /**
     * 按工作站统计待报工任务数，口径与 {@link #selectPendingFeedbackPage} 一致
     *
     * @return 每行包含 workstation_id、pending_count
     */
    default List<Map<String, Object>> selectPendingFeedbackCountByWorkstation() {
        return selectMaps(new QueryWrapper<MesProTaskDO>()
                .select("workstation_id", "COUNT(*) AS pending_count")
                .in("status", MesProTaskStatusEnum.PREPARE.getStatus(),
                        MesProTaskStatusEnum.PRODUCING.getStatus(), MesProTaskStatusEnum.PARTIAL_FINISHED.getStatus(),
                        MesProTaskStatusEnum.PAUSED.getStatus())
                .apply("(quantity - IFNULL(produced_quantity, 0)) > 0")
                .isNotNull("workstation_id")
                .groupBy("workstation_id"));
    }

    default List<MesProTaskDO> selectListByWorkOrderId(Long workOrderId) {
        return selectList(new LambdaQueryWrapperX<MesProTaskDO>()
                .eqIfPresent(MesProTaskDO::getWorkOrderId, workOrderId)
                .orderByDesc(MesProTaskDO::getId));
    }

    default MesProTaskDO selectByCode(String code) {
        return selectOne(MesProTaskDO::getCode, code);
    }

    default List<MesProTaskDO> selectListByWorkOrderIds(java.util.Collection<Long> workOrderIds) {
        return selectList(new LambdaQueryWrapperX<MesProTaskDO>()
                .in(MesProTaskDO::getWorkOrderId, workOrderIds)
                .orderByDesc(MesProTaskDO::getId));
    }

    /**
     * 班组报工工时统计：按班组聚合任务数与生产时长（duration 人日）
     *
     * @param startTime 生产开始时间下限
     * @param endTime   生产开始时间上限
     */
    default List<Map<String, Object>> selectWorkHourSummary(LocalDateTime startTime, LocalDateTime endTime) {
        return selectMaps(new QueryWrapper<MesProTaskDO>()
                .select("team_id", "team_name",
                        "COUNT(*) AS task_count",
                        "SUM(IFNULL(duration, 0)) AS total_duration")
                // 报工口径：排除草稿/已取消，仅统计实际进入生产的任务
                .notIn("status", MesProTaskStatusEnum.PREPARE.getStatus(), MesProTaskStatusEnum.CANCELED.getStatus())
                .ge(startTime != null, "start_time", startTime)
                .le(endTime != null, "start_time", endTime)
                .groupBy("team_id", "team_name"));
    }

    default void updateProducedQuantity(Long id,
@@ -82,7 +125,27 @@
                .eq(MesProTaskDO::getId, id)
                .setSql("produced_quantity = IFNULL(produced_quantity, 0) + " + incrProducedQuantity)
                .setSql("qualify_quantity = IFNULL(qualify_quantity, 0) + " + incrQualifyQuantity)
                .setSql("unqualify_quantity = IFNULL(unqualify_quantity, 0) + " + incrUnqualifyQuantity));
                .setSql("unqualify_quantity = IFNULL(unqualify_quantity, 0) + " + incrUnqualifyQuantity)
                .setSql("status = CASE WHEN IFNULL(produced_quantity, 0) > 0 THEN "
                        + MesProTaskStatusEnum.PARTIAL_FINISHED.getStatus() + " ELSE "
                        + MesProTaskStatusEnum.PRODUCING.getStatus() + " END")
                .setSql("first_feedback_time = IFNULL(first_feedback_time, NOW())")
                .setSql("last_feedback_time = NOW()"));
    }

    /**
     * 查询工作站上正在运行的任务：生产中、部分完成、已停工
     *
     * <p>数采模块根据设备上报的数据定位当前任务时使用。</p>
     */
    default MesProTaskDO selectRunningTaskByWorkstationId(Long workstationId) {
        return selectOne(new LambdaQueryWrapperX<MesProTaskDO>()
                .eq(MesProTaskDO::getWorkstationId, workstationId)
                .in(MesProTaskDO::getStatus, MesProTaskStatusEnum.PRODUCING.getStatus(),
                        MesProTaskStatusEnum.PARTIAL_FINISHED.getStatus(), MesProTaskStatusEnum.PAUSED.getStatus())
                .orderByDesc(MesProTaskDO::getLastFeedbackTime)
                .orderByDesc(MesProTaskDO::getId)
                .last("LIMIT 1"));
    }

}