package com.fzzy.sys.manager.security;

import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.fzzy.igds.data.IgdsBaseParam;
import com.fzzy.igds.domain.PatrolRecord;
import com.fzzy.igds.service.PatrolRecordService;
import com.fzzy.igds.utils.ContextUtil;
import com.ruoyi.common.utils.StringUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.List;

@Slf4j
@Component
public class PatrolRecordManager {

    @Resource
    private PatrolRecordService patrolRecordService;

    /**
     * 分页查询数据
     *
     * @param param
     * @return
     */
    public Page<PatrolRecord> pageData(IgdsBaseParam param) {
        if (StringUtils.isEmpty(param.getCompanyId())) {
            param.setCompanyId(ContextUtil.getCompanyId());
        }

        Page<PatrolRecord> corePage = new Page<>(param.getPage(), param.getLimit());
        patrolRecordService.listPage(corePage, param);

        if (null == corePage.getRecords() || corePage.getRecords().isEmpty()) {
            return corePage.setRecords(new ArrayList<>());
        }

        return corePage;
    }

    /**
     * 获取所有数据
     *
     * @param param
     * @return
     */
    public List<PatrolRecord> listAll(IgdsBaseParam param) {
        if (StringUtils.isEmpty(param.getCompanyId())) {
            param.setCompanyId(ContextUtil.getCompanyId());
        }

        return patrolRecordService.listAll(param);
    }
}