|
package org.springblade.mdm.program.service;
|
|
import com.alibaba.csp.sentinel.util.StringUtil;
|
import lombok.AllArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.io.FilenameUtils;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springblade.core.log.exception.ServiceException;
|
import org.springblade.core.mp.base.BizServiceImpl;
|
import org.springblade.core.oss.OssTemplate;
|
import org.springblade.mdm.basesetting.machine.service.MachineService;
|
import org.springblade.mdm.basesetting.machine.entity.Machine;
|
import org.springblade.mdm.flow.entity.FlowProgramFile;
|
import org.springblade.mdm.flow.service.FlowProgramFileService;
|
import org.springblade.mdm.program.entity.NcNode;
|
import org.springblade.mdm.program.entity.NcProgramApproved;
|
import org.springblade.mdm.program.entity.NcProgramExchange;
|
import org.springblade.mdm.program.mapper.NcProgramExchangeMapper;
|
import org.springblade.mdm.program.service.programannotation.AnnotationData;
|
import org.springblade.mdm.program.service.programannotation.AnnotationProcessor;
|
import org.springblade.mdm.program.service.programannotation.MachineAnnotationConfig;
|
import org.springblade.mdm.program.service.programannotation.ProcessorHelper;
|
import org.springblade.system.feign.IDictClient;
|
import org.springblade.system.pojo.entity.DictBiz;
|
import org.springframework.stereotype.Service;
|
|
import java.io.*;
|
import java.util.*;
|
import java.util.zip.ZipEntry;
|
import java.util.zip.ZipOutputStream;
|
|
/**
|
* 下发/回传程序统计
|
*
|
* @author yangys
|
*/
|
@Slf4j
|
@Service
|
@AllArgsConstructor
|
public class NcProgramExportDNCService extends BizServiceImpl<NcProgramExchangeMapper, NcProgramExchange> {
|
|
private final IDictClient dictClient;
|
private final MachineService machineService;
|
private final NcProgramApprovedService approvedService;
|
private final NcNodeService ncNodeService;
|
private final OssTemplate ossTemplate;
|
private final FlowProgramFileService flowProgramFileService;
|
private final ProgramAnnotationService programAnnotationService;
|
|
private final MachineAnnotationConfig annotationConfig;
|
/**
|
* 导出dnc压缩包
|
* @param approvedIdArray 待导出审批表id数组
|
*/
|
public void exportDnc(Long[] approvedIdArray, OutputStream os) throws IOException {
|
//FileOutputStream fos = new FileOutputStream("d:/exportDnc.zip");
|
try (ZipOutputStream zipOut = new ZipOutputStream(os);os) {//os
|
|
for (Long approvedId : approvedIdArray) {
|
NcProgramApproved approved = approvedService.getById(approvedId);
|
addProgramPackageToZip(zipOut,approved);
|
|
approved.setStatus(NcProgramApproved.STATUS_EXPORTED);
|
approvedService.updateById(approved);
|
}
|
|
}catch(Exception e){
|
Throwable [] err = e.getSuppressed();
|
throw new ServiceException("导出工控网错误"+e.getMessage());
|
}
|
}
|
|
/**
|
* 将程序包和下属文件加入压缩包
|
* @param zipOut
|
* @param approved
|
*/
|
private void addProgramPackageToZip(ZipOutputStream zipOut, NcProgramApproved approved) throws IOException{
|
NcNode packageNode = ncNodeService.getById(approved.getNcNodeId());
|
|
String packageFolder = packageNode.getName()+"/";
|
ZipEntry zipEntry = new ZipEntry(packageFolder);// "/"结尾表示文件夹
|
zipOut.putNextEntry(zipEntry);
|
zipOut.closeEntry();
|
|
Machine machine = machineService.getByCode(packageNode.getMachineCode());//程序包节点，获取注释用
|
|
List<NcNode> programNodes = ncNodeService.lambdaQuery().eq(NcNode::getIsLastEdition,1).eq(NcNode::getParentId, approved.getNcNodeId()).list();
|
|
FlowProgramFile programFile;
|
NcNode programPackageNode = this.ncNodeService.getById(approved.getNcNodeId());
|
String status = "SQ";
|
if(programPackageNode.isDeviationProgram()){
|
status = "PL";
|
}else if(programPackageNode.hasCured()){
|
status = "GH";
|
}
|
long deviationSerial = -1;//偏离单序号 ，-1为非偏离单，不需要加入
|
if(packageNode.isDeviationProgram()) {
|
deviationSerial = ncNodeService.getDeviationSerialForNode(packageNode);
|
}
|
|
//List<DictBiz> annotationDictList= programAnnotationService.getAnnotionDictList();
|
for (NcNode node : programNodes) {
|
String filePathInZip = genFilePathInZip(packageFolder,node,deviationSerial);
|
programFile = this.flowProgramFileService.getById(node.getFlowProgramFileId());
|
if(programFile.isProgram()) {//程序文件，才会加入压缩包
|
InputStream inputStream = ossTemplate.statFileStream(programFile.getOssName());
|
String sendDir = machine.getProgSendDir()==null? StringUtil.EMPTY:machine.getProgSendDir();
|
|
AnnotationProcessor annoProcessor = ProcessorHelper.getProcessor(machine.getControlSystem(),annotationConfig);
|
AnnotationData annoData = new AnnotationData();
|
annoData.setSendPath(sendDir);
|
annoData.setProgramStatus(status);
|
annoData.setDeviation(programPackageNode.getDeviation());
|
annoData.setFilename(programFile.getName());
|
InputStream addedInsFinal = annoProcessor.setAnnotation(annoData,inputStream);
|
addedInsFinal.reset();
|
/*
|
InputStream addedInsFinal;
|
InputStream addedIns2 = programAnnotationService.setSendDirAndStatusAnnotation(sendDir,status,inputStream,machine.getControlSystem());
|
if(StringUtils.isNotBlank(programPackageNode.getDeviation())) {
|
//偏离单，增加偏离单号
|
addedInsFinal = programAnnotationService.setPldAnnotation(programPackageNode.getDeviation(),addedIns2, machine.getControlSystem(), annotationDictList);;
|
}else{
|
addedInsFinal = addedIns2;
|
}*/
|
this.addInputStreamToZip(zipOut, addedInsFinal, filePathInZip);
|
}
|
}
|
|
}
|
|
/**
|
* 生成在zip中的文件路径
|
* @param packageFolder zip中的文件夹
|
* @param programNode 程序节点
|
* @param deviationSerial 偏离单序号
|
* @return zip中的文件路径
|
*/
|
String genFilePathInZip(String packageFolder,NcNode programNode,long deviationSerial){
|
//为文件名增加偏离单号
|
String filePathInZip;
|
|
if(deviationSerial != -1){
|
//加入偏离单序号
|
String ext = FilenameUtils.getExtension(programNode.getName());
|
String dotExt = StringUtils.isNotBlank(ext)?"."+ext:ext;//带点的扩展名
|
|
String temp = programNode.getName();
|
if(StringUtils.isNotBlank(ext)) {
|
temp = StringUtils.removeEnd(temp,dotExt);
|
}
|
filePathInZip = packageFolder + temp+"-P"+deviationSerial+dotExt;
|
}else{
|
//非偏离单，直接使用原文件名
|
filePathInZip = packageFolder + programNode.getName();
|
}
|
return filePathInZip;
|
}
|
/**
|
* 将 输入流 中的内容写入zip
|
* @param zipOut zip输出流
|
* @param inputStream 输入流
|
* @param entryName 文件名
|
* @throws IOException
|
*/
|
void addInputStreamToZip(ZipOutputStream zipOut, InputStream inputStream, String entryName)
|
throws IOException {
|
// 创建新的 ZIP 条目
|
ZipEntry zipEntry = new ZipEntry(entryName);
|
zipOut.putNextEntry(zipEntry);
|
|
// 将输入流写入 ZIP 输出流
|
byte[] buffer = new byte[1024];
|
int length;
|
while ((length = inputStream.read(buffer)) >= 0) {
|
zipOut.write(buffer, 0, length);
|
}
|
|
// 关闭当前条目
|
zipOut.closeEntry();
|
}
|
}
|
