Skip to content

Commit f9446dc

Browse files
authored
EventStream Fixes (#1647)
1 parent b03e3da commit f9446dc

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

codegen/smithy-aws-go-codegen/src/main/java/software/amazon/smithy/aws/go/codegen/AwsEventStreamUtils.java

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -780,7 +780,7 @@ private static void generateEventStreamWriter(
780780
var writeCloser = getSymbol("WriteCloser", SmithyGoDependency.IO);
781781
var signerInterface = getModuleSymbol(settings, EVENT_STREAM_SIGNER_INTERFACE);
782782

783-
var messageSymbol = getEventStreamSymbol("Message", false);
783+
var messageSymbol = getEventStreamSymbol("Message", true);
784784

785785
if (withInitialMessages) {
786786
generateEventStreamWriterMessageWrapper(eventStream, service, symbolProvider, writer, eventUnionSymbol);
@@ -937,6 +937,9 @@ private static void generateEventStreamWriter(
937937
writer.write("w.serializationBuffer.Reset()").write("")
938938
.write("eventMessage := $T{}", messageSymbol).write("");
939939

940+
var eventStreamSerializerName = getEventStreamSerializerName(eventStream, service,
941+
context.getProtocolName());
942+
940943
if (withInitialMessages) {
941944
var initialRequestType = getWriterEventWrapperInitialRequestType(symbolProvider, eventStream,
942945
service);
@@ -956,14 +959,14 @@ private static void generateEventStreamWriter(
956959
default:
957960
return nil, $T("unknown event wrapper type: %v", event)
958961
}
959-
""", initialRequestType, messageEventType, errorf);
962+
""", initialRequestType, messageEventType, eventStreamSerializerName, errorf);
960963
} else {
961964
writer.write("""
962965
if err := $L(event, &eventMessage); err != nil {
963966
return nil, err
964967
}
965968
""",
966-
getEventStreamSerializerName(eventStream, service, context.getProtocolName()));
969+
eventStreamSerializerName);
967970
}
968971

969972
writer.write("""

0 commit comments

Comments
 (0)