Skip to content

Commit 58d4642

Browse files
authored
fix: temporarily handle httpchecksum trait the same as httpchecksumrequired (#608)
1 parent 5732b83 commit 58d4642

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

smithy-kotlin-codegen/src/main/kotlin/software/amazon/smithy/kotlin/codegen/rendering/protocol/HttpProtocolClientGenerator.kt

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
*/
55
package software.amazon.smithy.kotlin.codegen.rendering.protocol
66

7+
import software.amazon.smithy.aws.traits.HttpChecksumTrait
78
import software.amazon.smithy.codegen.core.Symbol
89
import software.amazon.smithy.kotlin.codegen.core.*
910
import software.amazon.smithy.kotlin.codegen.integration.SectionId
@@ -248,7 +249,7 @@ abstract class HttpProtocolClientGenerator(
248249
.forEach { middleware ->
249250
middleware.render(ctx, op, writer)
250251
}
251-
if (op.hasTrait<HttpChecksumRequiredTrait>()) {
252+
if (op.checksumRequired()) {
252253
writer.addImport(RuntimeTypes.Http.Middlware.Md5ChecksumMiddleware)
253254
writer.write("op.install(#T())", RuntimeTypes.Http.Middlware.Md5ChecksumMiddleware)
254255
}
@@ -267,3 +268,7 @@ abstract class HttpProtocolClientGenerator(
267268
*/
268269
protected open fun renderAdditionalMethods(writer: KotlinWriter) { }
269270
}
271+
272+
// TODO https://github.com/awslabs/aws-sdk-kotlin/issues/557
273+
private fun OperationShape.checksumRequired(): Boolean =
274+
hasTrait<HttpChecksumRequiredTrait>() || getTrait<HttpChecksumTrait>()?.isRequestChecksumRequired == true

0 commit comments

Comments
 (0)