Add setting content-length middleware into operation stack

Adds the ContentLengthMiddleware to the operation's stack, and updates generated clients and protocol tests to use the content-length check. Depends on: * aws/smithy-go#108 * smithy-lang/smithy#491
jasdel · Jul 14, 2020 · 0de0a87 · 0de0a87
1 parent 8ed465e
commit 0de0a87
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/...-codegen/src/main/java/software/amazon/smithy/aws/go/codegen/AssembleMiddlewareStack.java b/...-codegen/src/main/java/software/amazon/smithy/aws/go/codegen/AssembleMiddlewareStack.java
@@ -4,6 +4,7 @@
 import java.util.Map;
 import software.amazon.smithy.aws.traits.auth.SigV4Trait;
 import software.amazon.smithy.aws.traits.auth.UnsignedPayloadTrait;
+import software.amazon.smithy.go.codegen.SmithyGoDependency;
 import software.amazon.smithy.go.codegen.SymbolUtils;
 import software.amazon.smithy.go.codegen.integration.GoIntegration;
 import software.amazon.smithy.go.codegen.integration.MiddlewareRegistrar;
@@ -43,6 +44,16 @@ public List<RuntimeClientPlugin> getClientPlugins() {
                         )
                         .build(),
 
+                // Add ContentLengthMiddleware to operation stack
+                RuntimeClientPlugin.builder()
+                        .registerMiddleware(MiddlewareRegistrar.builder()
+                                .resolvedFunction(SymbolUtils.createValueSymbolBuilder(
+                                        "AddContentLengthMiddleware", SmithyGoDependency.SMITHY_HTTP_TRANSPORT)
+                                        .build())
+                                .build()
+                        )
+                        .build(),
+
                 // Add endpoint serialize middleware to operation stack
                 RuntimeClientPlugin.builder()
                         .registerMiddleware(MiddlewareRegistrar.builder()

diff --git a/...-aws-go-codegen/src/main/java/software/amazon/smithy/aws/go/codegen/AwsProtocolUtils.java b/...-aws-go-codegen/src/main/java/software/amazon/smithy/aws/go/codegen/AwsProtocolUtils.java
@@ -76,7 +76,6 @@ static void generateHttpProtocolTests(GenerationContext context) {
                             writer.addUseImports(SmithyGoDependency.SMITHY_MIDDLEWARE);
                             writer.openBlock("[]APIOptionFunc{", "},", () -> {
                                 writer.openBlock("func(s *middleware.Stack) error {", "},", () -> {
-                                    writer.write("s.Build.Clear()");
                                     writer.write("s.Finalize.Clear()");
                                     writer.write("return nil");
                                 });