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.SafetyTrainingRecordMapper;
import com.ruoyi.safety.pojo.SafetyTrainingRecord;
import com.ruoyi.safety.service.SafetyTrainingRecordService;
import org.springframework.stereotype.Service;

@Service
public class SafetyTrainingRecordServiceImpl
        extends SafetyBaseServiceImpl<SafetyTrainingRecordMapper, SafetyTrainingRecord>
        implements SafetyTrainingRecordService {

    @Override
    public IPage<SafetyTrainingRecord> queryPage(Page<SafetyTrainingRecord> page, SafetyTrainingRecord query) {
        QueryWrapper<SafetyTrainingRecord> wrapper =
                new QueryWrapper<SafetyTrainingRecord>().orderByDesc("create_time");
        if (query != null) {
            if (query.getPlanId() != null) {
                wrapper.eq("plan_id", query.getPlanId());
            }
            if (query.getEmployeeId() != null) {
                wrapper.eq("employee_id", query.getEmployeeId());
            }
            if (hasText(query.getEmployeeName())) {
                wrapper.like("employee_name", query.getEmployeeName());
            }
            if (query.getStatus() != null) {
                wrapper.eq("status", query.getStatus());
            }
        }
        return page(page, wrapper);
    }
}