package com.ruoyi.device.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.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.device.dto.LineInspectionDto;
import com.ruoyi.device.mapper.LineInspectionMapper;
import com.ruoyi.device.pojo.LineInspection;
import com.ruoyi.device.service.ILineInspectionService;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;

import jakarta.servlet.http.HttpServletResponse;
import java.util.List;

/**
 * 线路巡检Service实现类
 */
@Service
@RequiredArgsConstructor
public class LineInspectionServiceImpl extends ServiceImpl<LineInspectionMapper, LineInspection> implements ILineInspectionService {

    private final LineInspectionMapper lineInspectionMapper;

    /**
     * 分页查询线路巡检列表
     */
    @Override
    public IPage<LineInspectionDto> queryPage(Page<LineInspection> page, LineInspectionDto dto) {
        List<LineInspectionDto> list = lineInspectionMapper.selectLineInspectionList(dto);
        Page<LineInspectionDto> resultPage = new Page<>(page.getCurrent(), page.getSize(), list.size());
        resultPage.setRecords(list);
        return resultPage;
    }

    /**
     * 新增或修改线路巡检记录
     */
    @Override
    public int addOrEdit(LineInspectionDto dto) {
        // 如果是新增，设置初始状态
        if (dto.getId() == null) {
            dto.setStatus("待巡检");
        }
        return lineInspectionMapper.insertOrUpdate(dto) ? 1 : 0;
    }

    /**
     * 导出线路巡检记录
     */
    @Override
    public void export(HttpServletResponse response, LineInspectionDto dto) {
        List<LineInspectionDto> list = lineInspectionMapper.selectLineInspectionList(dto);
        ExcelUtil<LineInspectionDto> util = new ExcelUtil<>(LineInspectionDto.class);
        util.exportExcel(response, list, "线路巡检记录");
    }
}