FluxApi - 使用Spring进行调用Flux接口

好久不见各位，最近我也是已经实习有一段时间了，博客也有一段时间没有更新了，但是最近着手完成了一个图片生成的项目，后端使用的是Springboot，在图片处理使用了最近很火的黑森林研究所的Flux模型，如何使用Java来访问和调用这个接口呢，接下来我就来分享一下。

首先你需要在yml文件里进行比较隐私信息的配置：
 

image:edit:tokens:single-deduct: 100blend-deduct: 50api: https://api.bfl.ai/v1/flux-kontext-maxkey: your_keyblend-api: https://api.bfl.ai/v1/flux-kontext-pro

key是需要你去官网自行注册的，具体注册方法可以参考：
Flux爆火，全网最全面最详细的Flux使用教程！-CSDN博客

接下来就是最重要的接口部分：
 

    /*** 核心方法：调用Flux API处理图片* @param imageData 图片数据（可以是URL或base64字符串）* @param prompt 提示词* @param isBase64 是否为base64格式* @param aspectRatio 宽高比字符串，格式为 "width:height"，为null时不传递*/private String processImageWithFluxApi(String imageData, String prompt, boolean isBase64, String aspectRatio) throws Exception {FluxRequestDTO request = new FluxRequestDTO(prompt, imageData);if (aspectRatio != null) {request.setAspectRatio(aspectRatio);}HttpHeaders headers = new HttpHeaders();headers.setContentType(MediaType.APPLICATION_JSON);headers.set("x-key", fluxApiKey);HttpEntity<FluxRequestDTO> entity = new HttpEntity<>(request, headers);ResponseEntity<FluxResponseDTO> response = restTemplate.postForEntity(fluxApiUrl, entity, FluxResponseDTO.class);if (response.getStatusCode() != HttpStatus.OK || response.getBody() == null) {throw new RuntimeException("Flux API调用失败");}FluxResponseDTO fluxResponse = response.getBody();String taskId = fluxResponse.getId();String pollingUrl = fluxResponse.getPollingUrl();if (taskId == null || pollingUrl == null) {throw new RuntimeException("无效的Flux API响应");}return pollFluxResult(pollingUrl);}private String pollFluxResult(String pollingUrl) throws Exception {int maxRetries = 60;int retryInterval = 5;for (int attempt = 0; attempt < maxRetries; attempt++) {ResponseEntity<FluxResponseDTO> response = restTemplate.getForEntity(pollingUrl, FluxResponseDTO.class);if (response.getStatusCode() == HttpStatus.OK && response.getBody() != null) {FluxResponseDTO fluxResponse = response.getBody();String status = fluxResponse.getStatus();if ("Ready".equals(status)) {if (fluxResponse.getResult() != null &&fluxResponse.getResult().getSample() != null) {return fluxResponse.getResult().getSample();}throw new RuntimeException("未获取到处理后的图片URL");} else if ("failed".equals(status)) {throw new RuntimeException("Flux处理任务失败");}}Thread.sleep(retryInterval * 1000);}throw new RuntimeException("处理超时，请稍后重试");}

这段代码主要实现了与Flux API交互处理图片的功能，具体作用如下：

1. 核心方法`processImageWithFluxApi`负责构建请求并调用Flux API： - 根据传入的图片数据（URL或base64）、提示词等参数创建请求对象 - 设置请求头信息，包括内容类型和API密钥 - 发送POST请求到Flux API获取处理任务 - 从响应中提取任务ID和轮询URL，调用轮询方法获取最终结果

2. 辅助方法`pollFluxResult`负责轮询获取处理结果： - 设置最大重试次数（60次）和重试间隔（5秒） - 循环向轮询URL发送GET请求查询任务状态 - 当任务状态为"Ready"时，返回处理后的图片样本 - 当任务状态为"failed"时，抛出处理失败异常 - 超过最大重试次数仍未完成时，抛出超时异常 整体流程实现了一个异步API调用模式：先提交处理任务，然后通过轮询方式等待处理完成并获取结果，适用于处理时间不确定的图片处理场景。

这就是这次分享，希望可以对你们有帮助，以后会提高发的频率的。