package com.docus.server.infrastructure.dao.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.docus.infrastructure.core.db.dao.impl.BaseDaoImpl;
import com.docus.infrastructure.web.request.SearchDTO;
import com.docus.infrastructure.web.response.PageResult;
import com.docus.server.entity.segmentation.OcrRuleTest;
import com.docus.server.entity.segmentation.OcrSpecialRule;
import com.docus.server.infrastructure.dao.IOcrSpecialRuleDao;
import com.docus.server.infrastructure.mapper.OcrSpecialRuleMapper;
import com.github.pagehelper.PageHelper;
import org.springframework.stereotype.Repository;

import java.util.Date;
import java.util.List;

/**
 * 特殊策略表 数据访问实现类
 *
 * @author AutoGenerator
 * @since 2023-08-29
 */
@Repository
public class OcrSpecialRuleDaoImpl extends BaseDaoImpl<OcrSpecialRuleMapper, OcrSpecialRule> implements IOcrSpecialRuleDao {

    /**
     * 按主键查询
     *
     * @param id 主键Id
     * @return 实体
     */
    @Override
    public OcrSpecialRule findById(Long id) {
        return super.findById(id);
    }

    /**
     * 新增
     *
     * @param ocrSpecialRule 编辑参数
     * @return 成功或失败
     */
    @Override
    public boolean add(OcrSpecialRule ocrSpecialRule) {
        return super.save(ocrSpecialRule);
    }

    /**
     * 编辑
     *
     * @param ocrSpecialRule 编辑参数
     * @return 成功或失败
     */
    @Override
    public boolean edit(OcrSpecialRule ocrSpecialRule) {
        return super.updateById(ocrSpecialRule);
    }

    /**
     * 批量删除
     *
     * @param ids 主键ids
     * @return 成功或失败
     */
    @Override
    public int delete(List<Long> ids) {
        return super.deleteByIds(ids);
    }

    /**
     * 关键字搜索
     *
     * @param searchDTO 搜索参数
     * @return 分页列表
     */
    @Override
    public PageResult<OcrSpecialRule> search(SearchDTO searchDTO) {
        //分页
        PageHelper.startPage(searchDTO.getPageNum(), searchDTO.getPageSize());
        LambdaQueryWrapper<OcrSpecialRule> query = Wrappers.lambdaQuery();
        //时间范围，默认按createTime
        if (searchDTO.getBeginTime() != null) {
            query.ge(OcrSpecialRule::getCreateTime, searchDTO.getBeginTime());
        }
        if (searchDTO.getEndTime() != null) {
            Date endTime = searchDTO.getEndTime();
            query.le(OcrSpecialRule::getCreateTime, endTime);
        }

        Object version = searchDTO.getParams("version");
        query.eq(OcrSpecialRule::getVersion,version);

        //默认createTime倒序排序
        query.orderByDesc(OcrSpecialRule::getCreateTime);
        List<OcrSpecialRule> list = super.find(query);
        return new PageResult<>(list);
    }

    /**
     * 名称不重复
     *
     * @param id   主键
     * @param name 名称
     * @return 名称重复数量
     */
    @Override
    public int findByIdAndName(Long id, String name) {
        LambdaQueryWrapper<OcrSpecialRule> query = Wrappers.lambdaQuery();
        if (id != null) {
            query.ne(OcrSpecialRule::getId, id);
        }
        //if (Func.isNotBlank(name)) {
        //   query.eq(OcrSpecialRule::getName, name);
        //}

        return baseMapper.selectCount(query);
    }
}
