diff --git a/src/main/java/com/ai/da/common/utils/SendRequestUtil.java b/src/main/java/com/ai/da/common/utils/SendRequestUtil.java index 05cc4415..28ac9406 100644 --- a/src/main/java/com/ai/da/common/utils/SendRequestUtil.java +++ b/src/main/java/com/ai/da/common/utils/SendRequestUtil.java @@ -39,6 +39,24 @@ public class SendRequestUtil { return null; } + public String sendAliYunPost(String apiUrl, String requestBody){ + // 发送POST请求 todo 异常处理 + HttpResponse execute = HttpRequest.post(apiUrl) + .header(Header.AUTHORIZATION, "Bearer " + ALIYUN_API_KEY) + .header(Header.CONTENT_TYPE, "application/json") + .body(requestBody) + .timeout(20000) // 设置超时时间20秒 + .execute(); + int status = execute.getStatus(); + if (status == 200){ + String body = execute.body(); + JSONObject bodyJson = JSONUtil.parseObj(body); + return body; + } + log.warn("请求失败,状态码为 : {}", status); + return null; + } + public static final String FREE_PIK = "https://api.freepik.com/v1/ai/beta/text-to-image/reimagine-flux"; public String sendFreepikPost( String requestBody){ // 发送POST请求 todo 异常处理 diff --git a/src/main/java/com/ai/da/model/vo/GenerateResultVO.java b/src/main/java/com/ai/da/model/vo/GenerateResultVO.java index 4dd1cc5e..ce983138 100644 --- a/src/main/java/com/ai/da/model/vo/GenerateResultVO.java +++ b/src/main/java/com/ai/da/model/vo/GenerateResultVO.java @@ -17,7 +17,7 @@ public class GenerateResultVO { private String url; - // Success || Executing || Invalid || Failed + // Success || Executing || Invalid || Fail private String status; private String category; diff --git a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java index d6f99877..d610569b 100644 --- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java @@ -1016,7 +1016,7 @@ public class GenerateServiceImpl extends ServiceImpl i } catch (Exception e) { log.error("异步处理图片转sketch失败, taskId: {}", taskId, e); // 更新redis - redisUtil.addToString(generateResultKey + ":" + taskId, new Gson().toJson(new GenerateResultVO(taskId, "Failed")), CommonConstant.GENERATE_RESULT_EXPIRE_TIME); + redisUtil.addToString(generateResultKey + ":" + taskId, new Gson().toJson(new GenerateResultVO(taskId, "Fail")), CommonConstant.GENERATE_RESULT_EXPIRE_TIME); } }, asyncTaskExecutor); return taskId; @@ -1583,7 +1583,7 @@ public class GenerateServiceImpl extends ServiceImpl i return new GenerateResultVO(taskId, null, null, "Executing"); } else { log.warn("万象 异步接口返回生成状态为:{}", taskStatus); - return new GenerateResultVO(taskId, null, null, "Failed"); + return new GenerateResultVO(taskId, null, null, "Fail"); } } catch (ApiException | NoApiKeyException e){ throw new RuntimeException(e.getMessage()); @@ -1655,7 +1655,7 @@ public class GenerateServiceImpl extends ServiceImpl i requestBody.set("input", input); requestBody.set("parameters", parameters); - String response = sendRequestUtil.sendPost(IMAGE_DETECT, requestBody.toString()); + String response = sendRequestUtil.sendAliYunPost(IMAGE_DETECT, requestBody.toString()); System.out.println("API响应: " + response); JSONObject jsonResponse = JSONUtil.parseObj(response); @@ -1838,7 +1838,7 @@ public class GenerateServiceImpl extends ServiceImpl i break; case STATUS_FAILED: case STATUS_UNKNOWN: - poseTransformationVO.setStatus("Failed"); + poseTransformationVO.setStatus("Fail"); break; case STATUS_RUNNING: case STATUS_PENDING: @@ -1847,7 +1847,7 @@ public class GenerateServiceImpl extends ServiceImpl i break; default: log.info("未知状态: {}", status); - poseTransformationVO.setStatus("Failed"); + poseTransformationVO.setStatus("Fail"); } poseTransformationVO.setTaskId(taskId); @@ -2104,7 +2104,7 @@ public class GenerateServiceImpl extends ServiceImpl i String status = respObj.getStr("status"); switch (status){ case "Task not found": - return "Failed"; + return "Fail"; case "Pending": case "Request Moderated": case "Content Moderated": @@ -2120,7 +2120,7 @@ public class GenerateServiceImpl extends ServiceImpl i return userBucket + "/" + objectName; case "Error": // 出错 - return "Failed"; + return "Fail"; } return null; } @@ -2133,8 +2133,8 @@ public class GenerateServiceImpl extends ServiceImpl i String objectName = accountId + "/imageToSketch/" + taskId + ".png"; String fluxResult = getFluxResult(taskId, objectName); if (Objects.isNull(generateDetail)){ - if (fluxResult.equals("Failed") || fluxResult.equals("Pending")){ - String status = fluxResult.equals("Failed") ? "Failed" : "Executing"; + if (fluxResult.equals("Fail") || fluxResult.equals("Pending")){ + String status = fluxResult.equals("Fail") ? "Fail" : "Executing"; return new GenerateResultVO(taskId, status); }