function to find if operation payload is union

smithy-lang · Nov 10, 2023 · cae26c5 · cae26c5
1 parent 3e649eb
commit cae26c5
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/...hy-kotlin-codegen/src/main/kotlin/software/amazon/smithy/kotlin/codegen/model/ShapeExt.kt b/...hy-kotlin-codegen/src/main/kotlin/software/amazon/smithy/kotlin/codegen/model/ShapeExt.kt
@@ -192,6 +192,15 @@ val Shape.isSparse: Boolean
 val Shape.isStreaming: Boolean
     get() = hasTrait<StreamingTrait>()
 
+/**
+ * Returns boolean indicating if operation input is union shaped
+ */
+fun OperationShape.payloadIsUnionShape(model: Model): Boolean {
+    val requestShape = model.expectShape<StructureShape>(input.get())
+    val payload = requestShape.findMemberWithTrait<HttpPayloadTrait>(model)?.targetOrSelf(model)
+    return payload is UnionShape
+}
+
 /**
  * Test if a member targets an event stream
  */