aidlab/aida_back
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

generate 接口疏通

Browse Source
This commit is contained in:
xupei
2023-09-19 15:52:33 +08:00
parent aef2bb7457
commit 3afe1c715d
1 changed files with 22 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
src/main/java/com/ai/da/python/PythonService.java
View File

@@ -1636,7 +1636,7 @@ public class PythonService {
content.put("user_id", userId);
content.put("image_url", url);
content.put("category", category);
content.put("mode",mode); //
content.put("mode",mode);
content.put("str", text);
content.put("version",2);
RequestBody body = RequestBody.create(mediaType, JSON.toJSONString(content));
@@ -1650,32 +1650,32 @@ public class PythonService {
.build();
Response response = null;
String bodyString = null;
// try {
try {
log.info("generateSketchOrPrint请求入参content###{}", JSON.toJSONString(content));
// response = client.newCall(request).execute();
// bodyString = response.body().string();
bodyString = "{\n" +
" \"code\": 200,\n" +
" \"data\": {\n" +
" \"list\": [\n" +
" \"aida-users/12/print_1695088687_0.png\"\n" +
" ]\n" +
" },\n" +
" \"msg\": \"OK!\"\n" +
"}";
// } catch (IOException ioException) {
// log.error("PythonService##generateSketchOrPrint异常###{}", ExceptionUtil.getThrowableList(ioException));
// }
response = client.newCall(request).execute();
bodyString = response.body().string();
// bodyString = "{\n" +
// " \"code\": 200,\n" +
// " \"data\": {\n" +
// " \"list\": [\n" +
// " \"aida-users/12/print_1695088687_0.png\"\n" +
// " ]\n" +
// " },\n" +
// " \"msg\": \"OK!\"\n" +
// "}";
} catch (IOException ioException) {
log.error("PythonService##generateSketchOrPrint异常###{}", ExceptionUtil.getThrowableList(ioException));
}
//去除限流
AccessLimitUtils.validateOut("generateSketchOrPrint");
// 生成失败
// if (Objects.isNull(response) || StringUtil.isNullOrEmpty(bodyString)) {
// log.error("PythonService##generateSketchOrPrint异常###{}", "response or body is empty!");
// throw new BusinessException("generate exception!");
// }
if (Objects.isNull(response) || StringUtil.isNullOrEmpty(bodyString)) {
log.error("PythonService##generateSketchOrPrint异常###{}", "response or body is empty!");
throw new BusinessException("generate exception!");
}
JSONObject jsonObject = JSON.parseObject(bodyString);
// Boolean result = JSON.parseObject(JSON.toJSONString(response)).getBoolean("successful");
Boolean result = Boolean.TRUE;
Boolean result = JSON.parseObject(JSON.toJSONString(response)).getBoolean("successful");
// Boolean result = Boolean.TRUE;
if (result) {
return setGenerateImageList(jsonObject.getJSONObject("data"));
}