Check HttpChecksumRequiredTrait for HTTP checksum requirements (#47)

* Check HttpChecksumRequiredTrait * swift format
soto-project · Mar 25, 2022 · a804bd6 · a804bd6
1 parent 21d535b
commit a804bd6
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/Sources/SotoCodeGeneratorLib/AwsService+shapes.swift b/Sources/SotoCodeGeneratorLib/AwsService+shapes.swift
@@ -107,8 +107,13 @@ extension AwsService {
 
         // get payload options
         let operationShape = shape.trait(type: SotoRequestShapeTrait.self)?.operationShape
-        if operationShape?.hasTrait(type: HttpChecksumRequiredTrait.self) == true {
-            shapeOptions.append("md5ChecksumRequired")
+        if operationShape?.hasTrait(type: AwsHttpChecksumTrait.self) == true {
+            shapeOptions.append("checksumHeader")
+        }
+        if operationShape?.hasTrait(type: HttpChecksumRequiredTrait.self) == true ||
+            operationShape?.trait(type: AwsHttpChecksumTrait.self)?.requestChecksumRequired == true
+        {
+            shapeOptions.append("checksumRequired")
         }
         // search for content-md5 header
         if let members = shape.members {