package com.nanjing.water.repository.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.nanjing.water.common.PagerResult;
import com.nanjing.water.common.enums.EYesOrNo;
import com.nanjing.water.common.util.ListUtil;
import com.nanjing.water.common.util.NumericUtil;
import com.nanjing.water.common.util.StringUtil;
import com.nanjing.water.entity.dto.OrderByDTO;
import com.nanjing.water.entity.dto.SearchBasicDTO;
import com.nanjing.water.entity.enums.EOrderBy;
import com.nanjing.water.entity.search.SearchQuartzTaskInfo;
import com.nanjing.water.repository.BasicMapperImpl;
import com.nanjing.water.repository.mapper.QuartzTaskInfoMapper;
import com.nanjing.water.repository.po.QuartzTaskInfoPO;
import org.springframework.stereotype.Repository;

import java.util.List;

/**
 * 定时任务信息表
 *
 * @author lin.liu
 * @description 定时任务信息表
 */
@Repository
public class QuartzTaskInfoMapperImpl extends BasicMapperImpl<QuartzTaskInfoPO, QuartzTaskInfoMapper> {
    public QuartzTaskInfoMapperImpl(QuartzTaskInfoMapper mapper) {
        super(mapper);
    }

    /**
     * 分页查询
     *
     * @param request 请求参数
     */
    @Override
    public PagerResult<QuartzTaskInfoPO> search(SearchBasicDTO request) {
        // 查询条件类型还原
        SearchQuartzTaskInfo search = (SearchQuartzTaskInfo) request;

        LambdaQueryWrapper<QuartzTaskInfoPO> queryWrapper = this.query();
        // 状态非逻辑删除
        queryWrapper.eq(QuartzTaskInfoPO::getIsDelete, EYesOrNo.NO.getValue());
        // 状态
        queryWrapper.eq(NumericUtil.tryParseInt(search.getStatus()).compareTo(0) > 0, QuartzTaskInfoPO::getStatus, search.getStatus());
        // 状态列表
        queryWrapper.in(ListUtil.isNotNullOrEmpty(search.getListStatus()), QuartzTaskInfoPO::getStatus, search.getListStatus());
        // 任务编号
        queryWrapper.eq(StringUtil.isNotNullOrEmpty(search.getTaskNo()), QuartzTaskInfoPO::getTaskNo, search.getTaskNo());
        // 数据创建时间-起始
        queryWrapper.ge(NumericUtil.tryParseLong(search.getCreateTimeStart()).compareTo(0L) > 0, QuartzTaskInfoPO::getCreateTime, search.getCreateTimeStart());
        // 数据创建时间-截止
        queryWrapper.le(NumericUtil.tryParseLong(search.getCreateTimeEnd()).compareTo(0L) > 0, QuartzTaskInfoPO::getCreateTime, search.getCreateTimeEnd());
        // 关键字
        if (StringUtil.isNotNullOrEmpty(search.getKeywords())) {
            queryWrapper.like(QuartzTaskInfoPO::getTaskName, search.getKeywords());
        }
        // 排序处理
        if (ListUtil.isNotNullOrEmpty(search.getOrderBy())) {
            for (OrderByDTO item : search.getOrderBy()) {
                EOrderBy orderBy = EOrderBy.getByValue(item.getOrderBy());
                // 顺序排序
                if (item.getIsAsc()) {
                    switch (orderBy) {
                        // 主键
                        case ID:
                            queryWrapper.orderByAsc(QuartzTaskInfoPO::getId);
                            break;
                        // 数据创建时间
                        case CREATE_TIME:
                            queryWrapper.orderByAsc(QuartzTaskInfoPO::getCreateTime);
                            break;
                        // 最后更新时间
                        case UPDATE_TIME:
                            queryWrapper.orderByAsc(QuartzTaskInfoPO::getUpdateTime);
                            break;
                    }
                } else {
                    // 倒叙排序
                    switch (orderBy) {
                        // 主键
                        case ID:
                            queryWrapper.orderByAsc(QuartzTaskInfoPO::getId);
                            break;
                        // 数据创建时间
                        case CREATE_TIME:
                            queryWrapper.orderByAsc(QuartzTaskInfoPO::getCreateTime);
                            break;
                        // 最后更新时间
                        case UPDATE_TIME:
                            queryWrapper.orderByAsc(QuartzTaskInfoPO::getUpdateTime);
                            break;
                    }
                }
            }
        } else {
            queryWrapper.orderByDesc(QuartzTaskInfoPO::getId);
        }
        Page<QuartzTaskInfoPO> pageResult = DB.selectPage(new Page<>(search.getPage(), search.getLimit()), queryWrapper);
        return new PagerResult<>(pageResult.getSize(), pageResult.getCurrent(), pageResult.getTotal(), pageResult.getRecords());
    }

    public List<QuartzTaskInfoPO> list4Status(Integer status) {
        LambdaQueryWrapper<QuartzTaskInfoPO> queryWrapper = this.query();
        // 非逻辑删除
        queryWrapper.eq(QuartzTaskInfoPO::getIsDelete, EYesOrNo.NO.getValue());
        // 状态
        queryWrapper.eq(QuartzTaskInfoPO::getStatus, status);
        return DB.selectList(queryWrapper);
    }

    public QuartzTaskInfoPO get4No(String taskNo) {
        LambdaQueryWrapper<QuartzTaskInfoPO> queryWrapper = this.query();
        // 任务编号
        queryWrapper.eq(QuartzTaskInfoPO::getTaskNo, taskNo);
        return DB.selectOne(queryWrapper);
    }
}