如何在Ktor插件中计算请求体哈希以实现幂等POST插件？

核心结论 #

Ktor插件完全可以访问请求体，但要解决请求体流只能读取一次的问题，同时兼容Content Negotiation的解析逻辑，下面是具体实现方案：

1. 处理请求体缓存与哈希计算 #

请求体是ByteReadChannel，只能消费一次，所以必须先把内容缓存到内存（大请求可考虑磁盘缓存），计算哈希后再放回流，让后续的Content Negotiation能正常解析。

示例哈希计算工具函数：

fun calculateRequestBodyHash(bodyBytes: ByteArray): String {
    return MessageDigest.getInstance("SHA-256")
        .digest(bodyBytes)
        .joinToString("") { "%02x".format(it) }
}

2. 编写自定义幂等插件 #

用Ktor的createRouteScopedPlugin封装逻辑，在插件中拦截请求体，完成缓存、哈希计算，同时处理Idempotency-Key头：

val IdempotencyPlugin = createRouteScopedPlugin("IdempotencyPlugin") {
    onCall { call ->
        if (call.request.httpMethod == HttpMethod.Post) {
            // 读取并缓存请求体
            val originalBody = call.receiveChannel()
            val bodyBytes = originalBody.readRemaining().readBytes()
            // 计算请求体哈希
            val bodyHash = calculateRequestBodyHash(bodyBytes)
            
            // 将缓存的内容放回请求流，供后续Content Negotiation解析
            call.request.setBody(ByteReadChannel(bodyBytes))
            
            // 幂等逻辑处理：结合Idempotency-Key和bodyHash做判断
            val idempotencyKey = call.request.headers["Idempotency-Key"]
            idempotencyKey?.let { key ->
                // 示例：检查存储中是否已有该key+hash的请求记录
                // 存在则返回之前的响应，不存在则执行请求并记录结果
            }
        }
    }
}

3. 兼容Content Negotiation的关键 #

必须先安装自定义幂等插件，再安装Content Negotiation，确保请求体缓存逻辑先执行，Content Negotiation拿到的是缓存后的流，不会出现读取失败的情况：

fun Application.module() {
    install(IdempotencyPlugin) // 先装幂等插件
    install(ContentNegotiation) {
        json()
    }
    // 其他路由配置
}

4. 替代自定义POST方法的方案 #

插件方式是Ktor的标准扩展风格，可统一处理所有POST请求，也可指定特定路由生效，无需自定义POST方法：

routing {
    // 仅对该路由下的POST请求启用幂等逻辑
    route("/api/orders") {
        install(IdempotencyPlugin)
        post {
            // 正常处理请求，此时请求体已完成缓存与哈希计算
            val order = call.receive<Order>()
            // ... 业务逻辑
        }
    }
}

内容的提问来源于stack exchange，提问作者agathis