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.SafetyRiskMapper;
import com.ruoyi.safety.pojo.SafetyRisk;
import com.ruoyi.safety.service.SafetyRiskService;
import org.springframework.stereotype.Service;

@Service
public class SafetyRiskServiceImpl extends SafetyBaseServiceImpl<SafetyRiskMapper, SafetyRisk>
        implements SafetyRiskService {

    @Override
    public IPage<SafetyRisk> queryPage(Page<SafetyRisk> page, SafetyRisk query) {
        QueryWrapper<SafetyRisk> wrapper = new QueryWrapper<SafetyRisk>().orderByDesc("create_time");
        if (query != null) {
            if (hasText(query.getType())) {
                wrapper.like("type", query.getType());
            }
            if (hasText(query.getLevel())) {
                wrapper.eq("level", query.getLevel());
            }
            if (hasText(query.getPost())) {
                wrapper.like("post", query.getPost());
            }
        }
        return page(page, wrapper);
    }
}