Skip to content

Commit 8a6cb03

Browse files
committed
Add event-stream middleware if operation has eventstream input
1 parent f03dc94 commit 8a6cb03

File tree

3 files changed

+8
-4
lines changed

3 files changed

+8
-4
lines changed

smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/integration/EventStreamGenerator.java

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,12 @@ public List<RuntimeClientPlugin> getClientPlugins() {
111111
RuntimeClientPlugin.Convention.HAS_CONFIG
112112
)
113113
.servicePredicate(EventStreamGenerator::hasEventStream)
114-
.build()
114+
.build(),
115+
RuntimeClientPlugin.builder()
116+
.withConventions(
117+
TypeScriptDependency.MIDDLEWARE_EVENT_STREAM.dependency,
118+
"EventStream"
119+
).operationPredicate((m, s, o) -> operationHasEventStreamInput(m, o)).build()
115120
);
116121
}
117122

smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/integration/HttpBindingProtocolGenerator.java

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,6 @@
5151
import software.amazon.smithy.model.shapes.StructureShape;
5252
import software.amazon.smithy.model.shapes.TimestampShape;
5353
import software.amazon.smithy.model.shapes.UnionShape;
54-
import software.amazon.smithy.model.traits.EndpointTrait;
5554
import software.amazon.smithy.model.traits.ErrorTrait;
5655
import software.amazon.smithy.model.traits.EventHeaderTrait;
5756
import software.amazon.smithy.model.traits.EventPayloadTrait;

smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/integration/HttpProtocolGeneratorUtils.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,14 +24,14 @@
2424
import software.amazon.smithy.codegen.core.Symbol;
2525
import software.amazon.smithy.codegen.core.SymbolProvider;
2626
import software.amazon.smithy.codegen.core.SymbolReference;
27+
import software.amazon.smithy.model.Model;
2728
import software.amazon.smithy.model.pattern.Pattern;
2829
import software.amazon.smithy.model.shapes.MemberShape;
2930
import software.amazon.smithy.model.shapes.OperationShape;
3031
import software.amazon.smithy.model.shapes.Shape;
3132
import software.amazon.smithy.model.shapes.StructureShape;
32-
import software.amazon.smithy.model.traits.EndpointTrait;
33-
import software.amazon.smithy.model.Model;
3433
import software.amazon.smithy.model.shapes.UnionShape;
34+
import software.amazon.smithy.model.traits.EndpointTrait;
3535
import software.amazon.smithy.model.traits.TimestampFormatTrait.Format;
3636
import software.amazon.smithy.typescript.codegen.TypeScriptDependency;
3737
import software.amazon.smithy.typescript.codegen.TypeScriptWriter;

0 commit comments

Comments
 (0)