mdm.git - Gitblit

			@@ -1,16 +1,25 @@

			package org.springblade.mdm.program.service;

			import com.alibaba.fastjson.JSONArray;
			import com.alibaba.fastjson.JSONObject;
			import com.baomidou.mybatisplus.core.metadata.IPage;
			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.ZipArchiveInputStream;
			import org.apache.commons.compress.archivers.zip.ZipFile;
			import org.apache.commons.compress.utils.SeekableInMemoryByteChannel;
			import org.apache.commons.io.IOUtils;
			import org.springblade.core.mp.base.BizServiceImpl;
			import org.springblade.core.mp.support.Condition;
			import org.springblade.core.mp.support.Query;
			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.NcProgram;
			import org.springblade.mdm.program.entity.NcProgramExchange;
			import org.springblade.mdm.program.mapper.NcProgramExchangeMapper;
			import org.springblade.mdm.program.vo.DncSendBackData;
			@@ -23,10 +32,8 @@
			import java.nio.file.Files;
			import java.nio.file.Path;
			import java.nio.file.Paths;
			import java.util.ArrayList;
			import java.util.Collections;
			import java.util.Enumeration;
			import java.util.List;
			import java.time.LocalDateTime;
			import java.util.*;

			/**
			* 程序交换（dnc导入/导出）
			@@ -37,30 +44,26 @@
			@Service
			@AllArgsConstructor
			public class NcProgramExchangeService extends BizServiceImpl<NcProgramExchangeMapper, NcProgramExchange> {

			private final CureFlowService cureFlowService;
			private final NcProgramService ncProgramService;
			/**
			* dnc回传文件上传
			* @param file 文件
			* dnc回传文件上传（解析后保存入upload表）
			* @param file DNC回传文件
			* @return
			*/
			public List<DncSendBackData> dncSendBackUpload(MultipartFile file) {
			List<DncSendBackData> list ;
			List<DncSendBackData> list;
			try {
			String fileName = file.getOriginalFilename();
			//InputStream fileInputStream = file.getInputStream();
			InputStream zipFileInputStream = FileExchangeUtil.convertFileToZip(file.getInputStream());
			//String fileName = file.getOriginalFilename();
			//InputStream zipFileInputStream = FileExchangeUtil.convertFileToZip(file.getInputStream());
			InputStream zipFileInputStream = file.getInputStream();//test

			byte[] bytes = FileUtil.copyToByteArray(zipFileInputStream);
			list = parseDncZipFromByteArray(bytes);
			for(DncSendBackData dncSendBackData:list){
			NcProgramExchange exchange=new NcProgramExchange();
			exchange.setName(dncSendBackData.getProgramName());
			exchange.setStatus(1);//已导入
			this.save(exchange);

			}

			} catch (IOException e) {
			log.error("上传dnc文件失败",e);
			log.error("上传dnc回传文件失败",e);
			list = Collections.emptyList();
			}
			return list;
			@@ -92,90 +95,93 @@
			return tempFile.toFile();
			}
			public static List<DncSendBackData> parseDncZipFromByteArray(byte[] zipData) throws IOException {
			//List<DncSendBackData> datas = new ArrayList<>();
			List<DncSendBackData> datas = ZipFileDirectoryScanner.getFilesInDirectoryRecursive(zipData, "");
			List<DncSendBackData> list = new ArrayList<>();
			//List<DncSendBackData> datas = ZipFileDirectoryScanner.getFilesInDirectoryRecursive(zipData, "");



			/*
			try (ByteArrayInputStream bis = new ByteArrayInputStream(zipData);
			ZipArchiveInputStream zis = new ZipArchiveInputStream(bis)) {

			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;
			while ((entry = zis.getNextZipEntry()) != null) {
			Enumeration<ZipArchiveEntry> entries = zipFile.getEntries();
			while (entries.hasMoreElements()) {
			//while ((entry = zis.getNextZipEntry()) != null) {
			entry = entries.nextElement();
			DncSendBackData prog = new DncSendBackData();
			prog.setProgramName(entry.getName());
			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);
			DncSendBackData d = new DncSendBackData();
			d.setProgramName(jsonObject.getString("name"));
			d.setId(jsonObject.getLong("id"));
			d.setProgramNo(jsonObject.getString("code"));
			d.setFileBackTime(LocalDateTime.now());//到达时间

			fileDataMap.put(d.getProgramName(),d);
			list.add(d);
			}

			}
			}else{
			try (InputStream inputStream = zipFile.getInputStream(entry)) {
			fileMd5Map.put(entryName,DigestUtils.md5Hex(inputStream));//获取文件MD5
			}

			}
			System.out.println("文件名: " + entry.getName());
			System.out.println("大小: " + entry.getSize());

			// 读取文件内容到字节数组
			ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
			IOUtils.copy(zis, outputStream);
			byte[] fileContent = outputStream.toByteArray();

			// 处理文件内容...
			System.out.println("内容长度: " + fileContent.length);
			}else{
			//文件夹，读内部文件，获取文件列表

			System.out.println("文件夹程序："+entry.getName());
			List<String> children = new ArrayList<>();
			prog.setChildren(children);
			}

			datas.add(prog);
			}


			}*/
			return datas;
			}
			//设置md5值
			fileDataMap.forEach((k,v)->{
			if(fileMd5Map.containsKey(k)){
			v.setMd5(fileMd5Map.get(k));
			}
			});
			return list;
			}


			/**
			* 程序下发统计分页查询
			* @param query 查询参数
			* 入库回传文件,并启动固化流程
			* @param ids id列表逗号分隔
			* @return
			*/
			public void dncFileAccept(String ids) {
			List<Long> idList = Func.toLongList(ids);
			List<NcProgram> progList = ncProgramService.listByIds(idList);
			NcProgramExchange exchange;
			//NcProgram program;
			//NcNode programNode;

			public IPage<DncSendBackData> dncSendBackPageQuery(Query query) {
			for(NcProgram prog:progList){
			exchange = new NcProgramExchange();
			exchange.setName(prog.getName());
			exchange.setExchangeType(2);//回传
			exchange.setNcProgramId(prog.getId());

			IPage<DncSendBackData> page = this.getBaseMapper().dncSendBackpageQuery(Condition.getPage(query),query);
			this.save(exchange);

			return page;

			}
			}


			class ZipFileDirectoryScanner {

			public static List<DncSendBackData> getFilesInDirectoryRecursive(byte[] zipData, String dirPath) throws IOException {
			List<DncSendBackData> list = new ArrayList<>();

			if (!dirPath.endsWith("/")) {
			dirPath += "/";
			}

			try (SeekableInMemoryByteChannel channel = new SeekableInMemoryByteChannel(zipData);
			ZipFile zipFile = new ZipFile(channel)) {
			cureFlowService.startCure(progList);

			Enumeration<ZipArchiveEntry> entries = zipFile.getEntries();
			while (entries.hasMoreElements()) {
			ZipArchiveEntry entry = entries.nextElement();
			String entryName = entry.getName();
			DncSendBackData d = new DncSendBackData();
			d.setProgramName(entryName);
			if(entry.isDirectory()){
			d.setHasChildren(true);
			}
			list.add(d);

			//if (entryName.startsWith(dirPath) && !entry.isDirectory()) {
			// fileList.add(entryName);
			//}
			}
			}
			return list;
			}
			}