package com.gkhy.safePlatform.specialWork.repository;

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.gkhy.safePlatform.specialWork.entity.WorkInfo;
import com.gkhy.safePlatform.specialWork.entity.WorkInfoDO;
import com.gkhy.safePlatform.specialWork.enums.WorkingAbortStatusEnum;
import com.gkhy.safePlatform.specialWork.enums.WorkingStatusEnum;
import com.gkhy.safePlatform.specialWork.model.query.db.*;
import org.apache.ibatis.annotations.Param;
import org.springframework.stereotype.Repository;

import java.time.LocalDateTime;
import java.util.List;

@Repository
public interface WorkInfoRepository extends BaseMapper<WorkInfo> {

    int insertWorkInfo(WorkInfo workInfoEntity);

    int updateWorkInfoById(WorkInfo workInfoEntity);

    List<WorkInfoDO> listWorkInfoDOByPage(Page<WorkInfoDO> page,@Param("query") WorkPageDBQuery dbQuery);

    List<WorkInfoDO> listWorkInfoDOMyRefByPage(Page<WorkInfoDO> page, @Param("query")WorkMyRefPageDBQuery dbQuery);

    long countWorkInfoMyRef(@Param("query")WorkMyRefPageDBQuery dbQuery);

    long countWorkInfo(@Param("query")WorkPageDBQuery dbQuery);

    int updateWorkStatusByWorkId(Long workId, Byte workStatus);

    int updateWorkAnalysisStatusByWorkId(Long workId, Byte workAnalysisStatus);

    List<WorkInfoDO> listWorkInfoGuardianByPage(Page<WorkInfo> page, @Param("query") WorkGuardianPageDBQuery dbQuery);

    List<WorkInfoDO> listWorkInfoAcceptByPage(Page<WorkInfo> page, @Param("query") WorkAcceptPageDBQuery dbQuery);

    List<WorkInfoDO> listWorkInfoAnalysisByPage(Page<WorkInfo> page,@Param("query") WorkAnalysisPageDBQuery dbQuery);

    /**
     * 统计接口 - 指定作业类型，部门列表，时间区间，统计作业数量
     * @param dbQuery
     * @return
     */
    Integer countByOptions(WorkCountDbQuery dbQuery);

    Integer countAllWorkByDep(@Param("depId") Long depId,@Param("startTime") LocalDateTime startTime,@Param("endTime") LocalDateTime endTime);


    int updateWorkStatusAndAbortStatusByWorkId(Long workId, WorkingStatusEnum workStatus, WorkingAbortStatusEnum abortStatus);
}