fuliqi
2024-11-01 b91945ea32378a71d1e57356af6e16703a1ceaa7
ycl-server/src/main/java/com/ycl/platform/service/impl/ImageResourceSecurityServiceImpl.java
@@ -3,6 +3,7 @@
import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.read.listener.PageReadListener;
import com.ycl.handler.ImageResourceHandler;
import com.ycl.platform.domain.dto.ImageResourceDTO;
import com.ycl.platform.domain.entity.CheckIndexVideo;
import com.ycl.platform.domain.entity.ImageResourceSecurity;
import com.ycl.platform.domain.entity.ImageResourceSecurityDetail;
@@ -45,11 +46,20 @@
    @Override
    public List<ImageResourceSecurity> selectImageResourceSecurityList(ImageResourceSecurity imageResourceSecurity) {
        List<ImageResourceSecurity> list = imageResourceSecurityMapper.selectImageResourceSecurityList(imageResourceSecurity);
        Map<Long, CheckIndexVideo> map = checkIndexVideoMapper.selectLastOneByDept();
        list.forEach(item -> item.setImageResourceSecurity(
                Optional.ofNullable(map.get(item.getDeptId()))
                        .map(CheckIndexVideo::getImageResourceSecurity)
                        .orElse(BigDecimal.ZERO)));
        List<ImageResourceDTO> dtos = checkIndexVideoMapper.selectLastOneByDept();
        for (ImageResourceSecurity item : list) {
            Long deptId = item.getDeptId();
            boolean flag = false;
            for (ImageResourceDTO dto : dtos) {
                if(dto.getDeptId().equals(deptId)){
                    item.setImageResourceSecurity(dto.getImageResourceSecurity());
                    flag = true;
                }
                if(!flag){
                    item.setImageResourceSecurity(BigDecimal.ZERO);
                }
            }
        }
        return list;
    }