32第4天

第4天
昨天的成就: 实现了图片生成的Job状态查询循环，每3秒轮询一次直到完成
遇到的困难: 需要处理API返回的statusCode为字符串而非整数的情况
今天的任务: 添加图片URL提取和响应构建逻辑
java
public GenerationResponse generateImage(GenerationRequest request) {
logger.info("开始生成图片：{}", request.getPrompt());

try {
Credential cred = new Credential(
config.getSecretId(),
config.getSecretKey()
);

HttpProfile httpProfile = new HttpProfile();
httpProfile.setEndpoint(config.getEndpoint());

ClientProfile clientProfile = new ClientProfile();
clientProfile.setHttpProfile(httpProfile);

HunyuanClient client = new HunyuanClient(
cred,
"ap-guangzhou",
clientProfile
);

SubmitHunyuanImageJobRequest req = new SubmitHunyuanImageJobRequest();
req.setPrompt(buildPrompt(request));

SubmitHunyuanImageJobResponse resp = client.SubmitHunyuanImageJob(req);
String jobId = resp.getJobId();
logger.info("任务提交成功，JobId: {}", jobId);

// 轮询查询任务状态
com.tencentcloudapi.hunyuan.v20230901.models.QueryHunyuanImageJobRequest queryReq =
new com.tencentcloudapi.hunyuan.v20230901.models.QueryHunyuanImageJobRequest();
queryReq.setJobId(jobId);

com.tencentcloudapi.hunyuan.v20230901.models.QueryHunyuanImageJobResponse queryResp;
int maxRetries = 60; // 最多等待3分钟
int retryCount = 0;

do {
Thread.sleep(3000); // 每3秒查询一次
queryResp = client.QueryHunyuanImageJob(queryReq);
retryCount++;

logger.info("任务状态：{} - {} (第{}次查询)",
queryResp.getJobStatusCode(),
queryResp.getJobStatusMsg(),
retryCount);

if (retryCount >= maxRetries) {
return buildErrorResponse("任务超时，请稍后重试");
}

} while (!"5".equals(queryResp.getJobStatusCode())); // 5表示完成

// 提取图片URL
String[] images = queryResp.getResultImage();
if (images == null || images.length == 0) {
return buildErrorResponse("未获取到生成的图片");
}

String imageUrl = images[0];
logger.info("图片生成成功，URL: {}", imageUrl);

// 构建成功响应
GenerationResponse response = new GenerationResponse();
response.setSuccess(true);
response.setStatus("completed");
response.setResultUrl(imageUrl);
response.setTaskId(jobId);

return response;

} catch (InterruptedException e) {
Thread.currentThread().interrupt();
logger.error("线程被中断", e);
return buildErrorResponse("生成过程被中断");
} catch (Exception e) {
logger.error("图片生成失败", e);
return buildErrorResponse("生成失败：" + e.getMessage());
}
}