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.SafetyFacilityDto;
|
import com.ruoyi.device.mapper.SafetyFacilityMapper;
|
import com.ruoyi.device.pojo.SafetyFacility;
|
import com.ruoyi.device.service.ISafetyFacilityService;
|
import lombok.RequiredArgsConstructor;
|
import org.springframework.stereotype.Service;
|
|
import jakarta.servlet.http.HttpServletResponse;
|
import java.util.List;
|
|
/**
|
* 安全设施Service实现类
|
*/
|
@Service
|
@RequiredArgsConstructor
|
public class SafetyFacilityServiceImpl extends ServiceImpl<SafetyFacilityMapper, SafetyFacility> implements ISafetyFacilityService {
|
|
private final SafetyFacilityMapper safetyFacilityMapper;
|
|
/**
|
* 分页查询安全设施列表
|
*/
|
@Override
|
public IPage<SafetyFacilityDto> queryPage(Page<SafetyFacility> page, SafetyFacilityDto dto) {
|
List<SafetyFacilityDto> list = safetyFacilityMapper.selectSafetyFacilityList(dto);
|
Page<SafetyFacilityDto> resultPage = new Page<>(page.getCurrent(), page.getSize(), list.size());
|
resultPage.setRecords(list);
|
return resultPage;
|
}
|
|
/**
|
* 新增或修改安全设施
|
*/
|
@Override
|
public int addOrEdit(SafetyFacilityDto dto) {
|
// 如果是新增，设置初始状态
|
if (dto.getId() == null) {
|
dto.setStatus("正常");
|
}
|
return safetyFacilityMapper.insertOrUpdate(dto) ? 1 : 0;
|
}
|
|
/**
|
* 导出安全设施记录
|
*/
|
@Override
|
public void export(HttpServletResponse response, SafetyFacilityDto dto) {
|
List<SafetyFacilityDto> list = safetyFacilityMapper.selectSafetyFacilityList(dto);
|
ExcelUtil<SafetyFacilityDto> util = new ExcelUtil<>(SafetyFacilityDto.class);
|
util.exportExcel(response, list, "安全设施台账");
|
}
|
}
|
