Commit f54e1196 authored by linpeiqin's avatar linpeiqin

不知道为啥又行了

parent cd0f3f53
......@@ -33,7 +33,6 @@ import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import java.io.File;
import java.net.URI;
import java.util.List;
import java.util.concurrent.atomic.AtomicInteger;
......@@ -209,6 +208,7 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement
public void onOpen(ServerHandshake serverHandshake) {
log.info("-------------与大模型建立连接-------------");
}
@Override
public void onMessage(String message) {
log.info("收到来自服务端的消息:" + message);
......@@ -242,16 +242,18 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement
}
log.info("发送服务端的消息:" + sendJson.toJSONString());
if (receiveMsg.equals("process_completed")) {
// this.close();
if (receiveJson.getBoolean("success")){
// this.close();
if (receiveJson.getBoolean("success")) {
isSuccess.set(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());
......@@ -259,15 +261,16 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement
}.connect();
System.out.println("isSuccess.get():" + isSuccess.get());
if (isSuccess.get() == 1) {
saveAll(tuningRun,targetModelVersionURl,runPublishDto);
saveAll(tuningRun, targetModelVersionURl, runPublishDto);
tuningRun.setPublishStatus(1);
} else {
tuningRun.setPublishStatus(-1);
}
return this.updateById(tuningRun);
}
@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();
modelVersionS.setRunId(tuningRun.getRunId());
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