From b3dd2b79e47646f8161aa05cb87414b640544d82 Mon Sep 17 00:00:00 2001 From: HuangHai <10402852@qq.com> Date: Tue, 13 May 2025 09:20:37 +0800 Subject: [PATCH] 'commit' --- .../aiSupport/Util/KeLing/KlText2Image.java | 141 ++++++------------ 1 file changed, 46 insertions(+), 95 deletions(-) diff --git a/dsAiSupport/src/main/java/com/dsideal/aiSupport/Util/KeLing/KlText2Image.java b/dsAiSupport/src/main/java/com/dsideal/aiSupport/Util/KeLing/KlText2Image.java index aa664b39..ac3875f4 100644 --- a/dsAiSupport/src/main/java/com/dsideal/aiSupport/Util/KeLing/KlText2Image.java +++ b/dsAiSupport/src/main/java/com/dsideal/aiSupport/Util/KeLing/KlText2Image.java @@ -1,18 +1,18 @@ package com.dsideal.aiSupport.Util.KeLing; -import com.alibaba.fastjson.JSON; -import com.alibaba.fastjson.JSONArray; -import com.alibaba.fastjson.JSONObject; +import cn.hutool.core.io.FileUtil; +import cn.hutool.http.HttpRequest; +import cn.hutool.http.HttpResponse; +import cn.hutool.http.HttpUtil; +import cn.hutool.json.JSONArray; +import cn.hutool.json.JSONObject; +import cn.hutool.json.JSONUtil; import com.dsideal.aiSupport.Util.KeLing.Kit.KeLingJwtUtil; import com.dsideal.aiSupport.Util.KeLing.Kit.KlCommon; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import java.io.File; -import java.io.FileOutputStream; -import java.io.InputStream; -import java.net.HttpURLConnection; -import java.net.URL; import java.util.HashMap; import java.util.Map; @@ -39,42 +39,32 @@ public class KlText2Image extends KlCommon { requestBody.put("model_name", modelName); requestBody.put("prompt", prompt); - // 发送请求 - URL url = new URL(BASE_URL + GENERATION_PATH); - HttpURLConnection connection = (HttpURLConnection) url.openConnection(); - connection.setRequestMethod("POST"); - connection.setRequestProperty("Content-Type", "application/json"); - connection.setRequestProperty("Authorization", "Bearer " + jwt); - connection.setDoOutput(true); - - // 写入请求体 - String requestBodyJson = JSON.toJSONString(requestBody); - connection.getOutputStream().write(requestBodyJson.getBytes("UTF-8")); - - // 获取响应 - int responseCode = connection.getResponseCode(); - if (responseCode != 200) { - throw new Exception("请求失败，状态码：" + responseCode); + // 使用Hutool发送POST请求 + HttpResponse response = HttpRequest.post(BASE_URL + GENERATION_PATH) + .header("Content-Type", "application/json") + .header("Authorization", "Bearer " + jwt) + .body(JSONUtil.toJsonStr(requestBody)) + .execute(); + + // 检查响应状态码 + if (response.getStatus() != 200) { + throw new Exception("请求失败，状态码：" + response.getStatus()); } // 解析响应 - InputStream inputStream = connection.getInputStream(); - byte[] responseBytes = new byte[inputStream.available()]; - inputStream.read(responseBytes); - String responseBody = new String(responseBytes, "UTF-8"); - - JSONObject responseJson = JSON.parseObject(responseBody); + String responseBody = response.body(); + JSONObject responseJson = JSONUtil.parseObj(responseBody); log.info("生成图片响应：{}", responseBody); // 检查响应状态 - int code = responseJson.getIntValue("code"); + int code = responseJson.getInt("code"); if (code != 0) { - String message = responseJson.getString("message"); + String message = responseJson.getStr("message"); throw new Exception("生成图片失败：" + message); } // 获取任务ID - String taskId = responseJson.getJSONObject("data").getString("task_id"); + String taskId = responseJson.getJSONObject("data").getStr("task_id"); log.info("生成图片任务ID：{}", taskId); return taskId; @@ -91,32 +81,26 @@ public class KlText2Image extends KlCommon { // 获取JWT令牌 String jwt = KeLingJwtUtil.getJwt(); - // 发送请求 - URL url = new URL(BASE_URL + QUERY_PATH + taskId); - HttpURLConnection connection = (HttpURLConnection) url.openConnection(); - connection.setRequestMethod("GET"); - connection.setRequestProperty("Content-Type", "application/json"); - connection.setRequestProperty("Authorization", "Bearer " + jwt); - - // 获取响应 - int responseCode = connection.getResponseCode(); - if (responseCode != 200) { - throw new Exception("请求失败，状态码：" + responseCode); + // 使用Hutool发送GET请求 + HttpResponse response = HttpRequest.get(BASE_URL + QUERY_PATH + taskId) + .header("Content-Type", "application/json") + .header("Authorization", "Bearer " + jwt) + .execute(); + + // 检查响应状态码 + if (response.getStatus() != 200) { + throw new Exception("请求失败，状态码：" + response.getStatus()); } // 解析响应 - InputStream inputStream = connection.getInputStream(); - byte[] responseBytes = new byte[inputStream.available()]; - inputStream.read(responseBytes); - String responseBody = new String(responseBytes, "UTF-8"); - - JSONObject responseJson = JSON.parseObject(responseBody); + String responseBody = response.body(); + JSONObject responseJson = JSONUtil.parseObj(responseBody); log.info("查询任务状态响应：{}", responseBody); // 检查响应状态 - int code = responseJson.getIntValue("code"); + int code = responseJson.getInt("code"); if (code != 0) { - String message = responseJson.getString("message"); + String message = responseJson.getStr("message"); throw new Exception("查询任务状态失败：" + message); } @@ -131,61 +115,28 @@ public class KlText2Image extends KlCommon { * @throws Exception 下载过程中的异常 */ public static void downloadFile(String fileUrl, String saveFilePath) throws Exception { - URL url = new URL(fileUrl); - HttpURLConnection connection = (HttpURLConnection) url.openConnection(); - connection.setRequestMethod("GET"); - connection.setConnectTimeout(5000); - connection.setReadTimeout(60000); - - // 确保目录存在 - File file = new File(saveFilePath); - File parentDir = file.getParentFile(); - if (parentDir != null && !parentDir.exists()) { - parentDir.mkdirs(); - log.info("创建目录: {}", parentDir.getAbsolutePath()); - } - - // 获取输入流 - try (InputStream in = connection.getInputStream(); - FileOutputStream out = new FileOutputStream(saveFilePath)) { - - byte[] buffer = new byte[4096]; - int bytesRead; - - // 读取数据并写入文件 - while ((bytesRead = in.read(buffer)) != -1) { - out.write(buffer, 0, bytesRead); - } - - log.info("文件下载成功，保存路径: {}", saveFilePath); + try { + // 使用Hutool下载文件 + long fileSize = HttpUtil.downloadFile(fileUrl, FileUtil.file(saveFilePath)); + log.info("文件下载成功，保存路径: {}, 文件大小: {}字节", saveFilePath, fileSize); } catch (Exception e) { log.error("文件下载失败: {}", e.getMessage(), e); throw e; - } finally { - connection.disconnect(); } } public static void main(String[] args) throws Exception { - - // 确保目录存在 - File dir = new File(basePath); - if (!dir.exists()) { - dir.mkdirs(); - log.info("创建目录: {}", basePath); - } - // 提示词和模型名称 String prompt = "一只可爱的小猫咪在草地上玩耍，阳光明媚"; - String modelName = "kling-v2"; // 可选：kling-v1, kling-v1-5, kling-v2 + String modelName = "kling-v1"; // 可选：kling-v1, kling-v1-5, kling-v2 // 添加重试逻辑 int generateRetryCount = 0; int maxGenerateRetries = 1000; // 最大重试次数 int generateRetryInterval = 5000; // 重试间隔（毫秒） - String taskId = null; - while (generateRetryCount < maxGenerateRetries) { + String taskId; + while (true) { try { taskId = generateImage(prompt, modelName); break; @@ -214,10 +165,10 @@ public class KlText2Image extends KlCommon { while (queryRetryCount < maxQueryRetries) { JSONObject result = queryTaskStatus(taskId); JSONObject data = result.getJSONObject("data"); - String taskStatus = data.getString("task_status"); + String taskStatus = data.getStr("task_status"); if ("failed".equals(taskStatus)) { - String taskStatusMsg = data.getString("task_status_msg"); + String taskStatusMsg = data.getStr("task_status_msg"); log.error("任务失败: {}", taskStatusMsg); break; } else if ("succeed".equals(taskStatus)) { @@ -227,8 +178,8 @@ public class KlText2Image extends KlCommon { for (int i = 0; i < images.size(); i++) { JSONObject image = images.getJSONObject(i); - int index = image.getIntValue("index"); - String imageUrl = image.getString("url"); + int index = image.getInt("index"); + String imageUrl = image.getStr("url"); // 下载图片 String saveImagePath = basePath + "image_" + index + ".png";