package org.dromara.demo.service.impl; import cn.hutool.core.date.DateUtil; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import lombok.RequiredArgsConstructor; import org.dromara.common.core.utils.MapstructUtils; import org.dromara.common.core.utils.StringUtils; import org.dromara.common.mybatis.core.page.PageQuery; import org.dromara.common.mybatis.core.page.TableDataInfo; import org.dromara.demo.domain.RsDangerTotal; import org.dromara.demo.domain.bo.RsDangerTotalBo; import org.dromara.demo.domain.vo.RsDangerTotalVo; import org.dromara.demo.mapper.RsDangerTotalMapper; import org.dromara.demo.service.IRsDangerTotalService; import org.springframework.stereotype.Service; import java.time.LocalDate; import java.time.LocalDateTime; import java.time.LocalTime; import java.util.Collection; import java.util.Date; import java.util.List; import java.util.Map; /** * 安全隐患统计Service业务层处理 * * @author gonghl * @date 2024-02-29 */ @RequiredArgsConstructor @Service public class RsDangerTotalServiceImpl implements IRsDangerTotalService { private final RsDangerTotalMapper baseMapper; /** * 查询安全隐患统计 */ @Override public RsDangerTotalVo queryById(String id){ return baseMapper.selectVoById(id); } /** * 查询安全隐患统计列表 */ @Override public TableDataInfo queryPageList(RsDangerTotalBo bo, PageQuery pageQuery) { LambdaQueryWrapper lqw = buildQueryWrapper(bo); Page result = baseMapper.selectVoPage(pageQuery.build(), lqw); return TableDataInfo.build(result); } /** * 查询安全隐患统计列表 */ @Override public List queryList(RsDangerTotalBo bo) { LambdaQueryWrapper lqw = buildQueryWrapper(bo); return baseMapper.selectVoList(lqw); } private LambdaQueryWrapper buildQueryWrapper(RsDangerTotalBo bo) { Map params = bo.getParams(); LambdaQueryWrapper lqw = Wrappers.lambdaQuery(); lqw.like(StringUtils.isNotBlank(bo.getTownStreetName()), RsDangerTotal::getTownStreetName, bo.getTownStreetName()); lqw.eq(bo.getPeriod() != null, RsDangerTotal::getPeriod, bo.getPeriod()); lqw.eq(StringUtils.isNotBlank(bo.getPeriodDate()), RsDangerTotal::getPeriodDate, bo.getPeriodDate()); lqw.eq(bo.getFindSecurityRisks() != null, RsDangerTotal::getFindSecurityRisks, bo.getFindSecurityRisks()); lqw.eq(bo.getIssueNotice() != null, RsDangerTotal::getIssueNotice, bo.getIssueNotice()); lqw.eq(bo.getDealWithSecurityRisks() != null, RsDangerTotal::getDealWithSecurityRisks, bo.getDealWithSecurityRisks()); lqw.eq(bo.getStatus() != null, RsDangerTotal::getStatus, bo.getStatus()); lqw.orderByDesc(RsDangerTotal::getCreateTime); return lqw; } /** * 新增安全隐患统计 */ @Override public Boolean insertByBo(RsDangerTotalBo bo) { RsDangerTotal add = MapstructUtils.convert(bo, RsDangerTotal.class); validEntityBeforeSave(add); boolean flag = baseMapper.insert(add) > 0; if (flag) { bo.setId(add.getId()); } return flag; } /** * 修改安全隐患统计 */ @Override public Boolean updateByBo(RsDangerTotalBo bo) { RsDangerTotal update = MapstructUtils.convert(bo, RsDangerTotal.class); return baseMapper.updateById(update) > 0; } /** * 保存前的数据校验 */ private void validEntityBeforeSave(RsDangerTotal entity){ // 创建时间 if (entity.getCreateTime() == null) { entity.setCreateTime(new Date()); } } /** * 批量删除安全隐患统计 */ @Override public Boolean deleteWithValidByIds(Collection ids, Boolean isValid) { if(isValid){ //TODO 做一些业务上的校验,判断是否需要校验 } return baseMapper.deleteBatchIds(ids) > 0; } @Override public String importExcel(List list) { for (RsDangerTotal dangerTotal : list) { dangerTotal.setPeriod(dangerTotal.getPeriodDate().length() > 4 ? 1L : 2L); dangerTotal.setCreateTime(new Date()); dangerTotal.setStatus(2L); } return baseMapper.insertBatch(list) ? "恭喜您,数据已全部导入成功!共" + list.size() + "条!" : "很抱歉,导入失败!请检查数据格式!"; } }