Commit f2107704 authored by linpeiqin's avatar linpeiqin

小修改

parent 5ec4476b
...@@ -151,7 +151,8 @@ public class ModelCompressServiceImpl extends BaseService<ModelCompress, Long> i ...@@ -151,7 +151,8 @@ public class ModelCompressServiceImpl extends BaseService<ModelCompress, Long> i
log.info("发送服务端的消息:" + sendJson.toJSONString()); log.info("发送服务端的消息:" + sendJson.toJSONString());
this.send(sendJson.toJSONString()); this.send(sendJson.toJSONString());
} else if (receiveMsg.equals("process_completed")) { } else if (receiveMsg.equals("process_completed")) {
if (receiveJson.getBoolean("success") == true){ this.close();
if (receiveJson.getBoolean("success")){
ModelVersion modelVersionS = new ModelVersion(); ModelVersion modelVersionS = new ModelVersion();
modelVersionS.setTaskId(modelCompress.getTaskId()); modelVersionS.setTaskId(modelCompress.getTaskId());
modelVersionS.setIsCompress(1); modelVersionS.setIsCompress(1);
...@@ -175,7 +176,6 @@ public class ModelCompressServiceImpl extends BaseService<ModelCompress, Long> i ...@@ -175,7 +176,6 @@ public class ModelCompressServiceImpl extends BaseService<ModelCompress, Long> i
modelManageService.saveAndCreateVersion(modelManageS, modelVersionS); modelManageService.saveAndCreateVersion(modelManageS, modelVersionS);
} }
} }
this.close();
} }
} }
@Override @Override
......
...@@ -239,7 +239,10 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement ...@@ -239,7 +239,10 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement
} }
log.info("发送服务端的消息:" + sendJson.toJSONString()); log.info("发送服务端的消息:" + sendJson.toJSONString());
if (receiveMsg.equals("process_completed")) { if (receiveMsg.equals("process_completed")) {
if (receiveJson.getBoolean("success") == true){ log.info("成功啦!");
this.close();
if (receiveJson.getBoolean("success")){
log.info("成功啦!!");
ModelVersion modelVersionS = new ModelVersion(); ModelVersion modelVersionS = new ModelVersion();
modelVersionS.setRunId(tuningRun.getRunId()); modelVersionS.setRunId(tuningRun.getRunId());
modelVersionS.setTaskId(tuningRun.getTaskId()); modelVersionS.setTaskId(tuningRun.getTaskId());
...@@ -263,7 +266,6 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement ...@@ -263,7 +266,6 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement
} else { } else {
tuningRun.setPublishStatus(-1); tuningRun.setPublishStatus(-1);
} }
this.close();
} }
} }
@Override @Override
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment