Commit f54e1196 authored by linpeiqin's avatar linpeiqin

不知道为啥又行了

parent cd0f3f53
...@@ -33,7 +33,6 @@ import org.springframework.beans.factory.annotation.Autowired; ...@@ -33,7 +33,6 @@ import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional; import org.springframework.transaction.annotation.Transactional;
import java.io.File;
import java.net.URI; import java.net.URI;
import java.util.List; import java.util.List;
import java.util.concurrent.atomic.AtomicInteger; import java.util.concurrent.atomic.AtomicInteger;
...@@ -209,6 +208,7 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement ...@@ -209,6 +208,7 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement
public void onOpen(ServerHandshake serverHandshake) { public void onOpen(ServerHandshake serverHandshake) {
log.info("-------------与大模型建立连接-------------"); log.info("-------------与大模型建立连接-------------");
} }
@Override @Override
public void onMessage(String message) { public void onMessage(String message) {
log.info("收到来自服务端的消息:" + message); log.info("收到来自服务端的消息:" + message);
...@@ -242,16 +242,18 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement ...@@ -242,16 +242,18 @@ 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")) {
// this.close(); // this.close();
if (receiveJson.getBoolean("success")){ if (receiveJson.getBoolean("success")) {
isSuccess.set(1); isSuccess.set(1);
} }
} }
} }
@Override @Override
public void onClose(int i, String s, boolean b) { public void onClose(int i, String s, boolean b) {
log.info("关闭连接:::" + "i = " + i + ":::s = " + s + ":::b = " + b); log.info("关闭连接:::" + "i = " + i + ":::s = " + s + ":::b = " + b);
} }
@Override @Override
public void onError(Exception e) { public void onError(Exception e) {
log.error("报错了:::" + e.getMessage()); log.error("报错了:::" + e.getMessage());
...@@ -259,15 +261,16 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement ...@@ -259,15 +261,16 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement
}.connect(); }.connect();
System.out.println("isSuccess.get():" + isSuccess.get()); System.out.println("isSuccess.get():" + isSuccess.get());
if (isSuccess.get() == 1) { if (isSuccess.get() == 1) {
saveAll(tuningRun,targetModelVersionURl,runPublishDto); saveAll(tuningRun, targetModelVersionURl, runPublishDto);
tuningRun.setPublishStatus(1); tuningRun.setPublishStatus(1);
} else { } else {
tuningRun.setPublishStatus(-1); tuningRun.setPublishStatus(-1);
} }
return this.updateById(tuningRun); return this.updateById(tuningRun);
} }
@Transactional(rollbackFor = Exception.class) @Transactional(rollbackFor = Exception.class)
public void saveAll(TuningRun tuningRun,String targetModelVersionURl,RunPublishDto runPublishDto){ public void saveAll(TuningRun tuningRun, String targetModelVersionURl, RunPublishDto runPublishDto) {
ModelVersion modelVersionS = new ModelVersion(); ModelVersion modelVersionS = new ModelVersion();
modelVersionS.setRunId(tuningRun.getRunId()); modelVersionS.setRunId(tuningRun.getRunId());
modelVersionS.setTaskId(tuningRun.getTaskId()); modelVersionS.setTaskId(tuningRun.getTaskId());
......
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