|
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.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.entity.ApproveRecord;
|
import org.springblade.mdm.flow.service.ApproveRecordService;
|
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.MdmProgramImportVO;
|
import org.springblade.mdm.utils.CustomBinaryReader;
|
import org.springframework.stereotype.Service;
|
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.util.*;
|
|
/**
|
* MDM程序导入（工控网功能）
|
*
|
* @author yangys
|
*/
|
@Slf4j
|
@Service
|
@AllArgsConstructor
|
public class MdmProgramImportService extends BizServiceImpl<NcProgramExchangeMapper, NcProgramExchange> {
|
private final CureFlowService cureFlowService;
|
private final NcProgramService ncProgramService;
|
private final NcNodeService ncNodeService;
|
private final ApproveRecordService approveRecordService;
|
private final BladeRedis bladeRedis;
|
private final OssTemplate ossTemplate;
|
|
private String getFileKey(){
|
return "mdmimpfile-"+ AuthUtil.getUserId();
|
}
|
/**
|
* MDM导入文件上传
|
* @param file MDM涉密网导出文件
|
* @return
|
*/
|
public List<MdmProgramImportVO> mdmImportUpload(MultipartFile file) {
|
List<MdmProgramImportVO> list;
|
try {
|
BladeFile bfile = ossTemplate.putFile(file);//上传，供后续入库使用
|
|
//设置一个缓存，2小时过期
|
bladeRedis.setEx(getFileKey(),bfile.getName(), Duration.ofHours(2));
|
String pkgFileName = bladeRedis.get(getFileKey());
|
//String fileName = file.getOriginalFilename();
|
//InputStream zipFileInputStream = FileExchangeUtil.convertFileToZip(file.getInputStream());
|
InputStream zipFileInputStream = file.getInputStream();//test
|
|
byte[] bytes = FileUtil.copyToByteArray(zipFileInputStream);
|
list = parseMdmZipFromByteArray(bytes);
|
|
|
} catch (IOException e) {
|
log.error("上传dnc回传文件失败",e);
|
list = Collections.emptyList();
|
}
|
return list;
|
}
|
|
InputStream convertFileToZip(InputStream inputStream) throws IOException {
|
|
File tempFile = createTempFile();
|
FileOutputStream fos = new FileOutputStream(tempFile);
|
CustomBinaryReader.read(inputStream,fos);
|
|
|
FileInputStream dInstream = new FileInputStream(tempFile);
|
|
return dInstream;
|
}
|
|
/**
|
* 创建一个临时文件
|
* @return
|
* @throws IOException
|
*/
|
File createTempFile() throws IOException {
|
Path tempDir = Paths.get(System.getProperty("java.io.tmpdir"));
|
// 在临时目录中创建文件
|
String tfilename = "t"+System.currentTimeMillis();
|
Path tempFile = Files.createTempFile(tempDir, tfilename, ".tmp");
|
System.out.println("创建的临时文件: " + tempFile);
|
return tempFile.toFile();
|
}
|
public static List<MdmProgramImportVO> parseMdmZipFromByteArray(byte[] zipData) throws IOException {
|
List<MdmProgramImportVO> list = new ArrayList<>();
|
Map<String,String> fileMd5Map = new HashMap<>();
|
Map<String,MdmProgramImportVO> 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();
|
|
String entryName = entry.getName();
|
|
if (!entry.isDirectory()) {
|
//直接解析程序的json文件
|
if(entryName.equals(NcProgramExportDNCService.PROGRAM_JSON_FILE)){
|
|
try (InputStream inputStream = zipFile.getInputStream(entry)) {
|
String jsonStr = IoUtil.readToString(inputStream);
|
|
JSONArray jsonArray = JSONArray.parseArray(jsonStr);
|
for(int i=0;i<jsonArray.size();i++){
|
JSONObject jsonObject = jsonArray.getJSONObject(i);
|
MdmProgramImportVO d = new MdmProgramImportVO();
|
d.setName(jsonObject.getString("name"));
|
d.setId(jsonObject.getLong("id"));
|
d.setCode(jsonObject.getString("code"));
|
d.setDescription(jsonObject.getString("description"));
|
d.setCategory(jsonObject.getString("category"));
|
d.setBindNcNodeId(jsonObject.getLong("bindNcNodeId"));
|
d.setIsLastEdition(jsonObject.getInteger("isLastEdition"));
|
d.setIsLocked(jsonObject.getInteger("isLocked"));
|
d.setIsTest(jsonObject.getInteger("isTest"));
|
d.setMachineCode(jsonObject.getString("machineCode"));
|
d.setNcNodeId(jsonObject.getLong("ncNodeId"));
|
d.setDrawingNo(jsonObject.getString("drawingNo"));
|
d.setProcessEdition(jsonObject.getString("processEdition"));
|
|
fileDataMap.put(d.getName(),d);
|
|
list.add(d);
|
}
|
|
}
|
}else{
|
try (InputStream inputStream = zipFile.getInputStream(entry)) {
|
fileMd5Map.put(entryName,DigestUtils.md5Hex(inputStream));//获取文件MD5
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
//设置md5值
|
fileDataMap.forEach((k,v)->{
|
if(fileMd5Map.containsKey(k)){
|
v.setMd5(fileMd5Map.get(k));
|
}
|
});
|
return list;
|
}
|
|
|
/**
|
* 入库mdm涉密网文件
|
* @param ids id列表逗号分隔
|
* @return
|
*/
|
public void mdmFileAccept(String ids) throws IOException {
|
|
String pkgFileName = bladeRedis.get(getFileKey());
|
|
List<Long> idList = Func.toLongList(ids);
|
//
|
updateProgramFile(pkgFileName,idList);
|
|
List<NcProgram> progList = ncProgramService.listByIds(idList);
|
NcProgramExchange exchange;
|
|
for(NcProgram prog:progList){
|
exchange = new NcProgramExchange();
|
exchange.setName(prog.getName());
|
exchange.setExchangeType(2);//回传
|
exchange.setNcProgramId(prog.getId());
|
|
|
this.save(exchange);
|
|
}
|
|
|
}
|
|
/**
|
* 更新程序文件
|
* @param pkgFileName mdm程序包
|
* @param idList 选定入库的
|
*/
|
private void updateProgramFile(String pkgFileName,List<Long> idList) throws IOException {
|
InputStream inputStream = this.ossTemplate.statFileStream(pkgFileName);
|
byte[] bytes = FileUtil.copyToByteArray(inputStream);
|
|
List<NcNode> nodeList = new ArrayList<>();
|
List<NcProgram> progList = new ArrayList<>();
|
List<ApproveRecord> recordList = 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();
|
|
String entryName = entry.getName();
|
|
if (!entry.isDirectory()) {
|
if(entryName.equals(NcProgramExportDNCService.PROGRAM_JSON_FILE)){
|
|
try (InputStream insJson = zipFile.getInputStream(entry)) {
|
String jsonStr = IoUtil.readToString(insJson);
|
|
JSONArray jsonArray = JSONArray.parseArray(jsonStr);
|
for(int i=0;i<jsonArray.size();i++){
|
JSONObject jsonObject = jsonArray.getJSONObject(i);
|
NcProgram prog = new NcProgram();
|
prog.setId(jsonObject.getLong("id"));
|
prog.setName(jsonObject.getString("name"));
|
|
prog.setCode(jsonObject.getString("code"));
|
prog.setDescription(jsonObject.getString("description"));
|
prog.setCategory(jsonObject.getString("category"));
|
prog.setBindNcNodeId(jsonObject.getLong("bindNcNodeId"));
|
prog.setIsLastEdition(jsonObject.getInteger("isLastEdition"));
|
prog.setIsLocked(jsonObject.getInteger("isLocked"));
|
prog.setIsTest(jsonObject.getInteger("isTest"));
|
prog.setIsTextFile(jsonObject.getBoolean("isTextFile"));
|
prog.setMachineCode(jsonObject.getString("machineCode"));
|
prog.setNcNodeId(jsonObject.getLong("ncNodeId"));
|
prog.setDrawingNo(jsonObject.getString("drawingNo"));
|
prog.setProcessEdition(jsonObject.getString("processEdition"));
|
|
setBaseProperties(prog,jsonObject);
|
|
progList.add(prog);
|
}
|
|
}
|
}else if(entryName.equals(NcProgramExportDNCService.NODE_JSON_FILE)){
|
try (InputStream insJson = zipFile.getInputStream(entry)) {
|
String jsonStr = IoUtil.readToString(insJson);
|
JSONArray jsonArray = JSONArray.parseArray(jsonStr);
|
for (int i = 0; i < jsonArray.size(); i++) {
|
JSONObject jsonObject = jsonArray.getJSONObject(i);
|
NcNode node = new NcNode();
|
node.setId(jsonObject.getLong("id"));
|
node.setName(jsonObject.getString("name"));
|
node.setDescription(jsonObject.getString("description"));
|
node.setNodeType(jsonObject.getString("nodeType"));
|
node.setProcessName(jsonObject.getString("processName"));
|
node.setDrawingNo(jsonObject.getString("drawingNo"));
|
node.setMachineCode(jsonObject.getString("machineCode"));
|
node.setParentId(jsonObject.getLong("parentId"));
|
node.setParentIds(jsonObject.getString("parentIds"));
|
node.setIsCured(jsonObject.getInteger("isCured"));
|
|
setBaseProperties(node,jsonObject);
|
|
nodeList.add(node);
|
}
|
}
|
}else if(entryName.equals(NcProgramExportDNCService.APPROVE_RECORD_JSON_FILE)){
|
try (InputStream insJson = zipFile.getInputStream(entry)) {
|
String jsonStr = IoUtil.readToString(insJson);
|
JSONArray jsonArray = JSONArray.parseArray(jsonStr);
|
for (int i = 0; i < jsonArray.size(); i++) {
|
JSONObject jsonObject = jsonArray.getJSONObject(i);
|
ApproveRecord record = new ApproveRecord();
|
record.setId(jsonObject.getLong("id"));
|
record.setNcProgramId(jsonObject.getLong("ncProgramId"));
|
record.setUserId(jsonObject.getLong("userId"));
|
record.setUserNickname(jsonObject.getString("userNickname"));
|
record.setOperateResult(jsonObject.getString("operateResult"));
|
record.setOperateTime(jsonObject.getDate("operateTime"));
|
record.setComment(jsonObject.getString("comment"));
|
setBaseProperties(record,jsonObject);
|
|
recordList.add(record);
|
}
|
}
|
}
|
|
}
|
}
|
|
}
|
|
|
try (SeekableInMemoryByteChannel channel = new SeekableInMemoryByteChannel(bytes);
|
ZipFile zipFile = new ZipFile(channel)) {
|
|
ZipArchiveEntry entry;
|
Enumeration<ZipArchiveEntry> entries = zipFile.getEntries();
|
while (entries.hasMoreElements()) {
|
entry = entries.nextElement();
|
|
String entryName = entry.getName();
|
|
if (!entry.isDirectory() && !entryName.equals(NcProgramExportDNCService.PROGRAM_JSON_FILE)) {
|
|
for(NcProgram prog:progList){
|
if(prog.getName().equals(entryName)){
|
try (InputStream ncFileStream = zipFile.getInputStream(entry)) {
|
BladeFile bfile = this.ossTemplate.putFile(prog.getName(),ncFileStream);
|
prog.setOssName(bfile.getName());
|
prog.setUrl(bfile.getLink());
|
}
|
break;
|
}
|
}
|
|
}
|
}
|
}
|
|
for(NcNode node:nodeList){
|
NcNode nodeTemp = this.ncNodeService.getById(node.getId());
|
if(nodeTemp == null){
|
ncNodeService.save(node);
|
}else{
|
ncNodeService.updateById(node);
|
}
|
}
|
|
for(NcProgram prog:progList){
|
NcProgram ncTemp = ncProgramService.getById(prog.getId());
|
if(ncTemp == null){
|
ncProgramService.save(prog);
|
}else{
|
ncProgramService.updateById(prog);
|
}
|
}
|
|
for(ApproveRecord record:recordList) {
|
ApproveRecord recordTemp = this.approveRecordService.getById(record.getId());
|
if (recordTemp == null) {
|
approveRecordService.save(record);
|
} else {
|
approveRecordService.updateById(record);
|
}
|
}
|
|
}
|
|
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"));
|
}
|
}
|
