Skip to content

Commit 6a6615c

Browse files
authored
fix(codegen): fix header type codegen for event streams (#1694)
1 parent 246ab92 commit 6a6615c

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -333,9 +333,9 @@ private void writeEventHeaders(GenerationContext context, StructureShape event)
333333
for (MemberShape headerMember : headerMembers) {
334334
String memberName = headerMember.getMemberName();
335335
Shape target = model.expectShape(headerMember.getTarget());
336-
writer.openBlock("if (input.$L) {", "}", memberName, () -> {
336+
writer.openBlock("if (input.$L != null) {", "}", memberName, () -> {
337337
writer.write("headers[$1S] = { type: $2S, value: input.$1L }", memberName,
338-
getEventHeaderType(headerMember));
338+
getEventHeaderType(target));
339339
});
340340
}
341341
}
@@ -356,7 +356,7 @@ private String getEventHeaderType(Shape shape) {
356356
case BLOB:
357357
return "binary";
358358
default:
359-
return "binary";
359+
throw new IllegalArgumentException("Unsupported event header shape type: " + shape.getType());
360360
}
361361
}
362362

0 commit comments

Comments
 (0)