111

src/main/java/com/ekexiu/project/storage/diacrisis/AiCleanTask.java

@@ -1,5 +1,13 @@
package com.ekexiu.project.storage.diacrisis;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import org.jfw.util.ConstData;
import org.jfw.util.io.IoUtil;
import org.jfw.util.log.LogFactory;
import org.jfw.util.log.Logger;
@@ -20,7 +28,44 @@ public class AiCleanTask implements Runnable{
	@Override
	public void run() {
		aiTaskService.updateTaskState(taskId, 399, null, new AiStrengthenTask(taskId, taskType, aiTaskService));
		String errCode="ERR0209";//调用异常
		File file = new File(aiTaskService.getDstPath(), taskId);
		file = new File(file,"data");
			try {
				URL u = new URL("http://127.0.0.1:5001//desensitize?pi=" + file.getAbsolutePath());
				HttpURLConnection httpURLConnection = (HttpURLConnection) u.openConnection();
				if (httpURLConnection.getResponseCode() == 200) {
					InputStream in = httpURLConnection.getInputStream();
					try {
						ByteArrayOutputStream out = new ByteArrayOutputStream();
						byte[] buf = new byte[128];
						int len = 0;
						while ((len = in.read(buf)) != -1) {
							if (len > 0) {
								out.write(buf, 0, len);
							}
						}
						buf = out.toByteArray();
						if (buf.length > 0) {
							String ret = new String(buf, ConstData.UTF8);
							ret = ret.trim();
							if (ret.equals("1")) {
								aiTaskService.updateTaskState(taskId, 399, null, new AiStrengthenTask(taskId, taskType, aiTaskService));
								return;
							}
						}
						errCode="ERR0207";//调用协议返回错误 
					} catch (Exception e) {
						errCode="ERR0207";//调用协议传输异常 
					} finally {
						IoUtil.close(in);
					}
				}else{
					errCode="ERR0208";//调用协议状态异常  http status != 200
				} 
			} catch (Exception ee) {
			}
		aiTaskService.updateTaskState(taskId, 350,errCode, null);
	}
}

src/main/java/com/ekexiu/project/storage/diacrisis/AiFormatTask.java

