diff --git a/src/main/java/com/gunshi/project/xyt/service/AssessTemplateService.java b/src/main/java/com/gunshi/project/xyt/service/AssessTemplateService.java index 8a39826..750224e 100644 --- a/src/main/java/com/gunshi/project/xyt/service/AssessTemplateService.java +++ b/src/main/java/com/gunshi/project/xyt/service/AssessTemplateService.java @@ -10,8 +10,10 @@ import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.gunshi.project.xyt.entity.dto.InspectItemDto; import com.gunshi.project.xyt.entity.so.AttCctvBasePage; +import com.gunshi.project.xyt.mapper.AssessTaskMapper; import com.gunshi.project.xyt.mapper.AssessTemplateMapper; import com.gunshi.project.xyt.model.AssessIndicator; +import com.gunshi.project.xyt.model.AssessTask; import com.gunshi.project.xyt.model.AssessTemplate; import lombok.extern.slf4j.Slf4j; import org.springframework.beans.factory.annotation.Autowired; @@ -20,7 +22,9 @@ import org.springframework.transaction.annotation.Transactional; import java.util.Date; import java.util.List; +import java.util.Map; import java.util.Objects; +import java.util.stream.Collectors; /** * 描述: 考核模板 @@ -38,6 +42,9 @@ public class AssessTemplateService extends ServiceImpl res = this.page(page.getPageSo().toPage(), query); + if (res.getRecords() != null && res.getRecords().size() > 0) { + fillUsedInfo(res.getRecords()); + } return res; } + private void fillUsedInfo(List records) { + List ids = records.stream().map(AssessTemplate::getId).collect(Collectors.toList()); + List list = taskMapper.selectList(new QueryWrapper().in("template_id", ids)); + Map map = list.stream().collect(Collectors.groupingBy(AssessTask::getTemplateId, Collectors.counting())); + for (AssessTemplate record : records) { + record.setIsUsed(map.containsKey(record.getId()) ? 1 : 0); + } + } + public String startStop(InspectItemDto dto) { Integer status = dto.getStatus(); AssessTemplate template = super.getById(dto.getId()); diff --git a/src/main/java/com/gunshi/project/xyt/service/MessageCenterService.java b/src/main/java/com/gunshi/project/xyt/service/MessageCenterService.java index 8969000..2388fef 100644 --- a/src/main/java/com/gunshi/project/xyt/service/MessageCenterService.java +++ b/src/main/java/com/gunshi/project/xyt/service/MessageCenterService.java @@ -12,6 +12,7 @@ import com.gunshi.project.xyt.mapper.AttResBaseMapper; import com.gunshi.project.xyt.mapper.MessageCenterMapper; import com.gunshi.project.xyt.model.AttResBase; import com.gunshi.project.xyt.model.MessageCenter; +import com.gunshi.project.xyt.model.OsmoticWarnR; import com.gunshi.project.xyt.model.StRsvrR; import com.gunshi.project.xyt.util.DateUtil; import lombok.extern.slf4j.Slf4j; @@ -136,6 +137,36 @@ public class MessageCenterService extends ServiceImpl queryPage(WarnPageSo warnPageSo) { return this.baseMapper.queryPage(warnPageSo.getPageSo().toPage(),warnPageSo); } @@ -158,6 +161,8 @@ public class OsmoticWarnRService extends ServiceImpl