package com.ruoyi.safety.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.ruoyi.safety.mapper.SafetyAssessmentTaskMapper;
import com.ruoyi.safety.pojo.SafetyAssessmentTask;
import com.ruoyi.safety.service.SafetyAssessmentTaskService;
import org.springframework.stereotype.Service;

@Service
public class SafetyAssessmentTaskServiceImpl
        extends SafetyBaseServiceImpl<SafetyAssessmentTaskMapper, SafetyAssessmentTask>
        implements SafetyAssessmentTaskService {

    @Override
    public IPage<SafetyAssessmentTask> queryPage(Page<SafetyAssessmentTask> page, SafetyAssessmentTask query) {
        QueryWrapper<SafetyAssessmentTask> wrapper =
                new QueryWrapper<SafetyAssessmentTask>().orderByDesc("create_time");
        if (query != null) {
            if (hasText(query.getTitle())) {
                wrapper.like("title", query.getTitle());
            }
            if (query.getDeadline() != null) {
                wrapper.eq("deadline", query.getDeadline());
            }
            if (query.getStatus() != null) {
                wrapper.eq("status", query.getStatus());
            }
        }
        return page(page, wrapper);
    }
}