@@ -31,6 +31,7 @@ public class AiFormatTask implements Runnable {
	public void run() {
		File file = new File(aiTaskService.getDstPath(), taskId);
		file = new File(file,"data_enhance");
		if (!file.exists()) {
			LOG.error("Path[" + file.getAbsolutePath() + "] not exists");
			aiTaskService.updateTaskState(taskId, 552, "ERR0403", null);

src/main/java/com/ekexiu/project/storage/diacrisis/AiStrengthenTask.java

@@ -1,26 +1,67 @@
package com.ekexiu.project.storage.diacrisis;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import org.jfw.util.ConstData;
import org.jfw.util.io.IoUtil;
import org.jfw.util.log.LogFactory;
import org.jfw.util.log.Logger;
public class AiStrengthenTask implements Runnable{
public class AiStrengthenTask implements Runnable {
	private static Logger LOG = LogFactory.getLog(AiStrengthenTask.class);
	private String taskId;
	private int taskType;
	private AiTaskService aiTaskService;
	
	
	public AiStrengthenTask(String taskId,int taskType,AiTaskService aiTaskService){
	public AiStrengthenTask(String taskId, int taskType, AiTaskService aiTaskService) {
		this.taskId = taskId;
		this.taskType = taskType;
		this.aiTaskService = aiTaskService;
	}
	
	@Override
	public void run() {
		aiTaskService.updateTaskState(taskId, 499, null, new AiFormatTask(taskId, taskType, aiTaskService));
		String errCode = "ERR0309";// 调用异常
		File file = new File(aiTaskService.getDstPath(), taskId);
		file = new File(file, "data_desen");
		try {
			URL u = new URL("http://127.0.0.1:5003/enhance?pi=" + file.getAbsolutePath());
			HttpURLConnection httpURLConnection = (HttpURLConnection) u.openConnection();
			if (httpURLConnection.getResponseCode() == 200) {
				InputStream in = httpURLConnection.getInputStream();
				try {
					ByteArrayOutputStream out = new ByteArrayOutputStream();
					byte[] buf = new byte[128];
					int len = 0;
					while ((len = in.read(buf)) != -1) {
						if (len > 0) {
							out.write(buf, 0, len);
						}
					}
					buf = out.toByteArray();
					if (buf.length > 0) {
						String ret = new String(buf, ConstData.UTF8);
						ret = ret.trim();
						if (ret.equals("1")) {
							aiTaskService.updateTaskState(taskId, 499, null, new AiFormatTask(taskId, taskType, aiTaskService));
							return;
						}
					}
					errCode = "ERR0307";// 调用协议返回错误
				} catch (Exception e) {
					errCode = "ERR0307";// 调用协议传输异常
				} finally {
					IoUtil.close(in);
				}
			} else {
				errCode = "ERR0308";// 调用协议状态异常 http status != 200
			}
		} catch (Exception ee) {
		}
		aiTaskService.updateTaskState(taskId, 450, errCode, null);
	}
}

src/main/java/com/ekexiu/project/storage/diacrisis/AiTaskService.java

@@ -109,7 +109,6 @@ public class AiTaskService {
				con.close();
			}
		} catch (Exception e1) {
		}
	}
	

src/main/java/com/ekexiu/project/storage/diacrisis/AiTransferTask.java

@@ -25,37 +25,38 @@ public class AiTransferTask implements Runnable {
			LOG.error("Path[" + file.getAbsolutePath() + "] exists");
			aiTaskService.updateTaskState(taskId, 250, "ERR0199", null);
		} else {
			if (file.mkdir()) {
				if (taskType == 1) {
					for (int i = 1; i < 7; ++i) {
						File src = new File(aiTaskService.getSrcPath(), "IM_" + String.format("%04d.DICOM", i));
						File dst = new File(file, "IM_" + String.format("%04d.DICOM", i));
						if (!aiTaskService.copyFile(src, dst)) {
							LOG.error("copy File[" + src.getAbsolutePath() + "] error");
							aiTaskService.updateTaskState(taskId, 250, "ERR0199", null);
							return;
				file = new File(file,"data");
				if (file.mkdirs()) {
					if (taskType == 1) {
						for (int i = 1; i < 7; ++i) {
							File src = new File(aiTaskService.getSrcPath(), "IM_" + String.format("%04d.DICOM", i));
							File dst = new File(file, "IM_" + String.format("%04d.DICOM", i));
							if (!aiTaskService.copyFile(src, dst)) {
								LOG.error("copy File[" + src.getAbsolutePath() + "] error");
								aiTaskService.updateTaskState(taskId, 250, "ERR0199", null);
								return;
							}
						}
					}
				} else if (taskType == 2) {
					for (int i = 7; i < 14; ++i) {
						File src = new File(aiTaskService.getSrcPath(), "IM_" + String.format("%04d.DICOM", i));
						File dst = new File(file, "IM_" + String.format("%04d.DICOM", i));
						if (!!aiTaskService.copyFile(src, dst)) {
							LOG.error("copy File[" + src.getAbsolutePath() + "] error");
							aiTaskService.updateTaskState(taskId, 250, "ERR0199", null);
							return;
					} else if (taskType == 2) {
						for (int i = 7; i < 14; ++i) {
							File src = new File(aiTaskService.getSrcPath(), "IM_" + String.format("%04d.DICOM", i));
							File dst = new File(file, "IM_" + String.format("%04d.DICOM", i));
							if (!aiTaskService.copyFile(src, dst)) {
								LOG.error("copy File[" + src.getAbsolutePath() + "] error");
								aiTaskService.updateTaskState(taskId, 250, "ERR0199", null);
								return;
							}
						}
					} else {
						LOG.error("err task type[" + taskType + "]");
						aiTaskService.updateTaskState(taskId, 250, "ERR0199", null);
						return;
					}
					aiTaskService.updateTaskState(taskId, 299, null, new AiCleanTask(taskId, taskType, aiTaskService));
				} else {
					LOG.error("err task type[" + taskType + "]");
					LOG.error("mkdir File[" + file.getAbsolutePath() + "] error");
					aiTaskService.updateTaskState(taskId, 250, "ERR0199", null);
					return;
				}
				aiTaskService.updateTaskState(taskId, 299, null, new AiCleanTask(taskId, taskType, aiTaskService));
			} else {
				LOG.error("mkdir File[" + file.getAbsolutePath() + "] error");
				aiTaskService.updateTaskState(taskId, 250, "ERR0199", null);
			}
		}
	}