forked from gzzfw/backEnd/gz-dyh

zhouxiantao
2024-09-14 6f4ed9dbb428374e5513fbf88aae1377fd423a51
dyh-service/dyh-sys/src/main/java/cn/huge/module/file/service/FileInfoService.java
@@ -475,4 +475,49 @@
        }
    }
    /**
     * 根据多个所属编号查询附件
     * @param mainId
     * @return List
     */
    public List<FileInfoBaseDTO> listByMainId(String mainId){
        List<FileInfoBaseDTO> fileList = new ArrayList<>();
        Map<String, Object> terms = new HashMap<>();
        terms.put("mainId", mainId);
        List<FileInfo> fileInfoList = mapper.listFile(terms);
        if(CollectionUtils.isNotEmpty(fileInfoList)){
            for (FileInfo fileInfo : fileInfoList) {
                FileInfoBaseDTO fileInfoBaseDTO = new FileInfoBaseDTO();
                BeanUtils.copyProperties(fileInfo, fileInfoBaseDTO);
                fileInfoBaseDTO.setOwnerCatName(FileOwnerTypeBaseEnum.getDes(fileInfo.getOwnerCat()));
                fileInfoBaseDTO.setOwnerTypeName(FileOwnerTypeBaseEnum.getDes(fileInfo.getOwnerType()));
                fileList.add(fileInfoBaseDTO);
            }
        }
        return fileList;
    }
    /**
     * 根据多个所属编号查询附件
     * @param mainId,ownerType
     * @return List
     */
    public List<FileInfoBaseDTO> listByMainIdAndType(String mainId,String ownerType){
        List<FileInfoBaseDTO> fileList = new ArrayList<>();
        Map<String, Object> terms = new HashMap<>();
        terms.put("mainId", mainId);
        terms.put("types", ownerType);
        List<FileInfo> fileInfoList = mapper.listFile(terms);
        if(CollectionUtils.isNotEmpty(fileInfoList)){
            for (FileInfo fileInfo : fileInfoList) {
                FileInfoBaseDTO fileInfoBaseDTO = new FileInfoBaseDTO();
                BeanUtils.copyProperties(fileInfo, fileInfoBaseDTO);
                fileInfoBaseDTO.setOwnerCatName(FileOwnerTypeBaseEnum.getDes(fileInfo.getOwnerCat()));
                fileInfoBaseDTO.setOwnerTypeName(FileOwnerTypeBaseEnum.getDes(fileInfo.getOwnerType()));
                fileList.add(fileInfoBaseDTO);
            }
        }
        return fileList;
    }
}