package org.dromara.knowledge.service;

import com.baomidou.mybatisplus.extension.service.IService;
import org.dromara.common.mybatis.core.page.PageQuery;
import org.dromara.common.mybatis.core.page.TableDataInfo;
import org.dromara.knowledge.domain.KnowledgeJob;
import org.dromara.knowledge.domain.bo.KnowledgeJobBo;
import org.dromara.knowledge.domain.vo.KnowledgeJobVo;

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

/**
 * 知识库任务Service接口
 *
 * @author pengles
 */
public interface KnowledgeJobService extends IService<KnowledgeJob> {

    /**
     * 查询知识库任务
     */
    KnowledgeJobVo queryById(Long id);

    /**
     * 查询知识库任务列表
     */
    TableDataInfo<KnowledgeJobVo> queryPageList(KnowledgeJobBo bo, PageQuery pageQuery);

    /**
     * 查询知识库任务列表
     */
    List<KnowledgeJobVo> queryList(KnowledgeJobBo bo);

    /**
     * 新增知识库任务
     */
    Boolean insertByBo(KnowledgeJobBo bo);

    /**
     * 修改知识库任务
     */
    Boolean updateByBo(KnowledgeJobBo bo);

    /**
     * 校验并批量删除知识库任务信息
     */
    Boolean deleteWithValidByIds(Collection<Long> ids, Boolean isValid);

    /**
     * 根据任务ID查询知识库任务
     */
    KnowledgeJobVo queryByJobId(String jobId);

    /**
     * 根据知识库ID查询知识库任务列表
     */
    List<KnowledgeJobVo> queryByKnowledgeId(String knowledgeId);

    /**
     * 根据文档ID查询知识库任务列表
     */
    List<KnowledgeJobVo> queryByDocId(String docId);

    /**
     * 根据用户ID查询知识库任务列表
     */
    List<KnowledgeJobVo> queryByUserId(Long userId);

    /**
     * 根据任务类型查询知识库任务列表
     */
    List<KnowledgeJobVo> queryByJobType(String jobType);

    /**
     * 根据任务状态查询知识库任务列表
     */
    List<KnowledgeJobVo> queryByStatus(Integer status);

    /**
     * 查询正在运行的任务列表
     */
    List<KnowledgeJobVo> queryRunningJobs();

    /**
     * 查询待处理的任务列表
     */
    List<KnowledgeJobVo> queryPendingJobs();

    /**
     * 查询失败的任务列表
     */
    List<KnowledgeJobVo> queryFailedJobs();

    /**
     * 更新任务进度
     */
    Boolean updateProgress(Long id, Integer progress);

    /**
     * 更新任务状态
     */
    Boolean updateStatus(Long id, Integer status);

    /**
     * 增加重试次数
     */
    Boolean incrementRetryCount(Long id);

    /**
     * 更新任务项目计数
     */
    Boolean updateItemCounts(Long id, Integer totalItems, Integer processedItems, Integer successItems, Integer failedItems);

    /**
     * 根据优先级查询任务列表
     */
    List<KnowledgeJobVo> queryByPriority(Integer priority);

    /**
     * 查询超时的任务列表
     */
    List<KnowledgeJobVo> queryTimeoutJobs();

}