package com.qianwen.smartman.modules.dnc.service.impl;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.Optional;
|
import java.util.function.Function;
|
import java.util.stream.Collectors;
|
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.qianwen.core.log.exception.ServiceException;
|
import com.qianwen.core.mp.base.BaseServiceImpl;
|
import com.qianwen.core.mp.support.Condition;
|
import com.qianwen.core.mp.support.Query;
|
import com.qianwen.core.tool.utils.DateUtil;
|
import com.qianwen.core.tool.utils.Func;
|
import com.qianwen.smartman.common.constant.CommonConstant;
|
import com.qianwen.smartman.common.utils.MessageUtils;
|
import com.qianwen.smartman.modules.cps.entity.Workstation;
|
import com.qianwen.smartman.modules.cps.service.IWorkstationService;
|
import com.qianwen.smartman.modules.dnc.convert.DncArchiveDirectoryFileConvert;
|
import com.qianwen.smartman.modules.dnc.entity.DncArchiveDirectory;
|
import com.qianwen.smartman.modules.dnc.entity.DncArchiveDirectoryFile;
|
import com.qianwen.smartman.modules.dnc.entity.DncFactoryFile;
|
import com.qianwen.smartman.modules.dnc.entity.TransferDirectoryFile;
|
import com.qianwen.smartman.modules.dnc.enums.DncEnums;
|
import com.qianwen.smartman.modules.dnc.mapper.DncArchiveDirectoryFileMapper;
|
import com.qianwen.smartman.modules.dnc.service.IDncArchiveDirectoryFileService;
|
import com.qianwen.smartman.modules.dnc.service.IDncArchiveDirectoryService;
|
import com.qianwen.smartman.modules.dnc.service.IDncFactoryFileService;
|
import com.qianwen.smartman.modules.dnc.service.IFileService;
|
import com.qianwen.smartman.modules.dnc.service.IObsFileService;
|
import com.qianwen.smartman.modules.dnc.service.ITransferDirectoryFileService;
|
import com.qianwen.smartman.modules.dnc.vo.ArchiveDirectoryFileSearchVO;
|
import com.qianwen.smartman.modules.dnc.vo.DncArchiveDirectoryFileAddVO;
|
import com.qianwen.smartman.modules.dnc.vo.DncArchiveDirectoryFileVO;
|
import com.qianwen.smartman.modules.dnc.vo.DncArchiveFileAddVO;
|
import com.qianwen.smartman.modules.dnc.vo.VerifyArchiveFileCheckVO;
|
import com.qianwen.smartman.modules.dnc.vo.VerifyArchiveFileVO;
|
import com.qianwen.smartman.modules.system.entity.User;
|
import com.qianwen.smartman.modules.system.service.IUserService;
|
|
@Service
|
public class DncArchiveDirectoryFileServiceImpl extends BaseServiceImpl<DncArchiveDirectoryFileMapper, DncArchiveDirectoryFile> implements IDncArchiveDirectoryFileService {
|
private final IFileService fileService;
|
private final IUserService userService;
|
private final IWorkstationService workstationService;
|
private final IDncArchiveDirectoryService directoryService;
|
private final ITransferDirectoryFileService transferDirectoryFileService;
|
private final IDncFactoryFileService factoryFileService;
|
private final IObsFileService obsFileService;
|
|
public DncArchiveDirectoryFileServiceImpl(final IFileService fileService, final IUserService userService, final IWorkstationService workstationService, final IDncArchiveDirectoryService directoryService, final ITransferDirectoryFileService transferDirectoryFileService, final IDncFactoryFileService factoryFileService, final IObsFileService obsFileService) {
|
this.fileService = fileService;
|
this.userService = userService;
|
this.workstationService = workstationService;
|
this.directoryService = directoryService;
|
this.transferDirectoryFileService = transferDirectoryFileService;
|
this.factoryFileService = factoryFileService;
|
this.obsFileService = obsFileService;
|
}
|
|
@Override // org.springblade.modules.dnc.service.IDncArchiveDirectoryFileService
|
public IPage<DncArchiveDirectoryFileVO> getArFilePage(ArchiveDirectoryFileSearchVO searchVO, Query query) {
|
List<String> suffixs = this.fileService.getSuffix();
|
Integer fileType = searchVO.getFileType();
|
boolean result = false;
|
String noSuffix = "";
|
if (fileType != null && fileType.equals(DncEnums.FileType.PROCESSING_PROGRAM.getCode())) {
|
result = true;
|
|
List<String> collect = suffixs.stream().filter(CommonConstant.NOT_SUFFIX::equals).collect(Collectors.toList());
|
|
if (collect.size() > 0) {
|
noSuffix = CommonConstant.NOT_SUFFIX;
|
}
|
suffixs = suffixs.stream().filter(s -> {
|
return !CommonConstant.NOT_SUFFIX.equals(s);
|
}).collect(Collectors.toList());
|
}
|
String endTime = "";
|
if (Func.isNotEmpty(searchVO.getEndTime())) {
|
endTime = DateUtil.formatDate(searchVO.getEndTime()) + " 23:59:59";
|
}
|
Map<Long, User> userMap = this.userService.list().stream().collect(Collectors.toMap((v0) -> {
|
return v0.getId();
|
}, Function.identity()));
|
Map<Long, Workstation> workstationMap = this.workstationService.list(Wrappers.<Workstation>lambdaQuery().eq(Workstation::getStatus, CommonConstant.ENABLE)).stream().collect(Collectors.toMap((v0) -> {
|
return v0.getId();
|
}, Function.identity()));
|
List<String> finalSuffixs = suffixs;
|
String finalNoSuffix = noSuffix;
|
|
IPage<DncArchiveDirectoryFile> page = page(Condition.getPage(query), Wrappers.<DncArchiveDirectoryFile>lambdaQuery()
|
.eq(DncArchiveDirectoryFile::getArchiveDirectoryId, searchVO.getArId())
|
.eq(DncArchiveDirectoryFile::getIsCurrent, DncEnums.CurrentVersion.YES.getCode())
|
.and(result, i -> i.in(DncArchiveDirectoryFile::getSuffix, finalSuffixs).or().eq(Func.isNotBlank(finalNoSuffix), DncArchiveDirectoryFile::getSuffix, ""))
|
.eq(Func.isNotEmpty(searchVO.getWorkstationId()), DncArchiveDirectoryFile::getWorkstationId, searchVO.getWorkstationId())
|
.ge(Func.isNotEmpty(searchVO.getStartTime()), DncArchiveDirectoryFile::getCreateTime, searchVO.getStartTime())
|
.le(Func.isNotBlank(endTime), DncArchiveDirectoryFile::getCreateTime, endTime)
|
.and(Func.isNotEmpty(searchVO.getKeywords()), i -> i.likeRight(DncArchiveDirectoryFile::getFilename, searchVO.getKeywords()).or().likeRight(DncArchiveDirectoryFile::getSuffix, searchVO.getKeywords())));
|
|
/*
|
IPage<DncArchiveDirectoryFile> page = page(Condition.getPage(query), ((LambdaQueryWrapper) ((LambdaQueryWrapper) Wrappers.lambdaQuery().eq((v0) -> {
|
return v0.getArchiveDirectoryId();
|
}, searchVO.getArId())).eq((v0) -> {
|
return v0.getIsCurrent();
|
}, DncEnums.CurrentVersion.YES.getCode())).and(result, i -> {
|
((LambdaQueryWrapper) ((LambdaQueryWrapper) i.in((v0) -> {
|
return v0.getSuffix();
|
}, finalSuffixs)).or()).eq(Func.isNotBlank(finalNoSuffix), (v0) -> {
|
return v0.getSuffix();
|
}, "");
|
}).eq(Func.isNotEmpty(searchVO.getWorkstationId()), (v0) -> {
|
return v0.getWorkstationId();
|
}, searchVO.getWorkstationId()).ge(Func.isNotEmpty(searchVO.getStartTime()), (v0) -> {
|
return v0.getCreateTime();
|
}, searchVO.getStartTime()).le(Func.isNotBlank(endTime), (v0) -> {
|
return v0.getCreateTime();
|
}, endTime).and(Func.isNotEmpty(searchVO.getKeywords()), i2 -> {
|
LambdaQueryWrapper lambdaQueryWrapper = (LambdaQueryWrapper) ((LambdaQueryWrapper) ((LambdaQueryWrapper) i2.likeRight((v0) -> {
|
return v0.getFilename();
|
}, searchVO.getKeywords())).or()).likeRight((v0) -> {
|
return v0.getSuffix();
|
}, searchVO.getKeywords());
|
}));
|
*/
|
IPage<DncArchiveDirectoryFileVO> convert = DncArchiveDirectoryFileConvert.INSTANCE.convert(page);
|
convert.getRecords().forEach(s2 -> {
|
User userCr = (User) userMap.get(s2.getCreateUser());
|
User userUp = (User) userMap.get(s2.getUpdateUser());
|
Workstation workstation = (Workstation) workstationMap.get(s2.getWorkstationId());
|
s2.setUpdateUserName(userCr != null ? userCr.getAccount() : "");
|
s2.setCreateUserName(userUp != null ? userUp.getAccount() : "");
|
s2.setWorkstationName(workstation != null ? workstation.getName() : "");
|
s2.setVersionDesc((Func.isNotBlank(s2.getVersion()) ? s2.getVersion() : CommonConstant.VERSION_NUM) + CommonConstant.FILE_SUFFIX);
|
});
|
return Optional.of(convert).orElse(new Page<>());
|
}
|
|
@Override // org.springblade.modules.dnc.service.IDncArchiveDirectoryFileService
|
@Transactional(rollbackFor = {Exception.class})
|
public DncArchiveFileAddVO addFile(DncArchiveFileAddVO archiveFileAddVOS) {
|
List<DncArchiveDirectoryFileAddVO> fileAddVOList = archiveFileAddVOS.getFileAddVOList();
|
List<String> archiveDirectoryIds = archiveFileAddVOS.getArchiveDirectoryIds();
|
List<DncArchiveDirectoryFile> fileList = new ArrayList<>();
|
archiveDirectoryIds.forEach(archiveDirectoryId -> {
|
if (Func.isNotEmpty(archiveDirectoryId)) {
|
DncArchiveDirectory directory = (DncArchiveDirectory) this.directoryService.getById(archiveDirectoryId);
|
if (directory == null) {
|
throw new ServiceException(MessageUtils.message("dnc.archive.directory.is.not.exists", new Object[0]));
|
}
|
}
|
if (Func.isNotEmpty(fileAddVOList)) {
|
List<DncArchiveDirectoryFileAddVO> workstationList = fileAddVOList.stream().filter(s -> {
|
return DncEnums.FileOwnership.WORKSTATION.getCode().equals(s.getFileOwnership());
|
}).collect(Collectors.toList());
|
if (Func.isNotEmpty(workstationList)) {
|
List<Long> workstationFileIds = workstationList.stream().map((v0) -> {
|
return v0.getWorkstationFileId();
|
}).distinct().collect(Collectors.toList());
|
fileList.addAll(addFile(Long.valueOf(Long.parseLong(archiveDirectoryId)), workstationList));
|
if (Func.isNotEmpty(workstationFileIds)) {
|
this.transferDirectoryFileService.update(Wrappers.<TransferDirectoryFile>lambdaUpdate()
|
.set(TransferDirectoryFile::getWhetherArchive, DncEnums.WhetherArchive.not_Archive.getCode())
|
.in(TransferDirectoryFile::getId, workstationFileIds));
|
/*
|
this.transferDirectoryFileService.update((Wrapper) ((LambdaUpdateWrapper) Wrappers.lambdaUpdate().set((v0) -> {
|
return v0.getWhetherArchive();
|
}, DncEnums.WhetherArchive.not_Archive.getCode())).in((v0) -> {
|
return v0.getId();
|
}, workstationFileIds));*/
|
}
|
}
|
List<DncArchiveDirectoryFileAddVO> factoryList = fileAddVOList.stream().filter(s2 -> {
|
return DncEnums.FileOwnership.FACTORY.getCode().equals(s2.getFileOwnership());
|
}).collect(Collectors.toList());
|
if (Func.isNotEmpty(factoryList)) {
|
List<Long> factoryIds = factoryList.stream().map((v0) -> {
|
return v0.getWorkstationFileId();
|
}).distinct().collect(Collectors.toList());
|
fileList.addAll(addFile(Long.valueOf(Long.parseLong(archiveDirectoryId)), factoryList));
|
if (Func.isNotEmpty(factoryIds)) {
|
this.factoryFileService.update(Wrappers.<DncFactoryFile>lambdaUpdate().set(DncFactoryFile::getWhetherArchive, DncEnums.WhetherArchive.not_Archive.getCode()).in(DncFactoryFile::getId, factoryIds));
|
/*
|
this.factoryFileService.update((Wrapper) ((LambdaUpdateWrapper) Wrappers.lambdaUpdate().set((v0) -> {
|
return v0.getWhetherArchive();
|
}, DncEnums.WhetherArchive.not_Archive.getCode())).in((v0) -> {
|
return v0.getId();
|
}, factoryIds));*/
|
}
|
}
|
}
|
});
|
if (Func.isNotEmpty(fileList)) {
|
saveBatch(fileList);
|
}
|
return archiveFileAddVOS;
|
}
|
|
private List<DncArchiveDirectoryFile> addFile(Long archiveDirectoryId, List<DncArchiveDirectoryFileAddVO> list) {
|
return list.stream().map(addVO -> {
|
DncArchiveDirectoryFile addFile = new DncArchiveDirectoryFile();
|
Long workstationId = addVO.getWorkstationId();
|
String filename = addVO.getFilename();
|
DncArchiveDirectoryFile directoryFile = verifyFileName(filename, workstationId, archiveDirectoryId);
|
String contentMd5 = addVO.getContentMd5();
|
if (Func.isNull(directoryFile)) {
|
addFile.setVersion(CommonConstant.VERSION_NUM);
|
addFile.setOriginalFilename(addVO.getOriginalFilename()).setIsCurrent(DncEnums.CurrentVersion.YES.getCode()).setFilename(addVO.getFilename()).setContentLength(addVO.getContentLength()).setContentMd5(contentMd5).setContentType(addVO.getContentType()).setObjectKey(addVO.getObjectKey()).setSuffix(addVO.getSuffix()).setLink(addVO.getLink()).setWorkstationFileId(addVO.getWorkstationFileId()).setWorkstationId(addVO.getWorkstationId()).setArchiveDirectoryId(archiveDirectoryId);
|
} else if (!contentMd5.equals(directoryFile.getContentMd5())) {
|
directoryFile.setIsCurrent(DncEnums.CurrentVersion.NO.getCode());
|
updateById(directoryFile);
|
addFile.setVersion(Integer.valueOf((directoryFile.getVersion() != null ? directoryFile.getVersion() : CommonConstant.VERSION_NUM).intValue() + CommonConstant.VERSION_NUM.intValue()));
|
addFile.setOriginalFilename(addVO.getOriginalFilename()).setIsCurrent(DncEnums.CurrentVersion.YES.getCode()).setFilename(addVO.getFilename()).setContentLength(addVO.getContentLength()).setContentMd5(contentMd5).setContentType(addVO.getContentType()).setObjectKey(addVO.getObjectKey()).setSuffix(addVO.getSuffix()).setLink(addVO.getLink()).setWorkstationFileId(addVO.getWorkstationFileId()).setWorkstationId(addVO.getWorkstationId()).setArchiveDirectoryId(archiveDirectoryId);
|
} else {
|
return null;
|
}
|
return addFile;
|
}).filter((v0) -> {
|
return Func.isNotEmpty(v0);
|
}).collect(Collectors.toList());
|
}
|
|
@Override // org.springblade.modules.dnc.service.IDncArchiveDirectoryFileService
|
public DncArchiveDirectoryFile verifyFileName(String fileName, Long workstationId, Long archiveDirectoryId) {
|
return (DncArchiveDirectoryFile)getOne(Wrappers.<DncArchiveDirectoryFile>lambdaQuery()
|
.eq(DncArchiveDirectoryFile::getIsCurrent, DncEnums.CurrentVersion.YES.getCode())
|
.eq(DncArchiveDirectoryFile::getFilename, fileName)
|
.eq(DncArchiveDirectoryFile::getWorkstationId, workstationId)
|
.eq(DncArchiveDirectoryFile::getArchiveDirectoryId, archiveDirectoryId));
|
/*
|
return (DncArchiveDirectoryFile) getOne((Wrapper) ((LambdaQueryWrapper) ((LambdaQueryWrapper) ((LambdaQueryWrapper) Wrappers.lambdaQuery().eq((v0) -> {
|
return v0.getIsCurrent();
|
}, DncEnums.CurrentVersion.YES.getCode())).eq((v0) -> {
|
return v0.getFilename();
|
}, fileName)).eq((v0) -> {
|
return v0.getWorkstationId();
|
}, workstationId)).eq((v0) -> {
|
return v0.getArchiveDirectoryId();
|
}, archiveDirectoryId));*/
|
}
|
|
@Override // org.springblade.modules.dnc.service.IDncArchiveDirectoryFileService
|
public boolean checkArchiveFile(VerifyArchiveFileVO verifyArchiveFile) {
|
List<DncArchiveDirectoryFile> list = new ArrayList<>();
|
List<String> archiveDirectoryIds = verifyArchiveFile.getArchiveDirectoryIds();
|
List<VerifyArchiveFileCheckVO> voList = verifyArchiveFile.getVoList();
|
for (String archiveDirectoryId : archiveDirectoryIds) {
|
voList.forEach(file -> {
|
String fileName = file.getFilename();
|
String workstationId = file.getWorkstationId();
|
DncArchiveDirectoryFile directoryFile = verifyFileName(fileName, Long.valueOf(Long.parseLong(workstationId)), Long.valueOf(Long.parseLong(archiveDirectoryId)));
|
if (directoryFile != null) {
|
if (Func.isNotBlank(file.getObjectKey())) {
|
boolean result = this.obsFileService.verifyFileContent(file.getObjectKey(), directoryFile.getContentMd5());
|
if (result) {
|
list.add(directoryFile);
|
}
|
} else if (!directoryFile.getContentMd5().equals(file.getContentMd5())) {
|
list.add(directoryFile);
|
}
|
}
|
});
|
}
|
return list.size() > 0;
|
}
|
}
|
