Spring Boot 3.4.5 集成即梦AI的Webhook回调与幂等性设计：Polling vs Webhook的选型对比与实战
上周有个需求：即梦AI图像生成任务平均耗时45秒，P99延迟飙到3分钟。团队默认用定时轮询查任务状态，结果DB连接池被打满，线上告警连续响了40分钟。问题不在即梦AI，在我们选错了任务同步策略。
即梦AI（Jimeng AI）是字节跳动旗下的AI创作平台，支持文生图、图生图、视频生成等能力。后端集成时，任务提交后需要获取结果——主动轮询（Polling）和被动回调（Webhook）是两条路。多数人默认用轮询，因为简单直接。但轮询在高并发下的代价远超想象：假设100个并发任务，每5秒轮询一次，10分钟内产生12000次无效请求，其中95%拿不到结果。
即梦AI的Webhook回调机制支持配置回调URL，任务完成后推送结果。回调请求包含签名头，用于验证请求合法性。幂等性设计确保同一任务不会因网络重试导致重复处理。这两个特性被大多数人忽略，恰恰是生产环境稳定性的关键。
方案简介
Polling方案（Spring Boot 3.4.5 + Spring Retry 2.0.5）：定时任务轮询即梦AI的任务查询接口，获取状态后更新本地DB。实现简单，但资源消耗随并发量线性增长，且无法处理网络抖动导致的重复请求。
Webhook方案（Spring Boot 3.4.5 + Redis 7.2.5 + Redisson 3.36.0）：配置即梦AI的回调URL，任务完成后接收推送。配合HMAC-SHA256签名验证和幂等性Key存储，确保请求合法且处理唯一。
混合方案（Polling + Webhook Fallback）：以Webhook为主通道，轮询作为兜底。Webhook失败时自动降级为轮询，确保任务不丢失。
多维度对比
| 维度 | Polling | Webhook | 混合方案 ||——|———|———|———-|| 实现复杂度 | 低（约20行） | 中（约80行） | 高（约150行） || 并发承载能力 | 低（每任务1连接） | 高（推送式，无连接占用） | 高 || 资源消耗 | 高（12000次/10min/100任务） | 低（仅任务完成时触发） | 中 || 延迟 | 取决于轮询间隔（5s） | 实时（<1s） | 实时+兜底 || 可靠性 | 中（依赖定时器） | 中（依赖回调可达性） | 高（双通道） || 安全验证 | 无（轮询请求无签名） | 有（HMAC-SHA256签名） | 有 || 幂等性保障 | 无（需额外实现） | 有（天然支持） | 有 || 调试难度 | 低 | 中（需模拟回调） | 高 |
官方文档推荐轮询方式，但在我们的场景下反而更糟——100并发时DB连接池被打满，而Webhook方案下连接池使用率不到10%。这个"简单"的推荐，在高并发下就是定时炸弹。
深入分析
Webhook签名验证
即梦AI的回调请求包含 X-Jimeng-Signature 头，使用HMAC-SHA256算法生成。验证逻辑如下：
```java@RestController@RequestMapping("/api/jimeng/callback")public class JimengWebhookController {
@Value("${jimeng.webhook.secret}")private String webhookSecret;
private final ObjectMapper objectMapper = new ObjectMapper();private final IdempotencyService idempotencyService;private final TaskService taskService;
public JimengWebhookController(IdempotencyService idempotencyService,TaskService taskService) {this.idempotencyService = idempotencyService;this.taskService = taskService;}
@PostMappingpublic ResponseEntity handleCallback(@RequestHeader("X-Jimeng-Signature") String signature,@RequestBody String payload) {
// HMAC-SHA256 签名验证String expectedSignature = computeHmacSha256(payload, webhookSecret);if (!MessageDigest.isEqual(expectedSignature.getBytes(),signature.getBytes())) {log.warn("Webhook signature mismatch, payload: {}", payload);return ResponseEntity.status(HttpStatus.FORBIDDEN).build();}
// 幂等性检查JimengCallbackEvent event = objectMapper.readValue(payload, JimengCallbackEvent.class);if (idempotencyService.isDuplicate(event.getTaskId())) {log.info("Duplicate callback for task: {}", event.getTaskId());return ResponseEntity.ok().build();}
// 业务处理taskService.processCallback(event);idempotencyService.markProcessed(event.getTaskId());return ResponseEntity.ok().build();}
private String computeHmacSha256(String data, String secret
你在实际项目中有遇到类似问题吗？欢迎在评论区分享你的经验和解决方案。
