Use xmlNameTrait when outputting xmlRootNodeName

soto-project · Oct 16, 2024 · 066e10e · 066e10e
1 parent 2a7dc09
commit 066e10e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/Sources/SotoCodeGeneratorLib/AwsService+shapes.swift b/Sources/SotoCodeGeneratorLib/AwsService+shapes.swift
@@ -174,7 +174,12 @@ extension AwsService {
             } else if isInput {
                 // set XML root node name.
                 if serviceProtocolTrait is AwsProtocolsRestXmlTrait {
-                    xmlRootNodeName = payloadMember.key
+                    // use XML name if trait exists
+                    if let xmlNameTrait = payloadMember.value.trait(type: XmlNameTrait.self) {
+                        xmlRootNodeName = xmlNameTrait.value
+                    } else {
+                        xmlRootNodeName = payloadMember.key
+                    }
                 }
                 // currently only support request streaming of blobs
                 if payload is BlobShape,