小修改

application-webadmin/src/main/java/com/yice/webadmin/app/service/impl/TuningRunServiceImpl.java

@@ -184,7 +184,7 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement
     }
 
     @SneakyThrows
-    @Transactional(rollbackFor = Exception.class)
+
     @Override
     public boolean publishToModelVersion(RunPublishDto runPublishDto) {
         String targetModelVersionURl;
@@ -239,10 +239,28 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement
                 }
                 log.info("发送服务端的消息:" + sendJson.toJSONString());
                 if (receiveMsg.equals("process_completed")) {
-                    log.info("成功啦！");
                     this.close();
                     if (receiveJson.getBoolean("success")){
-                        log.info("成功啦！！");
+                        saveAll(tuningRun,targetModelVersionURl,runPublishDto);
+                        tuningRun.setPublishStatus(1);
+                    } else {
+                        tuningRun.setPublishStatus(-1);
+                    }
+                }
+            }
+            @Override
+            public void onClose(int i, String s, boolean b) {
+                log.info("关闭连接:::" + "i = " + i + ":::s = " + s + ":::b = " + b);
+            }
+            @Override
+            public void onError(Exception e) {
+                log.error("报错了:::" + e.getMessage());
+            }
+        }.connect();
+        return this.updateById(tuningRun);
+    }
+    @Transactional(rollbackFor = Exception.class)
+    public void saveAll(TuningRun tuningRun,String targetModelVersionURl,RunPublishDto runPublishDto){
         ModelVersion modelVersionS = new ModelVersion();
         modelVersionS.setRunId(tuningRun.getRunId());
         modelVersionS.setTaskId(tuningRun.getTaskId());
@@ -262,22 +280,6 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement
             modelVersionS.setModelId(runPublishDto.getModelId());
             modelVersionService.saveNew(modelVersionS);
         }
-                        tuningRun.setPublishStatus(1);
-                    } else {
-                        tuningRun.setPublishStatus(-1);
-                    }
-                }
-            }
-            @Override
-            public void onClose(int i, String s, boolean b) {
-                log.info("关闭连接:::" + "i = " + i + ":::s = " + s + ":::b = " + b);
-            }
-            @Override
-            public void onError(Exception e) {
-                log.error("报错了:::" + e.getMessage());
-            }
-        }.connect();
-        return this.updateById(tuningRun);
     }
 
     /**