Fix API breakage of FlexibleChecksumsRequestInterceptor

awslabs · Feb 14, 2024 · b308381 · b308381
1 parent 73a39db
commit b308381
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...kotlin/aws/sdk/kotlin/codegen/customization/flexiblechecksums/FlexibleChecksumsRequest.kt b/...kotlin/aws/sdk/kotlin/codegen/customization/flexiblechecksums/FlexibleChecksumsRequest.kt
@@ -44,14 +44,15 @@ class FlexibleChecksumsRequest : KotlinIntegration {
 
         override fun render(ctx: ProtocolGenerator.GenerationContext, op: OperationShape, writer: KotlinWriter) {
             val interceptorSymbol = RuntimeTypes.HttpClient.Interceptors.FlexibleChecksumsRequestInterceptor
+            val inputSymbol = ctx.symbolProvider.toSymbol(ctx.model.expectShape(op.inputShape))
 
             val httpChecksumTrait = op.getTrait<HttpChecksumTrait>()!!
 
             val requestAlgorithmMember = ctx.model.expectShape<StructureShape>(op.input.get())
                 .members()
                 .first { it.memberName == httpChecksumTrait.requestAlgorithmMember.get() }
 
-            writer.write("op.interceptors.add(#T())", interceptorSymbol)
+            writer.write("op.interceptors.add(#T<#T>())", interceptorSymbol, inputSymbol)
             writer.withBlock("input.${requestAlgorithmMember.defaultName()}?.let {", "}") {
                 writer.write("op.context[#T.ChecksumAlgorithm] = it.value", RuntimeTypes.HttpClient.Operation.HttpOperationContext)
             }