|
package org.springblade.mdm.program.service;
|
|
import com.alibaba.fastjson.JSONArray;
|
import com.alibaba.fastjson.JSONObject;
|
import lombok.AllArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.codec.digest.DigestUtils;
|
import org.apache.commons.compress.archivers.zip.ZipArchiveEntry;
|
import org.apache.commons.compress.archivers.zip.ZipFile;
|
import org.apache.commons.compress.utils.SeekableInMemoryByteChannel;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springblade.core.log.exception.ServiceException;
|
import org.springblade.core.mp.base.BizEntity;
|
import org.springblade.core.mp.base.BizServiceImpl;
|
import org.springblade.core.oss.OssTemplate;
|
import org.springblade.core.oss.model.BladeFile;
|
import org.springblade.core.redis.cache.BladeRedis;
|
import org.springblade.core.secure.utils.AuthUtil;
|
import org.springblade.core.tool.utils.FileUtil;
|
import org.springblade.core.tool.utils.Func;
|
import org.springblade.core.tool.utils.IoUtil;
|
import org.springblade.mdm.flow.service.CureFlowService;
|
import org.springblade.mdm.program.entity.NcNode;
|
import org.springblade.mdm.program.entity.NcProgram;
|
import org.springblade.mdm.program.entity.NcProgramExchange;
|
import org.springblade.mdm.program.mapper.NcProgramExchangeMapper;
|
import org.springblade.mdm.program.vo.DncSendBackData;
|
import org.springblade.mdm.utils.CustomBinaryReader;
|
import org.springframework.beans.BeanUtils;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.io.*;
|
import java.nio.file.Files;
|
import java.nio.file.Path;
|
import java.nio.file.Paths;
|
import java.time.Duration;
|
import java.time.LocalDateTime;
|
import java.util.*;
|
|
/**
|
* DNC回传文件处理服务
|
*
|
* @author yangys
|
*/
|
@Slf4j
|
@Service
|
@AllArgsConstructor
|
public class DNCSendBackService extends BizServiceImpl<NcProgramExchangeMapper, NcProgramExchange> {
|
private final CureFlowService cureFlowService;
|
private final NcProgramService ncProgramService;
|
private final NcNodeService ncNodeService;
|
private final OssTemplate ossTemplate;
|
private final BladeRedis bladeRedis;
|
|
private String getFileKey(){
|
return "dncexpfile-"+ AuthUtil.getUserId();
|
}
|
/**
|
* dnc回传文件上传
|
* @param file DNC回传文件
|
* @return 压缩包内程序包名的列表
|
*/
|
public List<DncSendBackData> dncSendBackUpload(MultipartFile file) {
|
List<DncSendBackData> list;
|
try {
|
BladeFile bfile = ossTemplate.putFile(file);//上传，供后续入库使用
|
//设置一个缓存，2小时过期
|
bladeRedis.setEx(getFileKey(),bfile.getName(), Duration.ofHours(2));
|
|
InputStream zipFileInputStream = file.getInputStream();//test
|
|
byte[] bytes = FileUtil.copyToByteArray(zipFileInputStream);
|
list = parseDncZipFromByteArray(bytes);
|
|
} catch (IOException e) {
|
log.error("上传dnc回传文件失败",e);
|
throw new ServiceException("解析DNC回传数据失败");
|
}
|
return list;
|
}
|
|
|
/**
|
* 从压缩包 解析回传程序列表，这里解析目录即可，目录就是程序包名
|
* @param zipData 压缩包字节
|
* @return 回传程序列表
|
* @throws IOException
|
*/
|
List<DncSendBackData> parseDncZipFromByteArray(byte[] zipData) throws IOException {
|
List<DncSendBackData> list = new ArrayList<>();
|
|
//Map<String,String> fileMd5Map = new HashMap<>();
|
Map<String,DncSendBackData> fileDataMap = new HashMap<>();
|
try (SeekableInMemoryByteChannel channel = new SeekableInMemoryByteChannel(zipData);
|
ZipFile zipFile = new ZipFile(channel)) {
|
|
ZipArchiveEntry entry;
|
Enumeration<ZipArchiveEntry> entries = zipFile.getEntries();
|
while (entries.hasMoreElements()) {
|
entry = entries.nextElement();
|
DncSendBackData progData = new DncSendBackData();
|
String entryName = entry.getName();
|
if (entry.isDirectory()){
|
//目录，才是程序包，
|
//查询数据库，对应上才能确认时有效的程序包
|
String packageName = StringUtils.removeEnd(entryName,"/");
|
List<NcNode> pkgList = ncNodeService.lambdaQuery().eq(NcNode::getNodeType,NcNode.TYPE_PROGRAM_PACKAGE).eq(NcNode::getName, packageName).eq(NcNode::getIsLastEdition,1).list();
|
if(!pkgList.isEmpty()) {
|
NcNode programPackageNode = pkgList.get(0);
|
progData.setId(programPackageNode.getId());
|
progData.setProgramName(packageName);
|
progData.setFileBackTime(LocalDateTime.now());
|
progData.setProgramNo(programPackageNode.getProgramNo());
|
list.add(progData);
|
}
|
}
|
}
|
|
}
|
|
return list;
|
}
|
|
|
/**
|
* 入库回传文件,并启动固化流程
|
* @param ids id列表逗号分隔,程序包名 节点的id
|
* @return
|
*/
|
@Transactional
|
public void dncFileAccept(String ids) throws IOException {
|
List<Long> idList = Func.toLongList(ids);
|
//
|
NcProgramExchange exchange;
|
String pkgFileName = bladeRedis.get(getFileKey());
|
updateNodeDataByDNCBackData(pkgFileName,idList);
|
List<NcNode> pkgList = ncNodeService.listByIds(idList);
|
for(NcNode pkgNode :pkgList){
|
exchange = new NcProgramExchange();
|
exchange.setName(pkgNode.getName());
|
exchange.setExchangeType(2);//回传
|
exchange.setNcNodeId(pkgNode.getId());
|
|
this.save(exchange);
|
}
|
|
//cureFlowService.startCure(pkgList);
|
}
|
|
/**
|
* 更新节点，主要是创建 程序包名 的新版本。
|
* @param pkgFileName zip文件名
|
* @param programPackageIdList 程序包名 节点的id列表
|
* @throws IOException 访问文件异常
|
*/
|
void updateNodeDataByDNCBackData(String pkgFileName, List<Long> programPackageIdList) throws IOException {
|
InputStream inputStream = this.ossTemplate.statFileStream(pkgFileName);
|
byte[] bytes = FileUtil.copyToByteArray(inputStream);
|
|
List<NcNode> nodeList = new ArrayList<>();
|
List<String> entryNameList = new ArrayList<>();
|
try (SeekableInMemoryByteChannel channel = new SeekableInMemoryByteChannel(bytes);
|
ZipFile zipFile = new ZipFile(channel)) {
|
|
ZipArchiveEntry entry;
|
Enumeration<ZipArchiveEntry> entries = zipFile.getEntries();
|
while (entries.hasMoreElements()) {
|
entry = entries.nextElement();
|
entryNameList.add(entry.getName());
|
}
|
|
//根据内部文件，读取和分析程序包和程序文件数据
|
List<String> dirList = entryNameList.stream().filter(s -> s.endsWith("/")).toList();
|
for(String dir : dirList){
|
String programPackageName = StringUtils.removeEnd(dir,"/");
|
NcNode oriProgramPkg = this.ncNodeService.getLastEditionProgramPackage(programPackageName);
|
if(oriProgramPkg == null){
|
log.warn("未发现匹配的程序包名{}",programPackageName);
|
continue;
|
}
|
|
if(!programPackageIdList.contains(oriProgramPkg.getId())){
|
//不在勾选的范围内
|
continue;
|
}
|
//临时测试注释
|
|
NcNode newProgramPkg = new NcNode();
|
BeanUtils.copyProperties(oriProgramPkg, newProgramPkg);
|
|
clearBaseProperties(newProgramPkg);
|
newProgramPkg.setIsLastEdition(1);
|
|
ncNodeService.save(newProgramPkg);
|
|
//旧数据更新为老版本
|
oriProgramPkg.setIsLastEdition(0);;
|
ncNodeService.updateById(oriProgramPkg);
|
|
//查找包下的文件数据,
|
entryNameList.stream().filter(s -> s.startsWith(dir)).forEach(entryName -> {
|
log.info("{}下的文件：{}",dir,entryName);
|
if(!entryName.endsWith("/")){
|
//实际的文件
|
String fileName = StringUtils.removeStart(entryName,dir);//去除文件名路径部分
|
NcNode oldProgramNode = this.ncNodeService.getLastEditionProgramFile(fileName,oriProgramPkg.getId());
|
if(oldProgramNode == null){
|
log.info("{}找不到程序文件",entryName);
|
return;
|
}
|
|
NcNode newProgramNode = new NcNode();
|
BeanUtils.copyProperties(oriProgramPkg, newProgramNode);
|
clearBaseProperties(newProgramNode);
|
newProgramNode.setIsLastEdition(1);
|
ncNodeService.save(newProgramNode);
|
|
ncNodeService.updateById(oriProgramPkg);
|
|
}
|
});
|
}
|
|
}
|
|
}
|
|
void setBaseProperties(BizEntity entity, JSONObject jsonObject){
|
entity.setCreateTime(jsonObject.getDate("createTime"));
|
entity.setUpdateTime(jsonObject.getDate("updateTime"));
|
entity.setStatus(jsonObject.getInteger("status"));
|
entity.setCreateUser(jsonObject.getLong("createUser"));
|
entity.setUpdateUser(jsonObject.getLong("updateUser"));
|
}
|
|
void clearBaseProperties(BizEntity entity){
|
entity.setId(null);
|
entity.setCreateTime(null);
|
entity.setUpdateTime(null);
|
entity.setStatus(null);
|
entity.setCreateUser(null);
|
entity.setUpdateUser(null);
|
}
|
}
|
