feat: middleware changes (#1180)

* Add signing properties customization hook implementation to sigv4 auth scheme, used via polymorphism in AuthSchemeMiddleware. --------- Co-authored-by: Sichan Yoo <[email protected]>
awslabs · Oct 25, 2023 · f18ac03 · f18ac03
1 parent 981f9f9
commit f18ac03
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/Sources/Core/AWSClientRuntime/Auth/AuthSchemes/SigV4AuthScheme.swift b/Sources/Core/AWSClientRuntime/Auth/AuthSchemes/SigV4AuthScheme.swift
@@ -11,4 +11,11 @@ public struct SigV4AuthScheme: ClientRuntime.AuthScheme {
     public let schemeID: String = "aws.auth#sigv4"
     public let signer: ClientRuntime.Signer = AWSSigV4Signer()
     public let idKind: ClientRuntime.IdentityKind = .aws
+
+    public func customizeSigningProperties(signingProperties: Attributes, context: HttpContext) -> Attributes {
+        var copy = signingProperties
+        copy.set(key: AttributeKeys.bidirectionalStreaming, value: context.isBidirectionalStreamingEnabled())
+        copy.set(key: AttributeKeys.signingRegion, value: context.getSigningRegion())
+        return copy
+    }
 }