云
17 小时以前 ac2e2b5af3726ee64dc37c3ebf26989a0346f22a
yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/dal/mysql/pro/task/MesProTaskMapper.java
@@ -24,6 +24,15 @@
 */
@Mapper
public interface MesProTaskMapper extends BaseMapperX<MesProTaskDO> {
    @org.apache.ibatis.annotations.Select("SELECT id FROM mes_md_workstation WHERE deleted = 0 ORDER BY id FOR UPDATE")
    List<Long> lockSchedulingResources();
    default List<MesProTaskDO> selectSchedulingTasks() {
        return selectList(new LambdaQueryWrapperX<MesProTaskDO>()
                .and(query -> query.isNull(MesProTaskDO::getStatus).or()
                        .notIn(MesProTaskDO::getStatus, MesProTaskStatusEnum.CANCELED.getStatus(), MesProTaskStatusEnum.FINISHED.getStatus()))
                .orderByAsc(MesProTaskDO::getId));
    }
    default PageResult<MesProTaskDO> selectPage(MesProTaskPageReqVO reqVO) {
        MPJLambdaWrapperX<MesProTaskDO> query = new MPJLambdaWrapperX<>();
@@ -45,6 +54,7 @@
                .inIfPresent(MesProTaskDO::getStatus, reqVO.getStatuses())
                .betweenIfPresent(MesProTaskDO::getCreateTime, reqVO.getCreateTime())
                .orderByDesc(MesProTaskDO::getId);
        applyAssigneeFilter(query, reqVO.getAssigneeUserId());
        return selectPage(reqVO, query);
    }
@@ -64,7 +74,18 @@
                .eqIfPresent(MesProTaskDO::getWorkstationId, reqVO.getWorkstationId())
                .betweenIfPresent(MesProTaskDO::getCreateTime, reqVO.getCreateTime())
                .orderByDesc(MesProTaskDO::getId);
        applyAssigneeFilter(query, reqVO.getAssigneeUserId());
        return selectPage(reqVO, query);
    }
    /**
     * 限制只看当前用户是报工人的任务;未配置报工人的任务不参与匹配
     */
    default void applyAssigneeFilter(MPJLambdaWrapperX<MesProTaskDO> query, Long userId) {
        if (userId == null) {
            return;
        }
        query.apply("FIND_IN_SET({0}, assignee_user_ids)", userId);
    }
    /**
@@ -72,13 +93,14 @@
     *
     * @return 每行包含 workstation_id、pending_count
     */
    default List<Map<String, Object>> selectPendingFeedbackCountByWorkstation() {
    default List<Map<String, Object>> selectPendingFeedbackCountByWorkstation(Long assigneeUserId) {
        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")
                .apply(assigneeUserId != null, "FIND_IN_SET({0}, assignee_user_ids)", assigneeUserId)
                .isNotNull("workstation_id")
                .groupBy("workstation_id"));
    }