package cn.iocoder.yudao.module.mes.dal.mysql.safety;

import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.iocoder.yudao.module.mes.controller.admin.safety.vo.risk.MesSafetyRiskPageReqVO;
import cn.iocoder.yudao.module.mes.dal.dataobject.safety.MesSafetyRiskDO;
import org.apache.ibatis.annotations.Mapper;

import java.util.Collection;
import java.util.List;

/**
 * 安全生产 - 风险事件清单 Mapper
 *
 * @author 超级管理员
 */
@Mapper
public interface MesSafetyRiskMapper extends BaseMapperX<MesSafetyRiskDO> {

    default PageResult<MesSafetyRiskDO> selectPage(MesSafetyRiskPageReqVO reqVO) {
        return selectPage(reqVO, new LambdaQueryWrapperX<MesSafetyRiskDO>()
                .likeIfPresent(MesSafetyRiskDO::getCode, reqVO.getCode())
                .likeIfPresent(MesSafetyRiskDO::getName, reqVO.getName())
                .eqIfPresent(MesSafetyRiskDO::getUnitId, reqVO.getUnitId())
                .eqIfPresent(MesSafetyRiskDO::getAccidentType, reqVO.getAccidentType())
                .eqIfPresent(MesSafetyRiskDO::getRiskLevel, reqVO.getRiskLevel())
                .eqIfPresent(MesSafetyRiskDO::getStatus, reqVO.getStatus())
                .orderByDesc(MesSafetyRiskDO::getRiskValue)
                .orderByDesc(MesSafetyRiskDO::getId));
    }

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

    default List<MesSafetyRiskDO> selectListByUnitId(Long unitId) {
        return selectList(new LambdaQueryWrapperX<MesSafetyRiskDO>()
                .eq(MesSafetyRiskDO::getUnitId, unitId)
                .orderByDesc(MesSafetyRiskDO::getRiskValue));
    }

    default Long selectCountByUnitIds(Collection<Long> unitIds) {
        return selectCount(new LambdaQueryWrapperX<MesSafetyRiskDO>()
                .in(MesSafetyRiskDO::getUnitId, unitIds));
    }

    default List<MesSafetyRiskDO> selectListByStatus(Integer status) {
        return selectList(new LambdaQueryWrapperX<MesSafetyRiskDO>()
                .eqIfPresent(MesSafetyRiskDO::getStatus, status)
                .orderByDesc(MesSafetyRiskDO::getRiskValue));
    }

    /**
     * 按风险等级统计数量（用于看板/统计）
     */
    default List<MesSafetyRiskDO> selectAllList() {
        return selectList(new LambdaQueryWrapperX<MesSafetyRiskDO>()
                .eq(MesSafetyRiskDO::getStatus, 1)
                .orderByDesc(MesSafetyRiskDO::getRiskValue));
    }

}