allow override of union deserializer (#1045)

smithy-lang · Oct 17, 2023 · 9c303eb · 9c303eb
1 parent 0cbd15a
commit 9c303eb
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/.changeset/green-files-crash.md b/.changeset/green-files-crash.md
@@ -0,0 +1,2 @@
+---
+---
diff --git a/...a/software/amazon/smithy/typescript/codegen/integration/HttpBindingProtocolGenerator.java b/...a/software/amazon/smithy/typescript/codegen/integration/HttpBindingProtocolGenerator.java
@@ -2407,7 +2407,7 @@ private HttpBinding readPayload(
                     + "= __expectObject(await parseBody(output.body, context));");
         } else if (target instanceof UnionShape) {
             // If payload is a Union, then we need to parse the string into JavaScript object.
-            writer.addImport("expectUnion", "__expectUnion", TypeScriptDependency.AWS_SMITHY_CLIENT);
+            importUnionDeserializer(writer);
             writer.write("const data: Record<string, any> | undefined "
                     + "= __expectUnion(await parseBody(output.body, context));");
         } else if (target instanceof StringShape || target instanceof DocumentShape) {
@@ -2422,6 +2422,10 @@ private HttpBinding readPayload(
         return binding;
     }
 
+    protected void importUnionDeserializer(TypeScriptWriter writer) {
+        writer.addImport("expectUnion", "__expectUnion", TypeScriptDependency.AWS_SMITHY_CLIENT);
+    }
+
     /**
      * Writes the code needed to deserialize the input payload of a request.
      *