package com.gk.hotwork.Service.ServiceImpl; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.gk.hotwork.Domain.RiskMap; import com.gk.hotwork.Domain.UserInfo; import com.gk.hotwork.Domain.Vo.RiskMapVO; import com.gk.hotwork.Mapper.RiskMapRepository; import com.gk.hotwork.Service.RiskMapService; import com.gk.hotwork.Service.UserService; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import java.util.ArrayList; import java.util.Date; import java.util.List; @Service("RiskMapService") public class RiskMapServiceImpl extends ServiceImpl implements RiskMapService { @Autowired private UserService userService; @Override public List listByStatus(Byte status) { List riskMapList = baseMapper.listByStatus(status); if(riskMapList == null || riskMapList.size() == 0) return null; List voList = new ArrayList<>(); for(RiskMap map : riskMapList){ RiskMapVO vo = new RiskMapVO(); BeanUtils.copyProperties(map,vo); voList.add(vo); } return voList; } @Override public String saveOne(RiskMapVO riskMapVO) { if(checkRiskMapVO(riskMapVO)){ //判断已有的图片数量 List riskMapList = baseMapper.listByStatus((byte)0); if(riskMapList != null && riskMapList.size() >= 20){ return "图片数量超过上限"; } RiskMap riskMap = new RiskMap(); BeanUtils.copyProperties(riskMapVO,riskMap); if(save(riskMap)){ return null; } } return "参数错误"; } @Override public boolean deleteOne(Long id,Long uid) { RiskMap riskMap = getById(id); if(riskMap != null && riskMap.getId() > 0){ if(riskMap.getStatus() == 0){ riskMap.setStatus((byte)1); return updateById(riskMap); } } return false; } private boolean checkRiskMapVO(RiskMapVO vo){ if(vo.getCreateUid() == null || vo.getCreateUid() <= 0) return false; if(vo.getImg() == null || vo.getImg().isEmpty()) return false; UserInfo userInfo = userService.getById(vo.getCreateUid()); if(userInfo.getType() != 2) return false; vo.setCreateUname(userInfo.getRealname()); vo.setEditUid(userInfo.getId()); vo.setEditUname(userInfo.getRealname()); vo.setCreateTime(new Date()); vo.setEditTime(vo.getCreateTime()); vo.setStatus((byte)0); if(userInfo == null || userInfo.getId() <= 0 || userInfo.getCompanyid() == null || userInfo.getCompanyid() <= 0) return false; return true; } }